New Method Improves Medical VLM Coverage for Rare Diseases

Mushir Akhtar, M. Tanveer· August 3, 2026 View original

Key takeaways

  • Medical VLMs often under-cover rare disease classes, even with high overall coverage.
  • CALCoDe is a new post-hoc method to mitigate this class-tail undercoverage.
  • It combines class-conditional and localized conformal thresholds for robust protection.
  • CALCoDe significantly outperforms other methods in ensuring high worst-class coverage under clinical shifts.

Who benefits

HealthcareMedical DiagnosticsPharmaceuticalsAI DevelopmentHealthTech

Summary

This paper introduces Class-Tail Adaptive Localized Conformal Deferral (CALCoDe), a post-hoc reliability layer for medical Vision-Language Models (VLMs) that mitigates class-tail undercoverage, especially for rare diseases, under clinical data shifts. It ensures high marginal and worst-class accepted coverage, outperforming existing conformal methods.

Medical Vision-Language Models (VLMs) can maintain high overall prediction coverage even when clinical data shifts occur, but they often severely under-cover specific disease classes, particularly rare ones. This undercoverage is problematic because the affected classes can vary unpredictably with different data acquisition protocols or model architectures. Standard conformal methods, which adapt to local test neighborhoods or source-frequency tails, fail to adequately address these class-wise coverage failures. To tackle this, researchers developed Class-Tail Adaptive Localized Conformal Deferral (CALCoDe), a post-hoc reliability layer designed for frozen medical VLMs. CALCoDe first identifies classes at risk of undercoverage using cross-fitted validation predictions. It then estimates class-conditional tail thresholds from a separate calibration split. These protected thresholds are combined with localized conformal thresholds, ensuring that all labels admitted by the localized rule are included, with additional protection for identified high-risk classes. An independently calibrated support audit further defers cases lacking sufficient inlier support. CALCoDe guarantees finite-sample coverage at a prespecified guard level for accepted examples within protected classes. Empirical evaluations across two dermatology dataset shifts and four VLM backbones showed that CALCoDe was the only approach to achieve 0.95 marginal and worst-class accepted coverage in all eight settings, significantly outperforming other conformal baselines.

Why it matters

For professionals developing and deploying AI in healthcare, especially in diagnostics, ensuring reliable and equitable performance across all disease classes, including rare ones, is paramount. CALCoDe offers a crucial tool to improve the trustworthiness and clinical utility of medical VLMs by addressing a critical safety and fairness concern.

How to implement this in your domain

  1. 1Integrate CALCoDe as a post-hoc reliability layer for existing medical Vision-Language Models to improve coverage for rare or underrepresented disease classes.
  2. 2Evaluate the performance of CALCoDe on proprietary medical imaging datasets, especially those with known class imbalances or data shifts.
  3. 3Develop robust validation pipelines to identify classes at risk of undercoverage in medical AI models.
  4. 4Collaborate with clinical experts to define acceptable coverage levels and integrate deferral mechanisms into diagnostic workflows.
  5. 5Benchmark CALCoDe against other conformal prediction methods to assess its effectiveness in mitigating class-tail undercoverage.

Original post by Mushir Akhtar, M. Tanveer

"arXiv:2607.28696v1 Announce Type: new Abstract: Medical vision-language models (VLMs) can retain high observed marginal coverage after clinical shift while substantially under-covering an individual disease class. The affected class varies with acquisition protocol and backbone g…"

View on X

Originally posted by Mushir Akhtar, M. Tanveer on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses