Uncertainty-Aware Variational Information Pursuit for Interpretable Medical Image Analysis

In medical imaging, AI decision-support systems must balance accuracy and interpretability to build user trust and support effective clinical decision-making. Recently, Variational Information Pursuit (V-IP) and its variants have emerged as interpretable-by-design modeling techniques, aiming to explain AI decisions in terms of human-understandable, clinically relevant concepts. However, existing V-IP methods overlook instance-level uncertainties in query-answer generation, which can arise from model limitations (epistemic uncertainty) or variability in expert responses (aleatoric uncertainty).This paper introduces Uncertainty-Aware V-IP (UAV-IP), a novel framework that integrates uncertainty quantification into the V-IP process. We evaluate UAV-IP across four medical imaging datasets, PH2, Derm7pt, BrEaST, and SkinCon, demonstrating an average AUC improvement of approximately 3.2% while generating 20% more concise explanations compared to baseline V-IP, without sacrificing informativeness. These findings highlight the importance of uncertainty-aware reasoning in interpretable by design models for robust and reliable medical decision-making.

@article{nahiduzzaman2025_2506.16742,
  title={ Uncertainty-Aware Variational Information Pursuit for Interpretable Medical Image Analysis },
  author={ Md Nahiduzzaman and Ruwan Tennakoon and Steven Korevaar and Zongyuan Ge and Alireza Bab-Hadiashar },
  journal={arXiv preprint arXiv:2506.16742},
  year={ 2025 }
}