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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Retrieval | Flickr | R@114.4 | 35 | |
| Brain Disease Classification | Pub-Brain-5-gt | BAcc (Stroke)99 | 8 | |
| Chest CT classification | CT-Rate 16 tasks (test) | Mean AUC83.2 | 7 | |
| Chest CT classification | Rad-ChestCT 14 tasks (external val) | Mean AUC74.7 | 7 | |
| 30-way classification | RSICD (test) | Mean Rank3.76 | 6 | |
| Brain Disease Classification | Pub-Brain-5 | Stroke BAcc97.3 | 6 | |
| Head CT classification | RSNA 5-tasks | mAUC91.5 | 6 | |
| Head CT classification | CQ500 10-tasks | mAUC90 | 6 | |
| Image-to-Text Retrieval | Flickr | T@119.2 | 6 | |
| Text-to-Image Retrieval | MSCOCO | I@16.1 | 6 |