DFBU-Net: Double-branch flat bottom U-Net for efficient medical image segmentation

被引:5
作者
Yin, Hao [1 ]
Wang, Yi [1 ]
Wen, Jing [1 ]
Wang, Guangxian [2 ]
Lin, Bo [3 ]
Yang, Weibin [3 ]
Ruan, Jian [3 ]
Zhang, Yi [3 ]
机构
[1] Chongqing Univ, Sch Comp Sci, Chongqing 400000, Peoples R China
[2] Chongqing Med Univ, Banan Hosp, Dept Radiol, Chongqing 400000, Peoples R China
[3] Chongqing Univ, Intelligent Oncol Res Ctr, Canc Hosp, Chongqing 400000, Peoples R China
基金
中国国家自然科学基金;
关键词
U-Net; Medical image segmentation; Medical characteristic; Double-branch; NETWORK; AGGREGATION;
D O I
10.1016/j.bspc.2023.105818
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
In the field of medical image processing, segmenting tissues and organs in CT/MRI and other medical sequence images is a vital yet challenging task. Analyzing the MICCAI competition, we have identified two problems in current methods for medical image organ segmentation: (1) There is a bottleneck in organ segmentation, with marginal room for improvement, as algorithmic capabilities have already surpassed the task's inherent difficulty. (2) Most current research focuses on stacking and enhancing new modules for segmentation while overlooking the inherent characteristics of medical sequence images. To overcome these two problems, firstly, we have encapsulated the three characteristics of CT/MRI medical sequence image segmentation: semantic correctness, edge accuracy, and 3D structure. Secondly, we delved into the most information-rich downsampling stage in terms of detail and semantics. Subsequently, we designed a flat-bottom double-branch network (DFBU-Net) based on the U-Net architecture. The high-resolution flat bottom branch of this network maintained a 1/4 feature map size to ensure the preservation of rich detail information, while the low-resolution branch underwent progressive downsampling to capture more semantic information. To prevent information loss, cross-fusion was performed at each stage of the model's two branches. Finally, DFBU-Net was evaluated on the MICCAI FLARE2021 dataset (DSC:93.61%, NSD:85.01%). Particularly, in the challenging task of pancreatic segmentation, our model outperformed the first-place model by 0.72% in DSC and 2.92% in NSD. Furthermore, in the MICCAI PARSE2022 competition, DFBU-Net ranked ninth with a DICE score of 79.28%, demonstrating its excellent segmentation performance and generalization ability.
引用
收藏
页数:11
相关论文
共 44 条
  • [1] Antonelli M, 2021, Arxiv, DOI arXiv:2106.05735
  • [2] Cao H., 2021, arXiv, DOI DOI 10.48550/ARXIV.2105.05537
  • [3] AssemblyNet: A Novel Deep Decision-Making Process for Whole Brain MRI Segmentation
    Coupe, Pierrick
    Mansencal, Boris
    Clement, Michael
    Giraud, Remi
    de Senneville, Baudouin Denis
    Vinh-Thong Ta
    Lepetit, Vincent
    Manjon, Jose V.
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT III, 2019, 11766 : 466 - 474
  • [4] DeU-Net 2.0: Enhanced deformable U-Net for 3D cardiac cine MRI segmentation
    Dong, Shunjie
    Pan, Zixuan
    Fu, Yu
    Yang, Qianqian
    Gao, Yuanxue
    Yu, Tianbai
    Shi, Yiyu
    Zhuo, Cheng
    [J]. MEDICAL IMAGE ANALYSIS, 2022, 78
  • [5] Optimized U-Net for Brain Tumor Segmentation
    Futrega, Michal
    Milesi, Alexandre
    Marcinkiewicz, Michal
    Ribalta, Pablo
    [J]. BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2021, PT II, 2022, 12963 : 15 - 29
  • [6] CE-Net: Context Encoder Network for 2D Medical Image Segmentation
    Gu, Zaiwang
    Cheng, Jun
    Fu, Huazhu
    Zhou, Kang
    Hao, Huaying
    Zhao, Yitian
    Zhang, Tianyang
    Gao, Shenghua
    Liu, Jiang
    [J]. IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (10) : 2281 - 2292
  • [7] SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentation
    Guo, Changlu
    Szemenyei, Marton
    Yi, Yugen
    Wang, Wenle
    Chen, Buer
    Fan, Changqi
    [J]. 2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 1236 - 1242
  • [8] UNETR: Transformers for 3D Medical Image Segmentation
    Hatamizadeh, Ali
    Tang, Yucheng
    Nath, Vishwesh
    Yang, Dong
    Myronenko, Andriy
    Landman, Bennett
    Roth, Holger R.
    Xu, Daguang
    [J]. 2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, : 1748 - 1758
  • [9] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [10] Huang HM, 2020, INT CONF ACOUST SPEE, P1055, DOI [10.1109/icassp40776.2020.9053405, 10.1109/ICASSP40776.2020.9053405]