LASER: A Corrective Lens for LVLMs via Visual Attention Preservation and Sink Suppression

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

Summary

LASER is a novel post-training framework designed to mitigate "visual forgetting" in Large Vision-Language Models (LVLMs) during long-horizon decoding. This issue, where attention drifts from visual evidence, has been re-evaluated, revealing that performance degradation stems from early-stage attention decay and concentration on task-irrelevant visual sink tokens, rather than solely late-stage decay. LASER addresses these factors by regulating both the visual attention trajectory and intra-visual token attention distribution. It employs two complementary rewards: a Visual Grounding Reward, which ensures sustained attention on semantically salient visual tokens, and a Sink Suppression Reward, which prevents excessive focus on uninformative sink tokens. This dual approach preserves early-stage grounding and prevents attention collapse. Extensive experiments across eight benchmark datasets demonstrate LASER's consistent superior performance over existing baselines, validating attention-aware training as an effective solution.

Key takeaway

For Machine Learning Engineers developing Large Vision-Language Models, if you are struggling with visual forgetting during long-horizon decoding, consider implementing attention-aware post-training frameworks like LASER. Your models' performance can significantly improve by explicitly regulating visual attention trajectory and suppressing focus on uninformative sink tokens. This approach directly addresses early-stage attention decay, ensuring more robust visual grounding and preventing attention collapse in complex reasoning tasks.

Key insights

LVLM visual forgetting stems from early attention decay and sink token concentration, correctable via attention-aware training.

Principles

Method

LASER applies a post-training framework using a Visual Grounding Reward to sustain attention on salient visual tokens and a Sink Suppression Reward to penalize focus on uninformative sink tokens.

In practice

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.