Semantic-guided Masked Mutual Learning for Multi-modal Brain Tumor Segmentation with Arbitrary Missing Modalities
Malignant brain tumors have become an aggressive and dangerous disease that leads to death worldwide. Multi-modal MRI data is crucial for accurate brain tumor segmentation, but missing modalities common in clinical practice can severely degrade the segmentation performance. While incomplete multi-modal learning methods attempt to address this, learning robust and discriminative features from arbitrary missing modalities remains challenging. To address this challenge, we propose a novel Semantic-guided Masked Mutual Learning (SMML) approach to distill robust and discriminative knowledge across diverse missing modality scenarios. Specifically, we propose a novel dual-branch masked mutual learning scheme guided by Hierarchical Consistency Constraints (HCC) to ensure multi-level consistency, thereby enhancing mutual learning in incomplete multi-modal scenarios. The HCC framework comprises a pixel-level constraint that selects and exchanges reliable knowledge to guide the mutual learning process. Additionally, it includes a feature-level constraint that uncovers robust inter-sample and inter-class relational knowledge within the latent feature space. To further enhance multi-modal learning from missing modality data, we integrate a refinement network into each student branch. This network leverages semantic priors from the Segment Anything Model (SAM) to provide supplementary information, effectively complementing the masked mutual learning strategy in capturing auxiliary discriminative knowledge. Extensive experiments on three challenging brain tumor segmentation datasets demonstrate that our method significantly improves performance over state-of-the-art methods in diverse missing modality settings.
Authors
- Jingyuan Chen (ORCID: https://orcid.org/0000-0003-0415-6937)
- Sai Wu (ORCID: https://orcid.org/0000-0002-7903-1496)
- Qin Zhou (ORCID: https://orcid.org/0000-0001-5334-7188)
- Guoyan Liang (ORCID: https://orcid.org/0000-0003-0072-3156)
- Chang Yao (ORCID: https://orcid.org/0000-0002-1187-6257)
- Yi Xu (ORCID: https://orcid.org/0000-0003-1679-3271)
- Kai Chen (ORCID: https://orcid.org/0000-0003-2986-8148)
- Bingcang Huang
- Zhe Wang (ORCID: https://orcid.org/0009-0002-3397-0938)
Institutions
- Zhejiang Sci-Tech University (CN)
- East China University of Science and Technology (CN)
- Second Military Medical University (CN)
- RIKEN Center for Advanced Intelligence Project (JP)
- Shanghai Pudong New Area Gongli Hospital (CN)
- Zhejiang University (CN)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i5.32545
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Science Foundation
- National Natural Science Foundation of China