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

Polos: Multimodal Metric Learning from Human Feedback for Image Captioning

About

Establishing an automatic evaluation metric that closely aligns with human judgments is essential for effectively developing image captioning models. Recent data-driven metrics have demonstrated a stronger correlation with human judgments than classic metrics such as CIDEr; however they lack sufficient capabilities to handle hallucinations and generalize across diverse images and texts partially because they compute scalar similarities merely using embeddings learned from tasks unrelated to image captioning evaluation. In this study, we propose Polos, a supervised automatic evaluation metric for image captioning models. Polos computes scores from multimodal inputs, using a parallel feature extraction mechanism that leverages embeddings trained through large-scale contrastive learning. To train Polos, we introduce Multimodal Metric Learning from Human Feedback (M$^2$LHF), a framework for developing metrics based on human feedback. We constructed the Polaris dataset, which comprises 131K human judgments from 550 evaluators, which is approximately ten times larger than standard datasets. Our approach achieved state-of-the-art performance on Composite, Flickr8K-Expert, Flickr8K-CF, PASCAL-50S, FOIL, and the Polaris dataset, thereby demonstrating its effectiveness and robustness.

Yuiga Wada, Kanta Kaneda, Daichi Saito, Komei Sugiura• 2024

Related benchmarks

TaskDatasetResultRank
Image Captioning EvaluationComposite
Kendall-c Tau_c57.6
92
Image Captioning EvaluationFlickr8K Expert (test)
Kendall tau_c56.4
76
Image Captioning EvaluationFlickr8k Expert
Kendall Tau-c (tau_c)56.4
73
Image Captioning EvaluationPascal-50S (test)
HC70
66
Image Captioning EvaluationFlickr8K-CF (test)
Kendall tau_b37.8
65
Image Captioning EvaluationFlickr8K-CF
Kendall-b Correlation (tau_b)37.8
62
Image Captioning EvaluationPascal-50S
Mean Score86.5
39
Hallucination DetectionFOIL
Accuracy (4 Refs)95.4
32
Image Captioning Hallucination DetectionFOIL (test)
Accuracy95.4
28
Correlation with human judgmentFlickr8K-CF
Tau B37.8
26
Showing 10 of 19 rows

Other info

Code

Follow for update