arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.14871cs.AIcs.LG
收藏

从噪声标签中学习鲁棒奖励机

Learning Robust Reward Machines from Noisy Labels

Imperial College London(伦敦帝国学院);University of Brescia(布雷西亚大学);Cardiff University(卡迪夫大学)

Roko Parac, Lorenzo Nodari, Leo Ardon, Daniel Furelos-Blanco, Federico Cerutti, Alessandra Russo

AI总结:

本文提出PROB-IRM方法,结合鲁棒归纳逻辑编程与贝叶斯后验置信度,从含噪声轨迹中学习奖励机,通过RM与策略交替学习及概率奖励塑形加速训练,使智能体性能接近手工RM的水平。

摘要:

本文提出PROB-IRM方法,用于为强化学习(RL)智能体从含噪声的执行轨迹中学习鲁棒奖励机(reward machines, RMs)。基于RM的强化学习的核心是利用有限状态机将智能体的任务分解为不同子任务。PROB-IRM采用对噪声样本具有鲁棒性的前沿归纳逻辑编程框架,借助贝叶斯后验置信度从含噪声轨迹中学习RM,从而确保对不一致数据的鲁棒性。RM学习与策略学习的交替是取得结果的关键:每当RL智能体生成一条被认为无法被当前RM接受的轨迹时,就学习新的RM。为加快RL智能体的训练速度,PROB-IRM采用概率化的奖励塑形公式,该公式使用从轨迹中推导得到的后验贝叶斯置信度。我们的实验分析表明,PROB-IRM能够从含噪声轨迹中学习(可能存在瑕疵的)RM,并利用这些RM训练RL智能体成功完成任务。尽管从含噪声轨迹中学习RM复杂度较高,但使用PROB-IRM训练的智能体性能与配备手工构建RM的智能体相当。

英文摘要:

This paper presents PROB-IRM, an approach that learns robust reward machines (RMs) for reinforcement learning (RL) agents from noisy execution traces. The key aspect of RM-driven RL is the exploitation of a finite-state machine that decomposes the agent's task into different subtasks. PROB-IRM uses a state-of-the-art inductive logic programming framework robust to noisy examples to learn RMs from noisy traces using the Bayesian posterior degree of beliefs, thus ensuring robustness against inconsistencies. Pivotal for the results is the interleaving between RM learning and policy learning: a new RM is learned whenever the RL agent generates a trace that is believed not to be accepted by the current RM. To speed up the training of the RL agent, PROB-IRM employs a probabilistic formulation of reward shaping that uses the posterior Bayesian beliefs derived from the traces. Our experimental analysis shows that PROB-IRM can learn (potentially imperfect) RMs from noisy traces and exploit them to train an RL agent to solve its tasks successfully. Despite the complexity of learning the RM from noisy traces, agents trained with PROB-IRM perform comparably to agents provided with handcrafted RMs.

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

↑