atlas

Research map / Large language models / Alignment and preferences

Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithms

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Reward model design and preference models · 0 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 a preference-based RL (ranked DPO) method that aligns a CLIP-style retrieval model with human aesthetics by distilling LLM query rephrasing and aesthetic re-ranker signals, plus two aesthetic retrieval benchmarks (HPIR and a GPT-4V win-rate judge).

Problem

Vision-language retrieval models trained on noisy web data often return images that match the query semantically but are visually unappealing or misaligned with user aesthetic intent, and existing retrieval benchmarks do not evaluate aesthetics.

Approach

Use LLM query rephrasing to enrich aesthetic expectations, build partially ordered retrieval sequences ranked by aesthetic and semantic re-rankers, and fine-tune the retrieval model with a ranked DPO loss plus pretraining loss so it internalizes these aesthetic preferences in a single-stage system.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply ranked DPO preference alignment fine-tuningOurs-PT · HPIR Aestheticaesthetic (%): 59.4 -> 67.6Helped
Apply ranked DPO preference alignment fine-tuningOurs-PT · MSCOCORecall@1 (%): 40.2 -> 40.2No clear effect
Apply ranked DPO preference alignment fine-tuningOurs-PT · Flickr30KRecall@1 (%): 66.5 -> 66.4No clear effect
Apply ranked DPO preference alignment fine-tuningCLIP · HPIR Aestheticaesthetic (%): 62.1 -> 71.7Helped
Apply ranked DPO preference alignment fine-tuningDataComp · HPIR Aestheticaesthetic (%): 62.7 -> 71.9Helped
Use LLM query rephrasing with <k list> rather than the original queryretrieval system · HPIR Aestheticaesthetic (%): 59.36 -> 72.42Helped
Use LLM query rephrasing with <detail> rather than the original queryretrieval system · GPT-4V win rate, aestheticwin rate (%): 68.75Helped
Repeat the original query to extend its lengthretrieval system · HPIR Aestheticaesthetic (%): 59.36 -> 65.67Helped
Replace ranked DPO with RRHF loss (ablation)alignment fine-tuning · HPIR Aestheticaesthetic (%): 59.4 -> 62.2Helped
Replace ranked DPO with ranked IPO loss (ablation)alignment fine-tuning · HPIR Aestheticaesthetic (%): 59.4 -> 66.8Helped
Replace ranked DPO with ranked IPO loss (ablation)alignment fine-tuning · HPIR Aestheticaesthetic (%): 67.6 -> 66.8Hurt
Add auto augmentation (ablation)alignment fine-tuning · HPIR Accuracyaccuracy (%): 71.7 -> 74.4Helped

Surprising findings

Limitations

Experimental setup

Datasets: DataComp ImageNet-22k SCVL (self-collected image-text pairs) ImageNet1K MSCOCO Flickr30k

Benchmarks: ImageNet1K zero-shot MSCOCO T2I R@1 Flickr30K T2I R@1 HPIR

Metrics: accuracy (confidence-weighted) aesthetic (confidence-weighted) win rate win-and-similar rate

Baselines: CLIP DataComp Bing image search Getty search Ours-PT + Rerank

Builds on: DPO CLIP RRHF IPO InstructGPT

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