从干扰视频中学习语义知识,提升强化学习样本效率。
Disentangled World Models: Learning to Transfer Semantic Knowledge from Distracting Videos for Reinforcement Learning
- 通过离线预训练提取语义特征,再经潜在蒸馏迁移到在线环境。
- 在多个基准上实现更优性能,样本效率显著提升。
- 适合需要跨域迁移的强化学习场景,尤其关注语义解耦。
在实际场景中训练视觉强化学习(RL)面临巨大挑战,即环境变化导致智能体样本效率低下。尽管已有方法尝试通过解耦表征学习缓解此问题,但通常从零开始训练,缺乏对世界先验知识的利用。本文提出一种可解释的基于模型的强化学习框架——解耦世界模型(Disentangled World Models, DisWM),通过离线到在线的潜在蒸馏,从干扰视频中学习并理解底层语义变化。具体而言,我们离线预训练一个无动作视频预测模型,并施加解耦正则化,以从干扰视频中提取语义知识。随后,通过潜在蒸馏将预训练模型的解耦能力迁移到世界模型中。在线微调阶段,利用预训练模型的知识,并引入解耦约束于世界模型。适应过程中,线上交互产生的动作与奖励丰富了数据多样性,进一步增强解耦表征学习。实验结果验证了该方法在多个基准上的优越性。
原文摘要 · Abstract (English)
Training visual reinforcement learning (RL) in practical scenarios presents a significant challenge, $\textit{i.e.,}$ RL agents suffer from low sample efficiency in environments with variations. While various approaches have attempted to alleviate this issue by disentangled representation learning, these methods usually start learning from scratch without prior knowledge of the world. This paper, in contrast, tries to learn and understand underlying semantic variations from distracting videos via offline-to-online latent distillation and flexible disentanglement constraints. To enable effective cross-domain semantic knowledge transfer, we introduce an interpretable model-based RL framework, dubbed Disentangled World Models (DisWM). Specifically, we pretrain the action-free video prediction model offline with disentanglement regularization to extract semantic knowledge from distracting videos. The disentanglement capability of the pretrained model is then transferred to the world model through latent distillation. For finetuning in the online environment, we exploit the knowledge from the pretrained model and introduce a disentanglement constraint to the world model. During the adaptation phase, the incorporation of actions and rewards from online environment interactions enriches the diversity of the data, which in turn strengthens the disentangled representation learning. Experimental results validate the superiority of our approach on various benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。