TY - GEN
T1 - Leveraging Graph Neural Networks in Transferring Multimodal Knowledge for Unimodal Segmentation
AU - Liu, Tianyi
AU - Wang, Jiongshu
AU - Jiang, Haochuan
AU - Huang, Kaizhu
N1 - Publisher Copyright:
© 2025 IEEE.
PY - 2025
Y1 - 2025
N2 - Accurate segmentation of brain tumors in MRI scans is crucial for effective treatment. Multimodal MRI, including FLAIR, Tlce, T1, and T2 images, provides valuable infor-mation for tumor delineation. However, challenges such data corruption and varying protocols can lead to missing modal-ities, affecting segmentation accuracy. Recent methods at-tempt to transfer knowledge from multimodal teacher mod-els to unimodal student models, but often struggle to capture complex structural relationships within the data. This paper introduces a novel approach called Graph Neural Network-based Knowledge Distillation (GKD), which uses Graph Neu-ral Networks to create a comprehensive graph connecting in-termediate layers of both teacher and student networks. This enables the student network to better learn essential features and their interactions. Evaluations on the BraTS2018 and BraTS2020 datasets demonstrate that GKD significantly en-hances brain tumor segmentation accuracy by deepening the connection between teacher and student insights. The code is publicly available at https://github.com/T-Y-Liu/GKD.
AB - Accurate segmentation of brain tumors in MRI scans is crucial for effective treatment. Multimodal MRI, including FLAIR, Tlce, T1, and T2 images, provides valuable infor-mation for tumor delineation. However, challenges such data corruption and varying protocols can lead to missing modal-ities, affecting segmentation accuracy. Recent methods at-tempt to transfer knowledge from multimodal teacher mod-els to unimodal student models, but often struggle to capture complex structural relationships within the data. This paper introduces a novel approach called Graph Neural Network-based Knowledge Distillation (GKD), which uses Graph Neu-ral Networks to create a comprehensive graph connecting in-termediate layers of both teacher and student networks. This enables the student network to better learn essential features and their interactions. Evaluations on the BraTS2018 and BraTS2020 datasets demonstrate that GKD significantly en-hances brain tumor segmentation accuracy by deepening the connection between teacher and student insights. The code is publicly available at https://github.com/T-Y-Liu/GKD.
KW - Brain Tumor Segmentation
KW - Knowledge Distillation
KW - Magnetic Resonance Imaging
KW - Missing Modality
UR - http://www.scopus.com/inward/record.url?scp=105005834826&partnerID=8YFLogxK
U2 - 10.1109/ISBI60581.2025.10981298
DO - 10.1109/ISBI60581.2025.10981298
M3 - Conference Proceeding
AN - SCOPUS:105005834826
T3 - Proceedings - International Symposium on Biomedical Imaging
BT - ISBI 2025 - 2025 IEEE 22nd International Symposium on Biomedical Imaging, Proceedings
PB - IEEE Computer Society
T2 - 22nd IEEE International Symposium on Biomedical Imaging, ISBI 2025
Y2 - 14 April 2025 through 17 April 2025
ER -