Brain tumor image segmentation based on prior knowledge via transformer

被引:3
作者
Li, Qiang [1 ]
Liu, Hengxin [1 ]
Nie, Weizhi [2 ]
Wu, Ting [3 ]
机构
[1] Tianjin Univ, Sch Microelect, Tianjin, Peoples R China
[2] Tianjin Univ, Sch Elect & Informat Engn, Tianjin 300072, Peoples R China
[3] Tianjin Univ, Chest Hosp, Dept Cardiac Surg, Tianjin, Peoples R China
基金
中国国家自然科学基金;
关键词
attention mechanism; brain tumor segmentation; prior knowledge; transformer; CONVOLUTIONAL NEURAL-NETWORKS;
D O I
10.1002/ima.22931
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Many researchers use AI to improve the accuracy of early diagnostic techniques. However, as a result of the tumor's uneven shape, fuzzy borders and too few data, existing tumor segmentation methods do not propose accurate segmentation results. We innovative introduces the prior knowledge learned to filter the noise information and guide the final network to generate a more accurate segmentation model. First, we introduce a classification network with an attention block to highlight the potential location of the brain tumor and also obtain the rough diagnosis result as the prior knowledge. Second, we provide a novel image fusion network consisting of a transformer with cross attention to merge tumor localization information with brain MRI images. Third, we propose a novel multilayer transformer experience information fusion network to combine the classic U-Net network to handle the guiding of prior knowledge. The higher performance of the suggested method is demonstrated by comparison with contemporary methods.
引用
收藏
页码:2073 / 2087
页数:15
相关论文
共 53 条
[1]  
[Anonymous], 2019, Identifying the best machine learning algorithms for brain tumor segmentation, progression assessment, and overall survival prediction in the BRATS challenge
[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]   A survey of MRI-based medical image analysis for brain tumor studies [J].
Bauer, Stefan ;
Wiest, Roland ;
Nolte, Lutz-P ;
Reyes, Mauricio .
PHYSICS IN MEDICINE AND BIOLOGY, 2013, 58 (13) :R97-R129
[4]   End-to-End Object Detection with Transformers [J].
Carion, Nicolas ;
Massa, Francisco ;
Synnaeve, Gabriel ;
Usunier, Nicolas ;
Kirillov, Alexander ;
Zagoruyko, Sergey .
COMPUTER VISION - ECCV 2020, PT I, 2020, 12346 :213-229
[5]  
Chen B., 2021, ARXIV210705274
[6]   3D Dilated Multi-fiber Network for Real-Time Brain Tumor Segmentation in MRI [J].
Chen, Chen ;
Liu, Xiaopeng ;
Ding, Meng ;
Zheng, Junfeng ;
Li, Jiangyun .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT III, 2019, 11766 :184-192
[7]  
Chen J. N., ARXIV210204306
[8]   Dual-force convolutional neural networks for accurate brain tumor segmentation [J].
Chen, Shengcong ;
Ding, Changxing ;
Liu, Minfeng .
PATTERN RECOGNITION, 2019, 88 :90-100
[9]  
Cicek O., 2016, P INT C MED IM COMP
[10]  
Dosovitskiy A., 2021, arXiv