ClaimDiff-RL: Fine-Grained Caption Reinforcement Learning through Visual Claim Comparison

· Source: cs.LG updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, extended

Summary

The ClaimDiff-RL framework addresses the reward granularity problem in long-form image captioning reinforcement learning by introducing reference-conditioned atomic claim differences as its reward unit. Traditional holistic rewards often compress local errors into a single sequence-level signal, obscuring the critical trade-off between caption factuality and coverage. ClaimDiff-RL employs a multimodal judge to enumerate visually grounded differences between an actor caption and a reference caption, verify each difference against the image, and assign open-vocabulary error types and severity levels. These per-difference statistics are then composed into scalar rewards, allowing for separate measurement and tuning of hallucinated claims and omitted salient facts. Experiments on a 160-image human-labeled diagnostic benchmark, public captioning benchmarks, and VQA benchmarks demonstrate that ClaimDiff-RL significantly improves the balance between hallucination and missing facts, preserves general capabilities, and surpasses Gemini-3-Pro-Preview on fine-grained dimensions like object counting and spatial relations.

Key takeaway

For Machine Learning Engineers developing long-form image captioning models, you should consider adopting fine-grained reward mechanisms like ClaimDiff-RL. Your current holistic scalar rewards might inadvertently encourage conservative under-captioning, sacrificing coverage for reduced hallucination. By implementing image-verified claim differences, you can achieve a more balanced and controllable trade-off between factuality and informativeness, potentially surpassing existing models like Gemini-3-Pro-Preview on specific visual attributes. Explore severity weighting and ambiguity penalties to fine-tune your model's behavior.

Key insights

ClaimDiff-RL uses image-verified claim differences as fine-grained RL rewards for balanced, factual, and informative long-form image captions.

Principles

Method

ClaimDiff-RL's multimodal judge identifies actor-reference visual differences, verifies them against the image, assigns typed errors (e.g., color_hallucination, count_mismatch), and composes these statistics into scalar relative or actor-only rewards for GRPO optimization.

In practice

Topics

Code references

Best for: Computer Vision Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.LG updates on arXiv.org.