LATA: Laplacian-Assisted Transductive Adaptation for Conformal Uncertainty in Medical VLMs
Behzad Bozorgtabar, Dwarikanath Mahapatra, Sudipta Roy, Muzammal Naseer, Imran Razzak, Zongyuan Ge
Abstract
Medical vision-language models (VLMs) are strong zero-shot recognizers for medical imaging, but their reliability under domain shift hinges on calibrated uncertainty with guarantees. Split conformal prediction (SCP) offers finite-sample coverage, yet prediction sets often become large (low efficiency) and class-wise coverage unbalanced-high class-conditioned coverage gap (CCV), especially in few-shot, imbalanced regimes; moreover, naively adapting to calibration labels breaks exchangeability and voids guarantees. We propose **LATA** (Laplacian-Assisted Transductive Adaptation), a training- and label-free refinement that operates on the joint calibration and test pool by smoothing zero-shot probabilities over an image-image k-NN graph using a small number of CCCP mean-field updates, preserving SCP validity via a deterministic transform. We further introduce a failure-aware conformal score that plugs into the vision-language uncertainty (ViLU) framework, providing instance-level difficulty and label plausibility to improve prediction set efficiency and class-wise balance at fixed coverage. **LATA** is black-box (no VLM updates), compute-light (windowed transduction, no backprop), and includes an optional prior knob that can run strictly label-free or, if desired, in a label-informed variant using calibration marginals once. Across three medical VLMs and nine downstream tasks, **LATA** consistently reduces set size and CCV while matching or tightening target coverage, outperforming prior transductive baselines and narrowing the gap to label-using methods, while using far less compute. Comprehensive ablations and qualitative analyses show that **LATA** sharpens zero-shot predictions without compromising exchangeability.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a5a36366-614e-4341-bc62-23cce9ee0ab7Builds on11
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- MedCLIP: Contrastive Learning from Unpaired Medical Images and TextZifeng Wang, Zhenbang Wu, Dinesh Agarwal, Jimeng SunEMNLP 2022 · 907 citations
- Class-Conditional Conformal Prediction with Many ClassesTiffany Ding, Anastasios Angelopoulos, Stephen Bates, Michael I. Jordan et al.NeurIPS 2023 · 160 citations
- Information Maximization for Few-Shot LearningMalik Boudiaf, Imtiaz Masud Ziko, Jérôme Rony, Jose Dolz et al.NeurIPS 2020 · 136 citations
- EASE: Unsupervised Discriminant Subspace Learning for Transductive Few-Shot LearningHao Zhu, Piotr KoniuszCVPR 2022 · 54 citations
Related papers
- Conformal Prediction for Zero-Shot ModelsJulio Silva-Rodríguez, Ismail Ben Ayed, Jose DolzCVPR 2025
- CAOS: Conformal Aggregation of One-Shot PredictorsMaja WaldronICML 2026
- Conformal Calibration TransferAchref DoulaICML 2026 · 5 citations
- Multi-Label Test-Time Adaptation with Bayesian Conditional PriorsQiru Li, Ao Zhou, Zhiwei Jiang, Zifeng Cheng et al.ICML 2026 · 1 citation
- Domain-Shift-Aware Conformal Prediction for Large Language ModelsZhexiao Lin, Yuanyuan Li, Neeraj Sarna, Yuanyuan Gao et al.ICML 2026 · 6 citations
