通过时空掩码增强对比学习,让视觉强化学习更懂状态演变。
MOOSS: Mask-Enhanced Temporal Contrastive Learning for Smooth State Evolution in Visual Reinforcement Learning
- 用图结构建模像素的时空关联,实现精准掩码
- 多层级对比学习提升状态表示的时序连续性
- 适合追求样本效率的视觉强化学习研究者
在基于像素的视觉强化学习中,从高维数据中提取有效状态表示面临巨大挑战,尤其影响样本效率。现有对比学习方法虽有进展,但难以捕捉状态的细微演化过程。为此,本文提出MOOSS框架,利用基于图的时空掩码与时间对比学习目标,显式建模视觉强化学习中的状态演化。具体地,设计了一种自监督双组件策略:(1)基于像素观测构建图结构以实现时空掩码;(2)采用多层级对比学习机制,通过强调状态的时序连续性与变化来丰富状态表示。通过破坏并学习时空相关性,MOOSS深化了对状态动态的理解,促进策略学习。在多个连续和离散控制基准测试中,MOOSS在样本效率上超越先前最先进方法,验证了其有效性。代码已开源:https://github.com/jsun57/MOOSS。
原文摘要 · Abstract (English)
In visual Reinforcement Learning (RL), learning from pixel-based observations poses significant challenges on sample efficiency, primarily due to the complexity of extracting informative state representations from high-dimensional data. Previous methods such as contrastive-based approaches have made strides in improving sample efficiency but fall short in modeling the nuanced evolution of states. To address this, we introduce MOOSS, a novel framework that leverages a temporal contrastive objective with the help of graph-based spatial-temporal masking to explicitly model state evolution in visual RL. Specifically, we propose a self-supervised dual-component strategy that integrates (1) a graph construction of pixel-based observations for spatial-temporal masking, coupled with (2) a multi-level contrastive learning mechanism that enriches state representations by emphasizing temporal continuity and change of states. MOOSS advances the understanding of state dynamics by disrupting and learning from spatial-temporal correlations, which facilitates policy learning. Our comprehensive evaluation on multiple continuous and discrete control benchmarks shows that MOOSS outperforms previous state-of-the-art visual RL methods in terms of sample efficiency, demonstrating the effectiveness of our method. Our code is released at https://github.com/jsun57/MOOSS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。