arXiv:2506.20307cs.LGcs.AI2025-06ICLR被引 2

用双重探索提升模仿学习,少样本也能超越专家表现。

Beyond-Expert Performance with Limited Demonstrations: Efficient Imitation Learning with Double Exploration

  • 引入双路探索:不确定性奖励优化策略,偏离演示轨迹则激发好奇心
  • 在少样本下实现超越专家性能,Atari与MuJoCo任务均优于现有方法
  • 适合追求高效模仿学习的算法研究者和机器人应用开发者

模仿学习是强化学习中的核心问题,目标是学习一个模仿专家行为的策略。现实中,由于状态空间复杂,仅凭少量示范难以准确学习专家策略。同时,为实现超越专家的表现,必须探索环境并收集数据。为此,我们提出一种新型模仿学习算法——双重探索模仿学习(ILDE),从两个方面实现探索:(1) 通过探索奖励机制,对高不确定性状态-动作对进行乐观策略优化,以加速收敛至专家策略;(2) 通过好奇心驱动,探索偏离示范轨迹的状态,以可能获得超越专家的性能。实验表明,ILDE在样本效率上优于当前最先进的模仿学习算法,并在较少示范条件下于Atari和MuJoCo任务中实现超越专家表现。我们还从理论上证明了ILDE是一种带有乐观探索的不确定性正则化策略优化方法,其累计遗憾随回合数亚线性增长。

原文摘要 · Abstract (English)

Imitation learning is a central problem in reinforcement learning where the goal is to learn a policy that mimics the expert's behavior. In practice, it is often challenging to learn the expert policy from a limited number of demonstrations accurately due to the complexity of the state space. Moreover, it is essential to explore the environment and collect data to achieve beyond-expert performance. To overcome these challenges, we propose a novel imitation learning algorithm called Imitation Learning with Double Exploration (ILDE), which implements exploration in two aspects: (1) optimistic policy optimization via an exploration bonus that rewards state-action pairs with high uncertainty to potentially improve the convergence to the expert policy, and (2) curiosity-driven exploration of the states that deviate from the demonstration trajectories to potentially yield beyond-expert performance. Empirically, we demonstrate that ILDE outperforms the state-of-the-art imitation learning algorithms in terms of sample efficiency and achieves beyond-expert performance on Atari and MuJoCo tasks with fewer demonstrations than in previous work. We also provide a theoretical justification of ILDE as an uncertainty-regularized policy optimization method with optimistic exploration, leading to a regret growing sublinearly in the number of episodes.

模仿学习探索策略强化学习少样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。