arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.08356cs.LGcs.AIstat.ML

Goal-conditioned Batch Reinforcement Learning for Rotation Invariant Locomotion

University of California, San Diego(加州大学圣迭戈分校)

Aditi Mavalankar

英文摘要:

We propose a novel approach to learn goal-conditioned policies for locomotion in a batch RL setting. The batch data is collected by a policy that is not goal-conditioned. For the locomotion task, this translates to data collection using a policy learnt by the agent for walking straight in one direction, and using that data to learn a goal-conditioned policy that enables the agent to walk in any direction. The data collection policy used should be invariant to the direction the agent is facing i.e. regardless of its initial orientation, the agent should take the same actions to walk forward. We exploit this property to learn a goal-conditioned policy using two key ideas: (1) augmenting data by generating trajectories with the same actions in different directions, and (2) learning an encoder that enforces invariance between these rotated trajectories with a Siamese framework. We show that our approach outperforms existing RL algorithms on 3-D locomotion agents like Ant, Humanoid and Minitaur.

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

↑