通过修正源域数据动态,提升跨域离线强化学习性能
Cross-Domain Offline Policy Adaptation via Selective Transition Correction
- 用逆策略和奖励模型修正源域动作与奖励
- 结合前向动力学筛选更匹配目标域的样本
- 适合有源域数据但环境动态不同的迁移场景
在强化学习中,跨域策略适配面临动态不匹配的挑战。本文研究跨域离线强化学习,即利用相似源域的离线数据来增强目标域数据上的策略学习。直接合并数据可能导致性能下降。现有方法多通过过滤源域转移或修改奖励缓解问题,但易低估源域数据价值。本文提出将源域数据转换为目标域数据:利用逆策略模型和奖励模型修正源域转移的动作与奖励,显式对齐目标域动态;为应对数据有限导致模型训练不准的问题,进一步采用前向动力学模型保留更符合目标域动态的修正样本。由此提出选择性转移修正(STC)算法,实现源域数据的可靠利用。在多种存在动态偏移的环境中实验表明,STC显著优于现有基线。
原文摘要 · Abstract (English)
It remains a critical challenge to adapt policies across domains with mismatched dynamics in reinforcement learning (RL). In this paper, we study cross-domain offline RL, where an offline dataset from another similar source domain can be accessed to enhance policy learning upon a target domain dataset. Directly merging the two datasets may lead to suboptimal performance due to potential dynamics mismatches. Existing approaches typically mitigate this issue through source domain transition filtering or reward modification, which, however, may lead to insufficient exploitation of the valuable source domain data. Instead, we propose to modify the source domain data into the target domain data. To that end, we leverage an inverse policy model and a reward model to correct the actions and rewards of source transitions, explicitly achieving alignment with the target dynamics. Since limited data may result in inaccurate model training, we further employ a forward dynamics model to retain corrected samples that better match the target dynamics than the original transitions. Consequently, we propose the Selective Transition Correction (STC) algorithm, which enables reliable usage of source domain data for policy adaptation. Experiments on various environments with dynamics shifts demonstrate that STC achieves superior performance against existing baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。