Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

KEPIL: Knowledge-Enhanced Prompt-Image Learning for Prompt-Robust Disease Detection

About

Vision--language models (VLMs) show promise for clinical decision support in radiology because they enable joint reasoning over radiological images and clinical text, thereby leveraging complementary clinical information. However, radiological findings are long-tailed in practice, leaving some conditions underrepresented and making zero-shot inference essential. Yet current CLIP-style medical VLMs are sensitive to prompt variations and often lack trustworthy external knowledge at inference time, which hinders reliable clinical deployment. We present \textit{KEPIL}, a prompt-robust framework that integrates curated medical knowledge to stabilize zero-shot generalization. KEPIL comprises: (i) \emph{dynamic prompt enrichment} using ontologies with LLM assistance, (ii) a \emph{semantic-aware contrastive loss} aligning embeddings of equivalent prompt variants via a dual-embedding objective, and (iii) \emph{entity-centric report standardization} to yield ontology-aligned representations. Across seven benchmarks, KEPIL achieves state-of-the-art zero-shot inference performance; under prompt-variation tests, it improves AUC by \(6.37\%\) on \textit{CheXpert} and by \(4.11\%\) on average. These results suggest that structured knowledge and robust prompt design are key to clinically reliable radiology-facing VLMs. Code will be released at https://github.com/Roypic/KEPIL.

Haozhe Luo, Shelley Zixin Shu, Ziyu Zhou, Robert Berke, Mauricio Reyes• 2026

Related benchmarks

TaskDatasetResultRank
ClassificationSIIM
AUC93.02
67
Lung nodule classificationLIDC-IDRI
AUC66.65
36
ClassificationRSNA Pneumonia
Accuracy86.24
32
Thoracic Disease ClassificationChestX-ray14
Average Performance83.23
28
Disease ClassificationCheXpert
AUROC0.9121
23
Disease ClassificationPadChest seen
AUC85.32
11
Disease ClassificationPadChest (unseen)
AUC79.05
11
Disease ClassificationPadChest rare
AUC78.75
11
Disease ClassificationCOVID-19 CXR-2
AUC0.7955
11
Showing 9 of 9 rows

Other info

Follow for update