CLCU-Net: Cross-level connecte d U-shape d network with selective feature aggregation attention module for brain tumor segmentation

被引:29
作者
Wang, Y. L. [1 ]
Zhao, Z. J. [1 ]
Hu, S. Y. [2 ]
Chang, F. L. [1 ]
机构
[1] Shandong Univ, Sch Control Sci & Engn, Jinan 250061, Peoples R China
[2] Shandong First Med Univ, Dept Gen Surg, Affiliated Hosp 1, Jinan 250012, Peoples R China
关键词
Deep learning; Brain tumor segmentation; Multi-scale feature connection; Segmented attention module; Selective feature aggregation; CONVOLUTIONAL NEURAL-NETWORK;
D O I
10.1016/j.cmpb.2021.106154
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Background and Objective: Brain tumors are among the most deadly cancers worldwide. Due to the development of deep convolutional neural networks, many brain tumor segmentation methods help clinicians diagnose and operate. However, most of these methods insufficiently use multi-scale features, reducing their ability to extract brain tumors' features and details. To assist clinicians in the accurate automatic segmentation of brain tumors, we built a new deep learning network to make full use of multi-scale features for improving the performance of brain tumor segmentation. Methods: We propose a novel cross-level connected U-shaped network (CLCU-Net) to connect different scales' features for fully utilizing multi-scale features. Besides, we propose a generic attention module (Segmented Attention Module, SAM) on the connections of different scale features for selectively aggregating features, which provides a more efficient connection of different scale features. Moreover, we employ deep supervision and spatial pyramid pooling (SSP) to improve the method's performance further. Results: We evaluated our method on the BRATS 2018 dataset by five indexes and achieved excellent performance with a Dice Score of 88.5%, a Precision of 91.98%, a Recall of 85.62%, a Params of 36.34M and Inference Time of 8.89ms for the whole tumor, which outperformed six state-of-the-art methods. Moreover, the performed analysis of different attention modules' heatmaps proved that the attention module proposed in this study was more suitable for segmentation tasks than the other existing popular attention modules. Conclusion: Both the qualitative and quantitative experimental results indicate that our cross-level connected U-shaped network with selective feature aggregation attention module can achieve accurate brain tumor segmentation and is considered quite instrumental in clinical practice implementation. (c) 2021 Elsevier B.V. All rights reserved.
引用
收藏
页数:9
相关论文
共 46 条
[1]  
[Anonymous], 2018, ARXIV181102629
[2]   Data Descriptor: Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features [J].
Bakas, Spyridon ;
Akbari, Hamed ;
Sotiras, Aristeidis ;
Bilello, Michel ;
Rozycki, Martin ;
Kirby, Justin S. ;
Freymann, John B. ;
Farahani, Keyvan ;
Davatzikos, Christos .
SCIENTIFIC DATA, 2017, 4
[3]  
Bakas S., 2017, Cancer Imaging Arch., V286, DOI DOI 10.7937/K9/TCIA.2017.KLXWJJ1Q
[4]   Data Descriptor: Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features [J].
Bakas, Spyridon ;
Akbari, Hamed ;
Sotiras, Aristeidis ;
Bilello, Michel ;
Rozycki, Martin ;
Kirby, Justin S. ;
Freymann, John B. ;
Farahani, Keyvan ;
Davatzikos, Christos .
SCIENTIFIC DATA, 2017, 4
[5]  
Bauer S, 2011, LECT NOTES COMPUT SC, V6893, P354, DOI 10.1007/978-3-642-23626-6_44
[6]   Fully Automatic Brain Tumor Segmentation using End-To-End Incremental Deep Neural Networks in MRI images [J].
Ben Naceur, Mostefa ;
Saouli, Rachida ;
Akil, Mohamed ;
Kachouri, Rostom .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2018, 166 :39-49
[7]   VoxResNet: Deep voxelwise residual networks for brain segmentation from 3D MR images [J].
Chen, Hao ;
Dou, Qi ;
Yu, Lequan ;
Qin, Jing ;
Heng, Pheng-Ann .
NEUROIMAGE, 2018, 170 :446-455
[8]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[9]  
Cicek Ozgun, 2016, Medical Image Computing and Computer-Assisted Intervention - MICCAI 2016. 19th International Conference. Proceedings: LNCS 9901, P424, DOI 10.1007/978-3-319-46723-8_49
[10]   Dense deformation field estimation for atlas-based segmentation of pathological MR brain images [J].
Cuadra, M. Bach ;
De Craene, M. ;
Duay, V. ;
Macq, B. ;
Pollo, C. ;
Thiran, J. -Ph. .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2006, 84 (2-3) :66-75