arXiv:2510.08442cs.CVcs.AI2025-10

用回报差异引导注意力,让智能体更高效聚焦关键视觉信息

Gaze on the Prize: Shaping Visual Attention with Return-Guided Contrastive Learning

  • 基于回报差异构建对比学习信号,指导注意力聚焦任务相关特征
  • 在ManiSkill3上提升2.52倍样本效率,解决基线无法完成的任务
  • 无需改动算法或超参数,适配性强,适合视觉强化学习场景

视觉强化学习智能体需从高维图像中决策,但仅少量像素与任务相关,导致探索和计算资源浪费,造成样本效率低且学习不稳定。受人类视网膜聚焦机制启发,我们提出Gaze on the Prize框架,引入可学习的中心注视注意力机制(Gaze),其由智能体追求更高回报(即“奖赏”)的经验生成的自监督信号引导。核心思想是:若两个相似表示产生不同回报,则其差异特征很可能任务相关,应予以关注。通过回报差异将相似视觉表示分为正负样本对,构造对比三元组,以此训练注意力机制区分成功与失败状态的特征。该方法在不修改基础算法或超参数的前提下,实现最高2.52倍的样本效率提升,并成功解决了基线无法学习的ManiSkill3挑战任务。

原文摘要 · Abstract (English)

Visual Reinforcement Learning (RL) agents must learn to act based on high-dimensional image data where only a small fraction of the pixels is task-relevant. This forces agents to waste exploration and computational resources on irrelevant features, leading to sample-inefficient and unstable learning. To address this, inspired by human visual foveation, we introduce Gaze on the Prize. This framework augments visual RL with a learnable foveal attention mechanism (Gaze), guided by a self-supervised signal derived from the agent's experience pursuing higher returns (the Prize). Our key insight is that return differences reveal what matters most: If two similar representations produce different outcomes, their distinguishing features are likely task-relevant, and the gaze should focus on them accordingly. This is realized through return-guided contrastive learning that trains the attention to distinguish between the features relevant to success and failure. We group similar visual representations into positives and negatives based on their return differences and use the resulting labels to construct contrastive triplets. These triplets provide the training signal that teaches the attention mechanism to produce distinguishable representations for states associated with different outcomes. Our method achieves up to 2.52x improvement in sample efficiency and can solve challenging tasks from the ManiSkill3 benchmark that the baseline fails to learn, without modifying the underlying algorithm or hyperparameters.

视觉强化学习注意力机制对比学习样本效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。