SAGE: Sink-Aware Grounded Decoding for Multimodal Hallucination Mitigation
Organizations: Georgia Institute of Technology
Abstract
Large vision-language models (VLMs) frequently suffer from hallucinations, generating content that is inconsistent with visual inputs. Existing methods typically address this problem through post-hoc filtering, additional training objectives, or external verification, but they do not intervene during the decoding process when hallucinations arise. In this work, we introduce SAGE, a Sink-Aware Grounded Decoding framework that mitigates hallucinations by dynamically modulating self-attention during generation. Hallucinations are strongly correlated with attention sink tokens - punctuation or function tokens that accumulate disproportionate attention despite carrying limited semantic content. SAGE leverages these tokens as anchors to monitor grounding reliability in real time. At each sink trigger, the method extracts semantic concepts from the generated sequence, estimates their visual grounding using both self-attention maps and gradient-based attribution, and measures their spatial agreement. Based on this signal, self-attention distributions are adaptively sharpened or broadened to reinforce grounded regions or suppress unreliable ones. Extensive experiments across diverse hallucination benchmarks demonstrate that SAGE consistently outperforms existing decoding strategies, achieving substantial reductions in hallucination while preserving descriptive coverage, without requiring model retraining or architectural modifications. Our method achieves an average relative improvement of 10.65% on MSCOCO and 7.19% on AMBER across diverse VLM architectures, demonstrating consistent gains in hallucination mitigation.
Figures & tables
| Method | LLaVA-1.5 | Shikra | MiniGPT-4 | InstructBLIP | Qwen2-VL-7B | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| OPERA | 45.2 | 12.4 | 37.2 | 12.3 | 26.0 | 9.1 | 46.7 | 14.6 | 23.7 | 7.1 |
| HALC | 36.9 | 9.8 | 38.3 | 15.9 | 28.5 | 10.3 | 49.4 | 14.2 | 22.5 | 6.9 |
| EAH | 37.8 | 10.4 | 47.5 | 13.2 | 30.7 | 9.9 | 56.4 | 9.9 | 22.4 | 6.5 |
| DML-LVLM | 25.7 | 7.2 | 24.6 | 9.9 | 23.5 | 8.8 | 32.5 | 13.8 | 25.7 | 7.2 |
| GIFT | 37.4 | 9.6 | 33.5 | 11.7 | 29.3 | 10.2 | 38.8 | 11.9 | 22.5 | 8.1 |
| Method | LLaVA-1.5 | Shikra | MiniGPT-4 | InstructBLIP | Qwen2-VL-7B | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| OPERA | 12.1 | 45.3 | 11.2 | 42.6 | 10.3 | 50.4 | 12.2 | 48.7 | 8.8 | 52.4 |
| HALC | 9.7 | 51.7 | 16.2 | 39.6 | 11.1 | 53.6 | 13.1 | 49.3 | 7.8 | 54.7 |
| EAH | 11.3 | 48.4 | 12.6 | 52.3 | 10.6 | 54.5 | 10.7 | 49.1 | 7.1 | 55.2 |
| DML-LVLM | 7.8 | 52.6 | 9.2 | 46.7 | 8.9 | 47.4 | 13.6 | 44.3 | 8.1 | 51.6 |
| GIFT | 8.9 | 50.6 | 10.2 | 47.2 | 9.6 | 41.8 | 11.2 | 39.4 | 6.3 | 58.7 |
| Method | LLaVA-1.5 | Qwen2-vl-7b | ||
|---|---|---|---|---|
| GIFT [ 33 ] | ||||
| LVLMs-Saliency [ 55 ] | ||||
| SAGE(Ours) | 6.9 | 6.6 | 8.2 | 8.1 |
| Method | LLaVA-1.5 | Shikra | MiniGPT-4 | InstructBLIP | Qwen2-VL-7B | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| SAGE w/o Grad-CAM | 47.4 | 13.2 | 50.8 | 13.9 | 49.6 | 14.6 | 54.2 | 13.7 | 38.3 | 11.2 |
| SAGE (Ours) | 21.3 | 5.2 | 24.3 | 8.7 | 23.8 | 8.2 | 25.7 | 9.4 | 21.1 | 5.9 |
| Hyperparameters | LLaVA-1.5 | Qwen2-VL-7B | ||||||
|---|---|---|---|---|---|---|---|---|
| Reinforce only ( ) | ||||||||
| 0.0 | 1.2 | 0.5 | 56.7 | 14.7 | 50.6 | 49.9 | 13.8 | 58.9 |
| 0.0 | 1.5 | 0.5 | 57.2 | 14.8 | 47.7 | 50.3 | 13.9 | 56.7 |
| 0.0 | 1.8 | 0.5 | 57.9 | 15.3 | 48.0 | 53.7 | 14.2 | 55.2 |
| 0.0 | 2.0 | 0.5 | 59.3 | 15.2 | 48.7 | 54.2 | 14.3 | 53.2 |
| Method | LLaVA-1.5 | Shikra | MiniGPT-4 | InstructBLIP | Qwen2-VL-7B | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| Periodic Modulation (every 10 tokens) | 39.4 | 12.5 | 41.3 | 12.9 | 39.9 | 13.1 | 38.2 | 11.7 | 34.6 | 10.8 |
| Sink-triggered Modulation (SAGE) | 21.3 | 5.2 | 24.3 | 8.7 | 23.8 | 8.2 | 25.7 | 9.4 | 21.1 | 5.9 |
| Component | Avg. Additional Cost (%) | Description |
|---|---|---|
| Concept Extraction (POS Tagging) | 1.2 | Extract key concepts from decoded text |
| Attention Map Extraction + IoU | 1.7 | Compute spatial overlap between maps |
| Grad-CAM Attribution | 7.6 | Compute visual attribution from vision encoder |
| Attention Modulation | 0.3 | Rescale attention weights |
| Total SAGE Overhead | 10.8 | Relative to standard decoding |
VLM Hallucination Mitigation
Vision-language model hallucination mitigation keeps generated claims grounded in visual evidence.
See all 228 papers