Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learning complete and explainable visual representations from itemized text supervision

About

Training vision models with language supervision enables general and transferable representations. However, many visual domains, especially non-object-centric domains such as medical imaging and remote sensing, contain itemized text annotations: multiple text items describing distinct and semantically independent findings within a single image. Such supervision differs from standard multi-caption supervision, where captions are redundant or highly overlapping. Here, we introduce ItemizedCLIP, a framework for learning complete and explainable visual representations from itemized text supervision. ItemizedCLIP employs a cross-attention module to produce text item-conditioned visual embeddings and a set of tailored objectives that jointly enforce item independence (distinct regions for distinct items) and representation completeness (coverage of all items). Across four domains with naturally itemized text supervision (brain MRI, head CT, chest CT, remote sensing) and one additional synthetically itemized dataset, ItemizedCLIP achieves substantial improvements in zero-shot performance and fine-grained interpretability over baselines. The resulting ItemizedCLIP representations are semantically grounded, item-differentiable, complete, and visually interpretable. Our code is available at https://github.com/MLNeurosurg/ItemizedCLIP.

Yiwei Lyu, Chenhui Zhao, Soumyanil Banerjee, Shixuan Liu, Akshay Rao, Akhil Kondepudi, Honglak Lee, Todd C. Hollon• 2025

Related benchmarks

TaskDatasetResultRank
Text-to-Image RetrievalFlickr
R@114.4
35
Brain Disease ClassificationPub-Brain-5-gt
BAcc (Stroke)99
8
Chest CT classificationCT-Rate 16 tasks (test)
Mean AUC83.2
7
Chest CT classificationRad-ChestCT 14 tasks (external val)
Mean AUC74.7
7
30-way classificationRSICD (test)
Mean Rank3.76
6
Brain Disease ClassificationPub-Brain-5
Stroke BAcc97.3
6
Head CT classificationRSNA 5-tasks
mAUC91.5
6
Head CT classificationCQ500 10-tasks
mAUC90
6
Image-to-Text RetrievalFlickr
T@119.2
6
Text-to-Image RetrievalMSCOCO
I@16.1
6
Showing 10 of 12 rows

Other info

Follow for update