LASER: A Corrective Lens for LVLMs via Visual Attention Preservation and Sink Suppression
About
Large vision-language models (LVLMs) exhibit strong reasoning ability but suffer from visual forgetting during long-horizon decoding, where attention progressively drifts away from visual evidence. Existing methods largely treat this issue as a late-stage attention decay problem or attempt to mitigate it through heuristic reminders or post-hoc attention lifting. Through systematic empirical analysis, we find that performance degradation under visual forgetting is largely driven by two overlooked factors: early-stage attention decay disrupts evidence acquisition, and attention concentration on a subset of task-irrelevant visual sink tokens. Motivated by these insights, we propose LASER, a post-training framework that regulates both the visual attention trajectory and intra-visual token attention distribution during reasoning. Technically, LASER introduces two complementary rewards: a Visual Grounding Reward, which encourages the model to maintain attention on semantically salient visual tokens throughout decoding, and a Sink Suppression Reward, which penalizes excessive attention concentration on visual sink tokens. Together, these rewards preserve early-stage grounding while preventing attention collapse onto uninformative regions. Extensive experiments on eight benchmark datasets demonstrate that LASER consistently outperforms strong baselines, validating attention-aware training as an effective remedy for visual forgetting.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Mathematical Reasoning | WeMath | Accuracy44.4 | 317 | |
| Mathematical Reasoning | MathVerse | Accuracy55.2 | 266 | |
| Logical reasoning | LogicVista | Accuracy51.2 | 163 | |
| General Reasoning | MMMU | Overall Score62.1 | 57 | |
| Visual Reasoning | HallusionBench | Accuracy60.7 | 43 | |
| Math Reasoning | MathVision | Accuracy44.6 | 20 | |
| General Reasoning | MMStar | Accuracy (General Reasoning)64.1 | 20 | |
| Math Reasoning | MathVista | Accuracy72.9 | 10 |