Brain tumor segmentation based on the fusion of deep semantics and edge information in multimodal MRI

被引:337
作者
Zhu, Zhiqin [1 ]
He, Xianyu [1 ]
Qi, Guanqiu [2 ]
Li, Yuanyuan [1 ]
Cong, Baisen [3 ]
Liu, Yu [4 ]
机构
[1] Chongqing Univ Posts & Telecommun, Coll Automation, Chongqing 400065, Peoples R China
[2] State Univ New York Buffalo State, Comp Informat Syst Dept, Buffalo, NY 14222 USA
[3] DH Shanghai Diagnost Co Ltd, Diagnost Digital, Shanghai 200335, Peoples R China
[4] Hefei Univ Technol, Dept Biomed Engn, Hefei 230009, Peoples R China
基金
中国国家自然科学基金;
关键词
Brain tumor segmentation; Transformer; Convolutional neural networks; Edge feature; Feature fusion; FEATURES; NETWORK; UNET; CRF;
D O I
10.1016/j.inffus.2022.10.022
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Brain tumor segmentation in multimodal MRI has great significance in clinical diagnosis and treatment. The utilization of multimodal information plays a crucial role in brain tumor segmentation. However, most existing methods focus on the extraction and selection of deep semantic features, while ignoring some features with specific meaning and importance to the segmentation problem. In this paper, we propose a brain tumor segmentation method based on the fusion of deep semantics and edge information in multimodal MRI, aiming to achieve a more sufficient utilization of multimodal information for accurate segmentation. The proposed method mainly consists of a semantic segmentation module, an edge detection module and a feature fusion module. In the semantic segmentation module, the Swin Transformer is adopted to extract semantic features and a shifted patch tokenization strategy is introduced for better training. The edge detection module is designed based on convolutional neural networks (CNNs) and an edge spatial attention block (ESAB) is presented for feature enhancement. The feature fusion module aims to fuse the extracted semantic and edge features, and we design a multi-feature inference block (MFIB) based on graph convolution to perform feature reasoning and information dissemination for effective feature fusion. The proposed method is validated on the popular BraTS benchmarks. The experimental results verify that the proposed method outperforms a number of state-of-the-art brain tumor segmentation methods. The source code of the proposed method is available at https://github.com/HXY-99/brats.
引用
收藏
页码:376 / 387
页数:12
相关论文
共 52 条
[1]   Devil is in the Edges: Learning Semantic Boundaries from Noisy Annotations [J].
Acuna, David ;
Kar, Amlan ;
Fidler, Sanja .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :11067-11075
[2]  
Bakas S, 2019, Arxiv, DOI arXiv:1811.02629
[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]   Explainable Artificial Intelligence (XAI): Concepts, taxonomies, opportunities and challenges toward responsible AI [J].
Barredo Arrieta, Alejandro ;
Diaz-Rodriguez, Natalia ;
Del Ser, Javier ;
Bennetot, Adrien ;
Tabik, Siham ;
Barbado, Alberto ;
Garcia, Salvador ;
Gil-Lopez, Sergio ;
Molina, Daniel ;
Benjamins, Richard ;
Chatila, Raja ;
Herrera, Francisco .
INFORMATION FUSION, 2020, 58 :82-115
[5]  
Chen J., 2021, arXiv, DOI DOI 10.48550/ARXIV.2102.04306
[6]   Image Manipulation Detection by Multi-View Multi-Scale Supervision [J].
Chen, Xinru ;
Dong, Chengbo ;
Ji, Jiaqi ;
Cao, Juan ;
Li, Xirong .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :14165-14173
[7]   Graph-Based Global Reasoning Networks [J].
Chen, Yunpeng ;
Rohrbach, Marcus ;
Yan, Zhicheng ;
Yan, Shuicheng ;
Feng, Jiashi ;
Kalantidis, Yannis .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :433-442
[8]   Boundary-Preserving Mask R-CNN [J].
Cheng, Tianheng ;
Wang, Xinggang ;
Huang, Lichao ;
Liu, Wenyu .
COMPUTER VISION - ECCV 2020, PT XIV, 2020, 12359 :660-676
[9]  
Ding Y., 2021, P IEEECVF INT C COMP, P3975
[10]   HyperDense-Net: A Hyper-Densely Connected CNN for Multi-Modal Image Segmentation [J].
Dolz, Jose ;
Gopinath, Karthik ;
Yuan, Jing ;
Lombaert, Herve ;
Desrosiers, Christian ;
Ben Ayed, Ismail .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (05) :1116-1126