arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.07784cs.LGcs.AI

ProgAgent:一种具有进度感知奖励的持续强化学习代理

ProgAgent:A Continual RL Agent with Progress-Aware Rewards

University of California, San Diego(加州大学圣地亚哥分校);Texas A\&M University-Commerce(德克萨斯A&M大学-科摩斯)

Jinzhou Tan, Gabriel Adineera, Jinoh Kim

AI总结:

ProgAgent通过进度感知奖励机制,结合PPO与coreset回放和突触智能,有效解决持续学习中的遗忘问题,提升学习效率。

摘要:

我们提出了ProgAgent,一种持续强化学习(CRL)代理,它将进度感知奖励学习与一个高吞吐量、原生JAX架构系统统一起来。终身机器人学习面临着灾难性遗忘和奖励规格的高成本问题。ProgAgent通过从未标记的专家视频中推导出密集的、形状化的奖励,通过一个感知模型来估计任务进度,该模型在初始、当前和目标观察之间进行估计。我们理论上将这一过程解释为一个学习的状态势函数,从而提供稳健的指导,与专家行为一致。为了在在线探索中维持稳定性——其中会出现新的、分布外状态——我们引入了对抗性推回细化,以正则化奖励模型,抑制对非专家轨迹的过度自信预测,并对抗分布偏移。通过将此奖励机制嵌入到JIT编译的循环中,ProgAgent支持大规模并行展开和完全可微分的更新,使复杂的统一目标成为可能:它将PPO与coreset回放和突触智能相结合,以实现增强的稳定性-可塑性平衡。在ContinualBench和Meta-World基准测试中的评估突显了ProgAgent的优势:它显著减少了遗忘,提高了学习速度,并在视觉奖励学习(如Rank2Reward、TCN)和持续学习(如Coreset、SI)中优于关键基线,甚至超越了理想化的完美记忆代理。实际机器人试验进一步验证了其从嘈杂的少样本人类演示中获得复杂操作技能的能力。

英文摘要:

We present ProgAgent, a continual reinforcement learning (CRL) agent that unifies progress-aware reward learning with a high-throughput, JAX-native system architecture. Lifelong robotic learning grapples with catastrophic forgetting and the high cost of reward specification. ProgAgent tackles these by deriving dense, shaped rewards from unlabeled expert videos through a perceptual model that estimates task progress across initial, current, and goal observations. We theoretically interpret this as a learned state-potential function, delivering robust guidance in line with expert behaviors. To maintain stability amid online exploration - where novel, out-of-distribution states arise - we incorporate an adversarial push-back refinement that regularizes the reward model, curbing overconfident predictions on non-expert trajectories and countering distribution shift. By embedding this reward mechanism into a JIT-compiled loop, ProgAgent supports massively parallel rollouts and fully differentiable updates, rendering a sophisticated unified objective feasible: it merges PPO with coreset replay and synaptic intelligence for an enhanced stability-plasticity balance. Evaluations on ContinualBench and Meta-World benchmarks highlight ProgAgent's advantages: it markedly reduces forgetting, boosts learning speed, and outperforms key baselines in visual reward learning (e.g., Rank2Reward, TCN) and continual learning (e.g., Coreset, SI) - surpassing even an idealized perfect memory agent. Real-robot trials further validate its ability to acquire complex manipulation skills from noisy, few-shot human demonstrations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑