Uncertainty-Aware Vision-Language Segmentation for Medical Imaging
Organizations: VIT Bhopal · SAHE, Andhra Pradesh · IIIT Delhi · Tezpur University Assam · IIT Kanpur
Abstract
We introduce a novel uncertainty-aware multimodal segmentation framework that leverages both radiological images and associated clinical text for precise medical diagnosis. We propose a Modality Decoding Attention Block (MoDAB) with a lightweight State Space Mixer (SSMix) to enable efficient cross-modal fusion and long-range dependency modelling. To guide learning under ambiguity, we propose the Spectral-Entropic Uncertainty (SEU) Loss, which jointly captures spatial overlap, spectral consistency, and predictive uncertainty in a unified objective. In complex clinical circumstances with poor image quality, this formulation improves model reliability. Extensive experiments on various publicly available medical datasets, QATA-COVID19, MosMed++, and Kvasir-SEG, demonstrate that our method achieves superior segmentation performance while being significantly more computationally efficient than existing State-of-the-Art (SoTA) approaches. Our results highlight the importance of incorporating uncertainty modelling and structured modality alignment in vision-language medical segmentation tasks. Code: https://github.com/arya-domain/UA-VLS
Figures & tables
| Modality | Method | Trainable Params (M) | Flops (G) | QATA-COV19 | MosMedData++ | Kvasir-Seg | |||
| Dice (%) | mIoU (%) | Dice (%) | mIoU (%) | Dice (%) | mIoU (%) | ||||
| MonoModels | U-Net [ 1 ] | 14.8 | 50.3 | 78.91 | 69.32 | 64.30 | 50.50 | 82.33 | 74.26 |
| UNet++ [ 3 ] | 74.5 | 94.6 | 79.47 | 70.05 | 71.63 | 58.14 | 82.79 | 73.94 | |
| AttUNet [ 2 ] | 34.9 | 101.9 | 79.11 | 69.83 | 66.07 | 52.68 | 82.94 | 74.17 | |
| nnUNet [ 9 ] | 19.1 | 412.7 | 80.30 | 70.62 | 72.32 | 60.14 | 84.22 | 75.41 | |
| TransUNet [ 10 ] | 105 | 56.7 | 78.44 | 68.84 | 71.13 | 58.28 | 90.53 | 85.94 | |
| Method | Dice (%) | mIoU (%) |
| Loss Function | ||
| Dice loss | 93.44 | 85.76 |
| BCE loss | 92.03 | 85.26 |
| Textual Guidance | ||
| Inference w/o Text Prompts | 87.28 | 81.32 |
| Training w/o MoDAB | 85.15 | 73.86 |