MedProbCLIP: Probabilistic Adaptation of Vision-Language Foundation Model for Reliable Radiograph-Report Retrieval
Organizations: Texas A&M University-San Antonio · Boise State University
Abstract
Vision-language foundation models have emerged as powerful general-purpose representation learners with strong potential for multimodal understanding, but their deterministic embeddings often fail to provide the reliability required for high-stakes biomedical applications. This work introduces MedProbCLIP, a probabilistic vision-language learning framework for chest X-ray and radiology report representation learning and bidirectional retrieval. MedProbCLIP models image and text representations as Gaussian embeddings through a probabilistic contrastive objective that explicitly captures uncertainty and many-to-many correspondences between radiographs and clinical narratives. A variational information bottleneck mitigates overconfident predictions, while MedProbCLIP employs multi-view radiograph encoding and multi-section report encoding during training to provide fine-grained supervision for clinically aligned correspondence, yet requires only a single radiograph and a single report at inference. Evaluated on the MIMIC-CXR dataset, MedProbCLIP outperforms deterministic and probabilistic baselines, including CLIP, CXR-CLIP, and PCME++, in both retrieval and zero-shot classification. Beyond accuracy, MedProbCLIP demonstrates superior calibration, risk-coverage behavior, selective retrieval reliability, and robustness to clinically relevant corruptions, underscoring the value of probabilistic vision-language modeling for improving the trustworthiness and safety of radiology image-text retrieval systems.
Figures & tables
| Method | i2t | t2i | RSUM | ||||||
| R@1 | R@5 | R@10 | R@100 | R@1 | R@5 | R@10 | R@100 | ||
| Random | |||||||||
| CLIP | |||||||||
| PCME++ | |||||||||
| CXR-CLIP | |||||||||
| MedProbCLIP | 21.02 | 46.88 | 58.91 | 92.41 | 19.96 | 47.44 | 59.42 | 92.58 | 438.62 |
| Method | EC | CA | LL | AO | ED | CO | PA | AT | PX | PE | PO | FR | SD | Mean |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| CLIP | 0.8453 | 0.6364 | 0.7925 | |||||||||||
| PCME++ | 0.8061 | 0.7150 | ||||||||||||
| CXR-CLIP | 0.7837 | 0.8167 | ||||||||||||
| MedProbCLIP | 0.7976 | 0.7292 | 0.7308 | 0.7802 | 0.8661 | 0.7273 | 0.7101 |
Radiology VLMs
Radiology vision-language models (VLMs) align medical images and text to support grounded diagnosis, reporting, retrieval, and clinical reasoning.
See all 109 papers