Search papers, labs, and topics across Lattice.
This paper introduces a probabilistic representation framework for robust brain tumor segmentation using multimodal MRI, addressing the challenge of incomplete modality acquisition during inference. By modeling representations as Gaussian distributions, the mean captures task information while the variance quantifies uncertainty due to missing evidence. The proposed method outperforms existing baselines in various missing-modality scenarios, highlighting the importance of accurately modeling uncertainty in medical imaging tasks.
Uncertainty modeling in brain tumor segmentation can significantly enhance reliability, even when critical MRI modalities are missing.
Multimodal MRI is essential for accurate brain tumor segmentation. However, acquiring all modalities at inference is often challenging in practice, which causes intrinsic uncertainty due to unavoidable information loss. Without modeling this uncertainty, existing methods encode incomplete evidence into deterministic representations that appear plausible but lack reliability. In this regime, we propose a probabilistic representation framework that models representations as Gaussian distributions, where their mean captures task information and their variance measures uncertainty from missing evidence. To make variance reflect information deficiency, we regularize the mean from each partial configuration toward its full-modality counterpart, while scaling the variance with the discrepancy between their aligned means. We further introduce a set-inclusive strategy that exploits the hierarchical structure of modality subsets and enforces an ordering constraint to maintain their consistent uncertainty relationships. Extensive experiments on BraTS 2018 and 2020 demonstrate that our approach offers superior performance over baselines across diverse missing-modality scenarios. Code and model checkpoint are available at https://github.com/atlas-sky/SIUM.
