atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

Beyond Accuracy: Ensuring Correct Predictions With Correct Rationales

NeurIPS 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Confidence calibration objectives

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes double-correct predictions, curating a GPT-4-generated structured rationale dataset for ImageNet categories and a rationale-informed optimization (disentanglement plus reconstruction constraints solved via Lagrangian) that fine-tunes CLIP ViT-B/32 to improve both prediction accuracy and rationale localization.

Problem

Foundation models such as CLIP are evaluated mainly on prediction accuracy, ignoring whether correct predictions rest on valid visual evidence, which is needed for safe deployment.

Approach

Decompose ViT image embeddings per token and layer (weighted by mean-ablation importance) to get rationale heatmaps, then train CLIP with self-supervised constraints that make heatmaps of different rationales disentangled while their aggregate reconstructs the category text embedding, requiring no pixel annotations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply rationale-informed optimizationCLIP ViT-B/32 · Nine classification datasets (AVG), zero-shotaccuracy (%): 2.6%Helped
Apply rationale-informed optimizationCLIP ViT-B/32 · Nine classification datasets (AVG), linear probeaccuracy (%): 2.0%Helped
Apply rationale-informed optimizationCLIP ViT-B/32 · CUB-Partrationale localizability (mIoU): 7.5%Helped
Apply rationale-informed optimizationCLIP ViT-B/32 · PartImageNetrationale localizability (mIoU): 6.0%Helped
Apply rationale-informed optimizationCLIP ViT-B/32 · Nine classification datasetsrationale disentanglability: 36.5%Helped
Add rationale disentanglement constraint (ablation)Rationale-informed optimization · CUB-Partrationale localizability (mIoU): 10.4%Helped
Add reconstruction constraint (ablation)Rationale-informed optimization · CUB-Partrationale localizability (mIoU): 13.3%Helped
Add reconstruction constraint (ablation)Rationale-informed optimization · CUBaccuracy (%): 30.2%Helped
Use an external concept set (ablation)Rationale-informed optimization · ImageNet, rationale-based predictionaccuracy (%)Helped
Replace rationale concepts with random strings (ablation)Rationale-informed optimization · ImageNet, rationale-based predictionaccuracy (%): -2.2Hurt

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet CUB-200-2011 CUB-Part PartImageNet CIFAR-10 CIFAR-100 Caltech-101 Oxford-IIIT Pets Food-101 SUN397 Stanford Cars DTD

Benchmarks: CIFAR-10 CIFAR-100 CUB-200-2011 Caltech-101 Oxford-IIIT Pets Food-101 SUN397 Stanford Cars DTD Flickr30K MSCOCO CUB-Part

Metrics: accuracy mIoU disentanglability (1 - |cosine|) Likert score

Baselines: CLIP DeCLIP NegCLIP FILIP PyramidCLIP CLIP-ft CLIP-ft-vision

Builds on: CLIP DCLIP WaffleCLIP InfoNCE

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy