Leveraging Graph Neural Networks in Transferring Multimodal Knowledge for Unimodal Segmentation

Tianyi Liu, Jiongshu Wang, Haochuan Jiang, Kaizhu Huang*

*Corresponding author for this work

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

Abstract

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.

Original languageEnglish
Title of host publicationISBI 2025 - 2025 IEEE 22nd International Symposium on Biomedical Imaging, Proceedings
PublisherIEEE Computer Society
ISBN (Electronic)9798331520526
DOIs
Publication statusPublished - 2025
Event22nd IEEE International Symposium on Biomedical Imaging, ISBI 2025 - Houston, United States
Duration: 14 Apr 202517 Apr 2025

Publication series

NameProceedings - International Symposium on Biomedical Imaging
ISSN (Print)1945-7928
ISSN (Electronic)1945-8452

Conference

Conference22nd IEEE International Symposium on Biomedical Imaging, ISBI 2025
Country/TerritoryUnited States
CityHouston
Period14/04/2517/04/25

Keywords

  • Brain Tumor Segmentation
  • Knowledge Distillation
  • Magnetic Resonance Imaging
  • Missing Modality

Fingerprint

Dive into the research topics of 'Leveraging Graph Neural Networks in Transferring Multimodal Knowledge for Unimodal Segmentation'. Together they form a unique fingerprint.

Cite this