EFU Net: Edge Information Fused 3D Unet for Brain Tumor Segmentation

被引:0
作者
Wang, Yu [1 ]
Tian, Hengyi [1 ]
Liu, Minhua [1 ]
机构
[1] Beijing Technol & Business Univ, Sch Comp & Artificial Intelligence, Beijing 100048, Peoples R China
基金
北京市自然科学基金;
关键词
Deep learning; brain tumor segmentation; encoder decoder structure; edge attention mechanism; hybrid loss function; CLASSIFICATION; FEATURES;
D O I
10.13164/re.2024.0387
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Brain tumors refer to abnormal cell proliferation formed in brain tissue, which can cause neurological dysfunction and cognitive impairment, posing a serious threat to human health. Therefore, it becomes a very challenging work to full-automaticly segment brain tumors using computers because of the mutual infiltration and fuzzy boundary between the focus areas and the normal brain tissue. To address the above issues, a segmentation method which integrates edge features is proposed in this paper. The overall segmentation architecture follows the encoder decoder structure, extracting rich features from the encoder. The first two layers of features are input to the edge attention module, and to extract tumor edge features which are fully fused with the features of the decoder segment. At the same time, an adaptive weighted mixed loss function is introduced to train the network by adaptively adjusting the weights of different loss parts in the training process. Relevant experiments were carried out using the public brain tumor data set. The Dice mean values of the proposed segmentation model in the whole tumor area (WT), the core tumor area (TC), and the enhancing tumor area (ET) reach 91.10%, 87.16%, and 88.86%, respectively, and the mean values of Hausdorff distance are 3.92, 5.12, and 1.92 mm, respectively. The experimental results showed that the proposed method can significantly improve segmentation accuracy, especially the segmentation effect of the edge part.
引用
收藏
页码:387 / 396
页数:10
相关论文
共 26 条
[1]   HTTU-Net: Hybrid Two Track U-Net for Automatic Brain Tumor Segmentation [J].
Aboelenein, Nagwa M. ;
Piao Songhao ;
Koubaa, Anis ;
Noor, Alam ;
Afifi, Ahmed .
IEEE ACCESS, 2020, 8 :101406-101415
[2]  
Al Nasim Md Abdullah, 2022, 2022 25th International Conference on Computer and Information Technology (ICCIT), P1027, DOI 10.1109/ICCIT57492.2022.10054934
[3]   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
[4]  
BEERS A., 2017, P IEEE INT C COMP VI, P235, DOI [10.48550/arXiv.1709.02967, DOI 10.48550/ARXIV.1709.02967]
[5]   ResUNet-a: A deep learning framework for semantic segmentation of remotely sensed data [J].
Diakogiannis, Foivos, I ;
Waldner, Francois ;
Caccetta, Peter ;
Wu, Chen .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2020, 162 (162) :94-114
[6]   Automatic Brain Tumor Detection and Segmentation Using U-Net Based Fully Convolutional Networks [J].
Dong, Hao ;
Yang, Guang ;
Liu, Fangde ;
Mo, Yuanhan ;
Guo, Yike .
MEDICAL IMAGE UNDERSTANDING AND ANALYSIS (MIUA 2017), 2017, 723 :506-517
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/CVPR.2018.00745, 10.1109/TPAMI.2019.2913372]
[9]   nnU-Net for Brain Tumor Segmentation [J].
Isensee, Fabian ;
Jaeger, Paul F. ;
Full, Peter M. ;
Vollmuth, Philipp ;
Maier-Hein, Klaus H. .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES (BRAINLES 2020), PT II, 2021, 12659 :118-132
[10]   THE NEW WHO CLASSIFICATION OF BRAIN-TUMORS [J].
KLEIHUES, P ;
BURGER, PC ;
SCHEITHAUER, BW .
BRAIN PATHOLOGY, 1993, 3 (03) :255-268