FSOU-Net: Feature supplement and optimization U-Net for 2D medical image segmentation

被引:2
|
作者
Wang, Yongtao [1 ,2 ]
Tian, Shengwei [1 ,2 ]
Yu, Long [3 ]
Wu, Weidong [4 ,5 ]
Zhang, Dezhi [4 ]
Wang, Junwen [1 ,2 ]
Cheng, Junlong [6 ]
机构
[1] Xinjiang Univ, Coll Software Engn, Urumqi, Xinjiang, Peoples R China
[2] Xinjiang Univ, Key Lab Software Engn Technol, Urumqi, Xinjiang, Peoples R China
[3] Xinjiang Univ, Coll Informat Sci & Engn, Urumqi, Xinjiang, Peoples R China
[4] Peoples Hosp Xinjiang Uygur Autonomous Reg, Urumqi, Xinjiang, Peoples R China
[5] China Xinjiang Key Lab Dermatol Res XJYS1707, Urumqi, Xinjiang, Peoples R China
[6] Sichuan Univ, Coll Comp Sci, Chengdu, Sichuan, Peoples R China
关键词
Deep learning algorithm; medical image analysis; semantic segmentation; multi-scale; convolutional neural networks; OPTIC DISC;
D O I
10.3233/THC-220174
中图分类号
R19 [保健组织与事业(卫生事业管理)];
学科分类号
摘要
BACKGROUND: The results of medical image segmentation can provide reliable evidence for clinical diagnosis and treatment. The U-Net proposed previously has been widely used in the field of medical image segmentation. Its encoder extracts semantic features of different scales at different stages, but does not carry out special processing for semantic features of each scale. OBJECTIVE: To improve the feature expression ability and segmentation performance of U-Net, we proposed a feature supplement and optimization U-Net (FSOU-Net). METHODS: First, we put forward the view that semantic features of different scales should be treated differently. Based on this view, we classify the semantic features automatically extracted by encoders into two categories: shallow semantic features and deep semantic features. Then, we propose the shallow feature supplement module (SFSM), which obtains fine-grained semantic features through up-sampling to supplement the shallow semantic information. Finally, we propose the deep feature optimization module (DFOM), which uses the expansive convolution of different receptive fields to obtain multi-scale features and then performs multi-scale feature fusion to optimize the deep semantic information. RESULTS: The proposed model is experimented on three medical image segmentation public datasets, and the experimental results prove the correctness of the proposed idea. The segmentation performance of the model is higher than the advanced models for medical image segmentation. Compared with baseline network U-NET, the main index of Dice index is 0.75% higher on the RITE dataset, 2.3% higher on the Kvasir-SEG dataset, and 0.24% higher on the GlaS dataset. CONCLUSIONS: The proposed method can greatly improve the feature representation ability and segmentation performance of the model.
引用
收藏
页码:181 / 195
页数:15
相关论文
共 50 条
  • [1] DRA U-Net: An Attention based U-Net Framework for 2D Medical Image Segmentation
    Zhang, Xian
    Feng, Ziyuan
    Zhong, Tianchi
    Shen, Sicheng
    Zhang, Ruolin
    Zhou, Lijie
    Zhang, Bo
    Wang, Wendong
    2021 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2021, : 3936 - 3942
  • [2] Edge-Boosted U-Net for 2D Medical Image Segmentation
    Zhao, Renjie
    Chen, Weiting
    Cao, Guitao
    IEEE ACCESS, 2019, 7 : 171214 - 171222
  • [3] MSU-Net: Multi-Scale U-Net for 2D Medical Image Segmentation
    Su, Run
    Zhang, Deyun
    Liu, Jinhuai
    Cheng, Chuandong
    FRONTIERS IN GENETICS, 2021, 12
  • [4] Wavelet U-Net for Medical Image Segmentation
    Ying Li
    Yu Wang
    Tuo Leng
    Wen Zhijie
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2020, PT I, 2020, 12396 : 800 - 810
  • [5] GLUNet: Global-Local Fusion U-Net for 2D Medical Image Segmentation
    Wang, Ning
    Quan, Hongyan
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2021, PT IV, 2021, 12894 : 74 - 85
  • [6] MFA U-Net: a U-Net like multi-stage feature analysis network for medical image segmentation
    Wang, Yupeng
    Wang, Suyu
    He, Jian
    PATTERN ANALYSIS AND APPLICATIONS, 2024, 27 (04)
  • [7] Medical Image Segmentation Based on 3D U-net
    Chen, Silu
    Hu, Guanghao
    Sun, Jun
    2020 19TH INTERNATIONAL SYMPOSIUM ON DISTRIBUTED COMPUTING AND APPLICATIONS FOR BUSINESS ENGINEERING AND SCIENCE (DCABES 2020), 2020, : 130 - 133
  • [8] Medical Image Segmentation based on U-Net: A Review
    Du, Getao
    Cao, Xu
    Liang, Jimin
    Chen, Xueli
    Zhan, Yonghua
    JOURNAL OF IMAGING SCIENCE AND TECHNOLOGY, 2020, 64 (02)
  • [9] Modified U-Net for cytological medical image segmentation
    Benazzouz, Mourtada
    Benomar, Mohammed Lamine
    Moualek, Youcef
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2022, 32 (05) : 1761 - 1773
  • [10] MIXED TRANSFORMER U-NET FOR MEDICAL IMAGE SEGMENTATION
    Wang, Hongyi
    Xie, Shiao
    Lin, Lanfen
    Iwamoto, Yutaro
    Han, Xian-Hua
    Chen, Yen-Wei
    Tong, Ruofeng
    2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2022, : 2390 - 2394