Abstract
Reinforced Attention Learning optimizes internal attention distributions in multimodal language models, improving information allocation and cross-modal alignment through policy-gradient methods.
Post-training with Reinforcement Learning (RL) has substantially improved reasoning in Large Language Models (LLMs) via test-time scaling. However, extending this paradigm to Multimodal LLMs (MLLMs) through verbose rationales yields limited gains for perception and can even degrade performance. We propose Reinforced Attention Learning (RAL), a policy-gradient framework that directly optimizes internal attention distributions rather than output token sequences. By shifting optimization from what to generate to where to attend, RAL promotes effective information allocation and improved grounding in complex multimodal inputs. Experiments across diverse image and video benchmarks show consistent gains over GRPO and other baselines. We further introduce On-Policy Attention Distillation, demonstrating that transferring latent attention behaviors yields stronger cross-modal alignment than standard knowledge distillation. Our results position attention policies as a principled and general alternative for multimodal post-training.
Community
arXivLens breakdown of this paper ๐ https://arxivlens.com/PaperView/Details/reinforced-attention-learning-1838-e9667192
- Executive Summary
- Detailed Breakdown
- Practical Applications
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- KEPO: Knowledge-Enhanced Preference Optimization for Reinforcement Learning with Reasoning (2026)
- LaViT: Aligning Latent Visual Thoughts for Multi-modal Reasoning (2026)
- CORD: Bridging the Audio-Text Reasoning Gap via Weighted On-policy Cross-modal Distillation (2026)
- Masking Teacher and Reinforcing Student for Distilling Vision-Language Models (2025)
- Reinforcement Learning via Self-Distillation (2026)
- Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models (2026)
- Reflective Preference Optimization (RPO): Enhancing On-Policy Alignment via Hint-Guided Reflection (2025)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment:
@librarian-bot
recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
