MAFUNet: Multi-Attention Fusion Network for Medical Image Segmentation

被引:1
作者
Wang, Lili [1 ]
Zhao, Jiayu [1 ]
Yang, Hailu [1 ]
机构
[1] Harbin Univ Sci & Technol, Sch Comp Sci & Technol, Harbin 150080, Peoples R China
基金
黑龙江省自然科学基金; 中国国家自然科学基金;
关键词
Deep learning; medical image segmentation; model lightweight; spatial attention mechanism;
D O I
10.1109/ACCESS.2023.3320685
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The purpose of medical image segmentation is to identify target organs, tissues or lesion areas at the pixel level to help doctors evaluate and prevent diseases. Therefore, the model is required to have higher accuracy and robust representation. At present, the proposed models focus on the improvement of performance, and ignore the number of trainable parameters. This paper proposes a lightweight ECA-Residual module to build a model encoder, which can effectively extract features while reducing the number of parameters. The feature fusion of encoder and decoder using simple skip connections will produce semantic differences. Therefore, in this paper a spatial attention gating module is designed to solve this problem. The module suppresses the image irrelevant area and improves the performance of the model while ensuring the computational efficiency. The experiment is carried out on the Synapse dataset, and the results are superior to the current advanced models in terms of accuracy and parameter quantity with 84.21 % dice coefficient and 12.79 HD95. In addition, the accuracy of the model on the ACDC dataset is also better than the current advanced model, which proves the robustness and generalization of the model.
引用
收藏
页码:109793 / 109802
页数:10
相关论文
共 29 条
  • [1] Azad R, 2022, Arxiv, DOI [arXiv:2212.13504, DOI 10.48550/ARXIV.2212.13504]
  • [2] Cao Hu, 2023, Computer Vision - ECCV 2022 Workshops: Proceedings. Lecture Notes in Computer Science (13803), P205, DOI 10.1007/978-3-031-25066-8_9
  • [3] Chen J., 2021, arXiv
  • [4] Howard AG, 2017, Arxiv, DOI [arXiv:1704.04861, DOI 10.48550/ARXIV.1704.04861]
  • [5] Zero-Reference Deep Curve Estimation for Low-Light Image Enhancement
    Guo, Chunle
    Li, Chongyi
    Guo, Jichang
    Loy, Chen Change
    Hou, Junhui
    Kwong, Sam
    Cong, Runmin
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 1777 - 1786
  • [6] Guo MH, 2022, Arxiv, DOI arXiv:2202.09741
  • [7] No one knows what attention is
    Hommel, Bernhard
    Chapman, Craig S.
    Cisek, Paul
    Neyedli, Heather F.
    Song, Joo-Hyun
    Welsh, Timothy N.
    [J]. ATTENTION PERCEPTION & PSYCHOPHYSICS, 2019, 81 (07) : 2288 - 2303
  • [8] Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/CVPR.2018.00745, 10.1109/TPAMI.2019.2913372]
  • [9] Huang HM, 2020, INT CONF ACOUST SPEE, P1055, DOI [10.1109/icassp40776.2020.9053405, 10.1109/ICASSP40776.2020.9053405]
  • [10] Huang X., 2021, arXiv