Mitigating Object Hallucinations in LVLMs via Attention Imbalance Rectification
Organizations: Shanghai Key Laboratory of Trustworthy Computing, East China Normal University
Abstract
Object hallucination in Large Vision-Language Models (LVLMs) severely compromises their reliability in real-world applications, posing a critical barrier to their deployment in high-stakes scenarios such as autonomous driving and medical image analysis. Through systematic empirical investigation, we identify that the imbalanced attention allocation, both across modalities (i.e., vision and language) and within modalities (among individual tokens), exhibits a strong causal correlation with the occurrence of object hallucination. Leveraging this insight, we introduce a novel concept termed attention imbalance, which not only quantifies the degree of attention disparity but also visually delineates the underlying patterns (e.g., over-attentiveness to irrelevant language tokens or under-attentiveness to discriminative visual features) that drive object hallucination. To mitigate object hallucination, we further propose Attention Imbalance Rectification (AIR), a lightweight decoding-time intervention method that reallocates attention weights and adjusts attention distributions to rectify modality-wise and token-wise imbalances. Extensive evaluations on four mainstream LVLMs and three benchmarks (CHAIR, POPE, and MM-Vet) with seven baselines demonstrate that AIR consistently reduces object hallucination rates, achieving up to a 35.1% reduction compared to the baselines, while improving up to 15.9% of LVLMs' general capability across diverse vision-language tasks.
Figures & tables
| Max New Tokens: 256 | Max New Tokens: 64 | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Methods | LLaVA-1.5 | MiniGPT4 | InstructBLIP | Shikra | LLaVA-1.5 | MiniGPT4 | InstructBLIP | Shikra | ||||||||
| Greedy | 51.8 | 13.7 | 43.0 | 13.4 | 54.7 | 20.2 | 55.4 | 15.2 | 20.8 | 6.2 | 28.8 | 11.4 | 29.8 | 14.2 | 22.3 | 8.2 |
| FarSight | 43.9 | 12.5 | 39.2 | 12.4 | 48.3 | 18.7 | 51.6 | 12.2 | 18.6 | 5.9 | 22.5 | 8.9 | 20.2 | 9.6 | 15.4 | 7.6 |
| VCD | 59.4 | 16.0 | 40.0 | 14.6 | 50.1 | 19.4 | 54.9 | 15.1 | 22.8 | 7.1 | 28.0 | 11.3 | 34.1 | 15.8 | 23.5 | 8.3 |
| DoLA | 54.0 | 14.2 | 42.0 | 13.5 | 52.8 | 19.1 | 55.7 | 14.8 | 22.2 | 6.7 | 28.6 | 11.5 | 28.1 | 14.2 | 20.2 | 9.6 |
| Method | Rec | OCR | Know | Gen | Spat | Math | Overall |
|---|---|---|---|---|---|---|---|
| Greedy | 32.7 | 17.6 | 14.5 | 15.3 | 21.9 | 7.7 | 27.6 |
| FarSight | 32.3 | 20.6 | 14.4 | 16.6 | 23.5 | 11.5 | 28.3 (+2.5%) |
| VCD | 32.5 | 21.5 | 15.0 | 17.6 | 24.8 | 3.8 | 28.3 (+2.5%) |
| DoLA | 32.1 | 18.3 | 13.8 | 14.6 | 22.0 | 7.7 | 27.6 (+0.0%) |
| HALC | 32.4 | 17.7 | 14.2 | 15.5 | 19.9 | 7.7 | 27.2 (-1.4%) |
| OPERA | 32.9 | 18.4 | 14.4 | 15.5 | 22.3 | 7.7 | 27.9 (+1.1%) |
| Method | Rec | OCR | Know | Gen | Spat | Math | Overall |
|---|---|---|---|---|---|---|---|
| Greedy | 22.3 | 14.7 | 13.6 | 10.0 | 21.7 | 7.7 | 20.0 |
| FarSight | 26.3 | 12.3 | 14.5 | 14.6 | 14.7 | 3.5 | 20.3 (+1.5%) |
| VCD | 17.1 | 10.0 | 7.9 | 7.8 | 11.7 | 7.3 | 14.2 (-29.0%) |
| DoLA | 20.7 | 14.4 | 13.6 | 10.0 | 21.7 | 7.3 | 18.8 (-6.0%) |
| HALC | 19.4 | 11.9 | 11.3 | 8.6 | 16.5 | 6.5 | 16.6 (-17.0%) |
| OPERA | 20.9 | 15.1 | 10.5 | 10.5 | 19.5 | 12.7 | 18.5 (-7.5%) |
| Method | MM-Vet Overall | |||
|---|---|---|---|---|
| None | 51.8 | 13.7 | 0.83 | 27.6 |
| R-only | 32.1 | 9.9 | 0.85 | 30.5 (+10.5%) |
| P-only | 48.6 | 12.1 | 0.84 | 28.0 (+1.4%) |
| Full (AIR) | 28.8 | 8.6 | 0.86 | 32.0 ( +15.9% ) |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| MM-Vet Overall | ||||
|---|---|---|---|---|
| 32.1 | 9.9 | 0.85 | 30.5 | |
| 28.8 | 8.6 | 0.86 | 32.0 ( +4.9% ) | |
| 30.4 | 9.0 | 0.86 | 31.9 (+4.6%) | |
| 32.0 | 9.2 | 0.85 | 27.4 (-10.2%) |
| Max New Tokens: 256 | ||||||||
| Methods | LLaVA-NeXT-7B [ 37 ] | Qwen-2.5‑VL-7B [ 2 ] | InternVL-3.5-8B [ 17 ] | DeepSeek-VL2-4.5B [ 63 ] | ||||
| Greedy | 31.2 | 8.6 | 38.0 | 9.4 | 32.9 | 9.0 | 42.4 | 11.6 |
| AIR (Ours) | 22.5 | 6.1 | 29.3 | 8.8 | 25.1 | 8.3 | 25.8 | 6.9 |
| 27.9% | 29.1% | 22.9% | 6.4% | 23.7% | 7.8% | 39.2% | 40.5% | |
| Max New Tokens: 256 | ||||||
| Methods | LLaVA-1.5-7B | LLaVA-1.5-13B | LLaVA-NeXT-34B | |||
| Greedy | 51.8 | 13.7 | 46.3 | 12.5 | 25.7 | 6.1 |
| AIR (Ours) | 28.8 | 8.6 | 31.0 | 9.0 | 23.2 | 6.0 |
| 44.4% | 37.2% | 33.0% | 28.0% | 9.7% | 1.6% | |
Object Hallucination in VLMs
Object hallucination in vision-language models is the production of object claims unsupported by visual evidence.
See all 128 papers