Multi-teacher cross-modal distillation with cooperative deep supervision fusion learning for unimodal segmentation

被引:8
作者
Ahmad, Saeed [1 ]
Ullah, Zahid [1 ]
Gwak, Jeonghwan [1 ,2 ,3 ,4 ]
机构
[1] Korea Natl Univ Transportat, Dept Software, Chungju 27469, South Korea
[2] Korea Natl Univ Transportat, Dept IT Energy Convergence BK21 FOUR, Chungju 27469, South Korea
[3] Korea Natl Univ Transportat, Dept Biomed Engn, Chungju 27469, South Korea
[4] Korea Natl Univ Transportat, Dept AI Robot Engn, Chungju 27469, South Korea
基金
新加坡国家研究基金会;
关键词
Brain tumor segmentation; Knowledge distillation; Cooperative learning; Feature fusion; Multi-teacher framework; SEMANTICS;
D O I
10.1016/j.knosys.2024.111854
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Accurate brain tumor segmentation is a labor-intensive and time-consuming task that requires automation to enhance its efficacy. Recent advanced techniques have shown promising results in segmenting brain tumors; however, their dependency on extensive multimodal magnetic resonance imaging (MRI) data limits their practicality in clinical environments where such data may not be readily available. To address this, we propose a novel multi-teacher cross-modal knowledge distillation framework, which utilizes the privileged multimodal data during training while relying solely on unimodal data for inference. Our framework is tailored to the unimodal segmentation of the T 1ce MRI sequence, which is prevalently available in clinical practice and structurally akin to the T1 1 modality, providing ample information for the segmentation task. Our framework introduces two learning strategies for knowledge distillation (KD): (1) performance-aware response-based KD and (2) cooperative deep supervision fusion learning (CDSFL). The first strategy involves dynamically assigning confidence weights to each teacher model based on its performance, ensuring that the KD is performance- driven, and the CDSFL module augments the learning capabilities of the multi-teacher models by fostering mutual learning. Moreover, the fused information is distilled into the student model to improve its deep supervision. Extensive experiments on BraTS datasets demonstrate that our framework achieves promising unimodal segmentation results on the T 1ce and T1 1 modalities and outperforms previous state-of-the-art methods. Code is available at https://github.com/ami-lab-knut/mtcm_kd.
引用
收藏
页数:13
相关论文
共 49 条
[41]   Unified Attentional Generative Adversarial Network for Brain Tumor Segmentation from Multimodal Unpaired Images [J].
Yuan, Wenguang ;
Wei, Jia ;
Wang, Jiabing ;
Ma, Qianli ;
Tasdizen, Tolga .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT III, 2019, 11766 :229-237
[42]   mustGAN: multi-stream Generative Adversarial Networks for MR Image Synthesis [J].
Yurt, Mahmut ;
Dar, Salman U. H. ;
Erdem, Aykut ;
Erdem, Erkut ;
Oguz, Kader K. ;
Cukur, Tolga .
MEDICAL IMAGE ANALYSIS, 2021, 70
[43]   Multi-Modal MRI Image Synthesis via GAN With Multi-Scale Gate Mergence [J].
Zhan, Bo ;
Li, Di ;
Wu, Xi ;
Zhou, Jiliu ;
Wang, Yan .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2022, 26 (01) :17-26
[44]   Cross-modality deep feature learning for brain tumor segmentation [J].
Zhang, Dingwen ;
Huang, Guohai ;
Zhang, Qiang ;
Han, Jungong ;
Han, Junwei ;
Yu, Yizhou .
PATTERN RECOGNITION, 2021, 110
[45]   Ensemble diverse hypotheses and knowledge distillation for unsupervised cross-subject adaptation [J].
Zhang, Kuangen ;
Chen, Jiahong ;
Wang, Jing ;
Chen, Xinxing ;
Leng, Yuquan ;
de Silva, Clarence W. ;
Fu, Chenglong .
INFORMATION FUSION, 2023, 93 :268-281
[46]   Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillation [J].
Zhang, Linfeng ;
Song, Jiebo ;
Gao, Anni ;
Chen, Jingwei ;
Bao, Chenglong ;
Ma, Kaisheng .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3712-3721
[47]   Hi-Net: Hybrid-Fusion Network for Multi-Modal MR Image Synthesis [J].
Zhou, Tao ;
Fu, Huazhu ;
Chen, Geng ;
Shen, Jianbing ;
Shao, Ling .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (09) :2772-2781
[48]   Feature fusion and latent feature learning guided brain tumor segmentation and missing modality recovery network [J].
Zhou, Tongxue .
PATTERN RECOGNITION, 2023, 141
[49]   Brain tumor segmentation based on the fusion of deep semantics and edge information in multimodal MRI [J].
Zhu, Zhiqin ;
He, Xianyu ;
Qi, Guanqiu ;
Li, Yuanyuan ;
Cong, Baisen ;
Liu, Yu .
INFORMATION FUSION, 2023, 91 :376-387