Anchor-Final Self-Supervision Drives Hallucination-Aware Optimization in Large Vision-Language Models
Abstract
Hallucinations in large vision-language models (LVLMs) remain a critical challenge, where models often generate tokens that fail to align with visual evidence. To address this issue, we propose AFS: Anchor-Final Self-Supervision, a novel framework for hallucination-aware optimization in LVLMs. By leveraging discrepancies between intermediate and final layer predictions, AFS selectively applies self-supervision to visually descriptive tokens, incorporates hallucination-aware token classification, and encourages consistency between intermediate and final layer distributions. Unlike traditional methods that rely on explicit supervision or post-hoc interventions, AFS optimizes the model via Group Relative Policy Optimization (GRPO), using token-specific rewards derived from internal model signals. Experiments demonstrate that AFS significantly reduces hallucinations without compromising recall in caption generation. Beyond captioning, AFS excels in discriminative tasks, improving the reliability of object existence predictions and multimodal reasoning. Furthermore, AFS demonstrates strong cross-dataset generalization, transferring effectively across diverse visual domains. Code is available at \url{https://github.com/guavayew/AFS}.
Lay Summary
Large vision-language models can sometimes produce outputs that are not supported by the visual input, reducing their reliability in real-world applications. We developed a new training framework, called AFS, that uses signals from the model’s internal prediction dynamics to encourage visually grounded outputs and mitigate hallucinations. Our method improves the reliability of multimodal reasoning and visual understanding across different tasks and visual domains.