用元学习优化动作排序,让低质量示范数据也能提升智能体性能
Imitation Learning from Suboptimal Demonstrations via Meta-Learning An Action Ranker
- 通过元学习构建动作排序器,动态加权示范数据
- 在少量专家数据基础上,显著提升对次优示范的利用效率
- 适合示范数据难获取或质量参差的强化学习场景
模仿学习面临的一大瓶颈是需要大量专家示范,而这些示范往往成本高或难以获取。从无需严格质量要求的补充示范中学习已成为解决该问题的有效方法。然而,以往方法常因丢弃非专家数据而未能充分挖掘其潜力。我们的核心洞察是:即使某些示范不在专家分布内,但若其表现优于当前策略,仍可提升策略性能。为此,我们提出一种新方法——基于元学习的动作排序器模仿学习(ILMAR)。ILMAR在有限专家示范与补充示范上实施加权行为克隆,利用优势函数的特性,选择性地融合补充示范的知识。为更有效利用补充示范,引入元目标,通过显式最小化当前策略与专家策略之间的距离来优化优势函数功能。在多种任务上的全面实验表明,ILMAR在处理次优示范方面显著优于现有方法。代码已公开于 https://github.com/F-GOD6/ILMAR。
原文摘要 · Abstract (English)
A major bottleneck in imitation learning is the requirement of a large number of expert demonstrations, which can be expensive or inaccessible. Learning from supplementary demonstrations without strict quality requirements has emerged as a powerful paradigm to address this challenge. However, previous methods often fail to fully utilize their potential by discarding non-expert data. Our key insight is that even demonstrations that fall outside the expert distribution but outperform the learned policy can enhance policy performance. To utilize this potential, we propose a novel approach named imitation learning via meta-learning an action ranker (ILMAR). ILMAR implements weighted behavior cloning (weighted BC) on a limited set of expert demonstrations along with supplementary demonstrations. It utilizes the functional of the advantage function to selectively integrate knowledge from the supplementary demonstrations. To make more effective use of supplementary demonstrations, we introduce meta-goal in ILMAR to optimize the functional of the advantage function by explicitly minimizing the distance between the current policy and the expert policy. Comprehensive experiments using extensive tasks demonstrate that ILMAR significantly outperforms previous methods in handling suboptimal demonstrations. Code is available at https://github.com/F-GOD6/ILMAR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。