atlas

Research map / Vision-language and multimodal / Cross-modal learning

Seeing the Image: Prioritizing Visual Correlation by Contrastive Alignment

NeurIPS 2024 · Vision-language and multimodal › Cross-modal learning · approach: Contrastive vision-language alignment · 1 citations (OpenAlex)

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 Contrastive Alignment (CAL), a training-time token re-weighting that weights each text token's MLE loss by its logit difference with vs. without the image, improving VLMs such as LLaVA-1.5, LLaVA-NeXT and MGM at negligible extra cost.

Problem

Standard autoregressive image-text alignment in VLMs weights all text tokens equally, so tokens weakly correlated with or contradicting the image receive as much training signal as visually grounded ones.

Approach

Use the logit difference between predictions with and without the image as a per-token visual-correlation signal; clamp and average-pool these differences and use them as weights in the cross-entropy loss, so visually correlated tokens are prioritized. This is a training-time analogue of visual contrastive decoding.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · VQA-DocANLS: 78.4 -> 80.1Helped
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · VQA-Chartrelaxed accuracy: 63.8 -> 67.2Helped
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · OCR-Benchscore: 553 -> 574Helped
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · COCO CaptionCIDEr: 118.5 -> 120.6Helped
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · TextCapsCIDEr: 118.2 -> 124.4Helped
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · RefCOCOg valIoU: 79.8 -> 80.4Helped
Apply CAL token re-weightingLLaVA-NeXT Vicuna-13B · RefCOCOg testIoU: 79.6 -> 80.3Helped
Apply CAL in IT stage only (ablation)LLaVA-NeXT Vicuna-13B · VQA-DocANLS: 78.4 -> 79.9Helped
Add CAL in PT stage in addition to IT (ablation)LLaVA-NeXT Vicuna-13B · VQA-DocANLS: 79.9 -> 80.1Helped
Remove average pooling (ablation)LLaVA-NeXT Vicuna-13B + CAL · TextCapsCIDEr: 124.4 -> 123.8Hurt
Set clamping bounds to [0, +∞] (ablation)LLaVA-NeXT Vicuna-13B + CAL · VQA-Chartrelaxed accuracy: 67.2 -> 66.6Hurt
Use random patch masking with mask ratio 0.7 instead of the standard image contrast (ablation)LLaVA-NeXT Vicuna-13B + CAL · OCR-Benchscore: 574 -> 541Hurt

Surprising findings

Limitations

Experimental setup

Datasets: LLaVA-665k VQA-Doc VQA-Chart ShareGPT4V LLaVA-filtered CC3M ALLaVA LAION-GPT-4V LIMA OpenAssistant2 DVQA AI2D

Benchmarks: VQA-Text VQA-Doc VQA-Chart OCR-Bench MMT-Bench MMStar SQA-I COCO Caption TextCaps RefCOCOg MME POPE

Metrics: accuracy ANLS relaxed accuracy CIDEr IoU

Baselines: LLaVA-1.5 LLaVA-NeXT MGM MGM-HD VCD IBD CRG

Builds on: LLaVA-1.5 LLaVA-NeXT Mini-Gemini visual contrastive decoding image-based decoding

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