IBA-U-Net: Attentive BConvLSTM U-Net with Redesigned Inception for medical image segmentation

被引:22
作者
Chen, Siyuan [1 ]
Zou, Yanni [1 ]
Liu, Peter X. [2 ]
机构
[1] Nanchang Univ, Sch Informat Engn, Nanchang 330031, Jiangxi, Peoples R China
[2] Carleton Univ, Dept Syst & Comp Engn, Ottawa, ON K1S 5B6, Canada
基金
中国国家自然科学基金;
关键词
Medical image segmentation; Deep convolutional neural networks; Multi-scale feature fusion; Attentive BConvLSTM; VESSEL SEGMENTATION; AUTOMATIC DETECTION; BLOOD-VESSELS; OPTIC DISC; NETWORK; MODEL;
D O I
10.1016/j.compbiomed.2021.104551
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
Accurate segmentation of medical images plays an essential role in their analysis and has a wide range of research and application values in fields of practice such as medical research, disease diagnosis, disease analysis, and auxiliary surgery. In recent years, deep convolutional neural networks have been developed that show strong performance in medical image segmentation. However, because of the inherent challenges of medical images, such as irregularities of the dataset and the existence of outliers, segmentation approaches have not demonstrated sufficiently accurate and reliable results for clinical employment. Our method is based on three key ideas: (1) integrating the BConvLSTM block and the Attention block to reduce the semantic gap between the encoder and decoder feature maps to make the two feature maps more homogeneous, (2) factorizing convolutions with a large filter size by Redesigned Inception, which uses a multiscale feature fusion method to significantly increase the effective receptive field, and (3) devising a deep convolutional neural network with multiscale feature fusion and a Attentive BConvLSTM mechanism, which integrates the Attentive BConvLSTM block and the Redesigned Inception block into an encoder-decoder model called Attentive BConvLSTM U-Net with Redesigned Inception (IBA-U-Net). Our proposed architecture, IBA-U-Net, has been compared with the U-Net and state-of-the-art segmentation methods on three publicly available datasets, the lung image segmentation dataset, skin lesion image dataset, and retinal blood vessel image segmentation dataset, each with their unique challenges, and it has improved the prediction performance even with slightly less calculation expense and fewer network parameters. By devising a deep convolutional neural network with a multiscale feature fusion and Attentive BConvLSTM mechanism, medical image segmentation of different tasks can be completed effectively and accurately with only 45% of U-Net parameters.
引用
收藏
页数:10
相关论文
共 58 条
[1]  
Abadi M, 2016, ACM SIGPLAN NOTICES, V51, P1, DOI [10.1145/2951913.2976746, 10.1145/3022670.2976746]
[2]  
Alom M.Z., 2018, CoRR
[3]   Bi-Directional ConvLSTM U-Net with Densley Connected Convolutions [J].
Azad, Reza ;
Asadi-Aghbolaghi, Maryam ;
Fathy, Mahmood ;
Escalera, Sergio .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, :406-415
[4]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[5]  
Becker C, 2013, LECT NOTES COMPUT SC, V8149, P526, DOI 10.1007/978-3-642-40811-3_66
[6]   Automatic detection and segmentation of brain metastases on multimodal MR images with a deep convolutional neural network [J].
Charron, Odelin ;
Lallement, Alex ;
Jarnet, Delphine ;
Noblet, Vincent ;
Clavier, Jean-Baptiste ;
Meyer, Philippe .
COMPUTERS IN BIOLOGY AND MEDICINE, 2018, 95 :43-54
[7]   DETECTION OF BLOOD-VESSELS IN RETINAL IMAGES USING TWO-DIMENSIONAL MATCHED-FILTERS [J].
CHAUDHURI, S ;
CHATTERJEE, S ;
KATZ, N ;
NELSON, M ;
GOLDBAUM, M .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 1989, 8 (03) :263-269
[8]  
Chen L.C., 2014, arXiv, V4, P357
[9]   Automated ventricular systems segmentation in brain CT images by combining low-level segmentation and high-level template matching [J].
Chen, Wenan ;
Smith, Rebecca ;
Ji, Soo-Yeon ;
Ward, Kevin R. ;
Najarian, Kayvan .
BMC MEDICAL INFORMATICS AND DECISION MAKING, 2009, 9
[10]  
Codella N., 2019, arXiv