用扩散模型生成轨迹,自动识别环境差异并调整策略。
DADiff: Diffusion-Driven Cross-Domain Policy Adaptation for Reinforcement Learning

- 基于扩散模型生成状态轨迹,捕捉源域与目标域的动态差异。
- 在多种环境迁移任务中表现优于现有方法,性能提升显著。
- 适合需要少量目标域交互的在线策略迁移场景。
在强化学习中,跨域策略迁移面临源域与目标域动力学不匹配的挑战。本文研究在线动力学自适应场景:源域有充足数据训练策略,但目标域仅允许有限交互。现有方法多依赖领域分类器、价值引导的数据过滤或表征学习。我们从生成建模视角出发,提出DADiff——一种基于扩散模型的框架,通过分析生成下一状态时源域与目标域生成轨迹的差异,估计动力学失配程度。设计了奖励修正与数据选择两种变体来适应目标域策略。理论分析表明,策略在两域间的性能差距受生成轨迹偏差约束。在多种环境偏移设置下进行大量实验,结果证明该方法优于现有方法,有效缓解动力学失配问题。代码已开源。
原文摘要 · Abstract (English)
Transferring policies across domains poses a vital challenge in reinforcement learning, due to the dynamics mismatch between the source and target domains. In this paper, we consider the setting of online dynamics adaptation, where policies are trained in the source domain with sufficient data, while only limited interactions with the target domain are allowed. There are a few existing works that address the dynamics mismatch by employing domain classifiers, value-guided data filtering, or representation learning. Instead, we study the domain adaptation problem from a generative modeling perspective. Specifically, we introduce DADiff, a diffusion-based framework that leverages the discrepancy between source and target domain generative trajectories in the generation process of the next state to estimate the dynamics mismatch. Both reward modification and data selection variants are developed to adapt the policy to the target domain. We also provide a theoretical analysis to show that the performance difference of a given policy between the two domains is bounded by the generative trajectory deviation. More discussions on the applicability of the variants and the connection between our theoretical analysis and the prior work are further provided. We conduct extensive experiments in environments with various shifts to validate the effectiveness of our method. The results demonstrate that our method provides superior performance compared to existing approaches, effectively addressing the dynamics mismatch. We provide the code of our method at https://github.com/hanyang-chen/DADiff-release
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。