OG-SPR:观察锚定的自预测强化学习用于视觉连续控制

Observation-Grounded Self-Predictive Reinforcement Learning for Visual Continuous Control

精选理由

OG-SPR把两种预测目标结合起来,28个控制任务上超过现有方法,难任务提升明显。搞视觉RL的可以看看。

AI 摘要

本文提出OG-SPR,一种用于视觉连续控制的模型无关强化学习算法,融合多步潜在自预测与下一观察预测两个辅助目标。在DeepMind Control Suite的28个视觉控制任务上,OG-SPR相比现有自预测和观察预测方法获得更高的总体性能,尤其在dog和humanoid等困难任务上提升显著。该方法为共享表征引入两个轻量适配器来执行潜在自预测,避免直接强制表征满足预测目标带来的过约束问题。

原文 · arXiv: Google DeepMind

Observation-Grounded Self-Predictive Reinforcement Learning for Visual Continuous Control

Sample-efficient policy learning from pixels is a long-standing challenge in reinforcement learning (RL). Recent dynamics-based representation learning methods have significantly improved the sample efficiency of model-free visual RL by learning dynamics-aware representations through auxiliary prediction performed either in latent space (self-prediction) or observation space (observation prediction). However, state-of-the-art methods from both categories still struggle on challenging visual control tasks when training data is limited. We posit that relying on either predictive objective alone may be insufficient. In contrast, observation prediction grounds learned representations in observation-level dynamics, but does not directly regularize the temporal predictability of latent representations over extended horizons. In this paper, we propose Observation-Grounded Self-Predictive Representations (OG-SPR), a model-free visual RL algorithm for continuous control that learns representations that are both temporally predictive in latent space and grounded in observation-level dynamics. OG-SPR incorporates two core auxiliary objectives: multi-step latent self-prediction and next-observation prediction. We empirically show that directly imposing latent self-prediction on the shared representation may over-constrain it and does not necessarily improve performance. To address this issue, OG-SPR introduces two lightweight adapters for latent self-prediction, allowing the shared representation to benefit from temporally predictive signals without being forced to directly satisfy the self-prediction objective. Experiments on 28 visual control tasks from the DeepMind Control Suite show that OG-SPR improves aggregate performance over state-of-the-art self-predictive and observation-predictive RL methods, with particularly pronounced gains in challenging domains such as dog and humanoid.