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

强化学习中的局部约束表示

Locally Constrained Representations in Reinforcement Learning

McGill University(麦吉尔大学);Mila-Quebec AI Institute(米拉-魁北克人工智能研究所);University of Massachusetts, Amherst(马萨诸塞大学阿默斯特分校);Dell Technologies, Research Office(戴尔科技公司研究办公室);École de technologie supérieure(高等技术学院);CIFAR AI Chair(加拿大高级研究所人工智能讲席)

Somjit Nath, Rushiv Arora, Samira Ebrahimi Kahou

AI总结:

提出局部约束表示方法,通过辅助损失使状态表示可由相邻状态预测,防止过度拟合价值损失,在连续控制任务中显著提升性能。

摘要:

强化学习(RL)的成功在很大程度上依赖于从环境观测中学习鲁棒表示的能力。在大多数情况下,仅由强化学习损失学习到的表示,会因价值函数的变化而在不同状态间产生巨大差异。然而,学习到的表示并不需要与当前任务高度特定相关。仅依赖RL目标可能导致表示在连续时间步之间变化剧烈。此外,由于RL损失具有变化的目标,学习到的表示将取决于当前价值/策略的好坏。因此,将表示与主任务解耦,不仅能让它们关注任务特定特征,还能关注环境动态。为此,我们提出了局部约束表示,其中辅助损失强制状态表示可由相邻状态的表示预测。这促使表示不仅由价值/策略学习驱动,还受到额外损失的约束,从而防止表示过度拟合价值损失。我们在多个已知基准上评估了所提出的方法,并观察到强劲的性能。特别是在连续控制任务中,我们的实验显示出显著的性能提升。

英文摘要:

The success of Reinforcement Learning (RL) heavily relies on the ability to learn robust representations from the observations of the environment. In most cases, the representations learned purely by the reinforcement learning loss can differ vastly across states depending on how the value functions change. However, the representations learned need not be very specific to the task at hand. Relying only on the RL objective may yield representations that vary greatly across successive time steps. In addition, since the RL loss has a changing target, the representations learned would depend on how good the current values/policies are. Thus, disentangling the representations from the main task would allow them to focus not only on the task-specific features but also the environment dynamics. To this end, we propose locally constrained representations, where an auxiliary loss forces the state representations to be predictable by the representations of the neighboring states. This encourages the representations to be driven not only by the value/policy learning but also by an additional loss that constrains the representations from over-fitting to the value loss. We evaluate the proposed method on several known benchmarks and observe strong performance. Especially in continuous control tasks, our experiments show a significant performance improvement.

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

↑