2408.14336

Total: 1

#1 Equivariant Reinforcement Learning under Partial Observability [PDF1] [Copy] [Kimi1] [REL]

Authors: Hai Nguyen, Andrea Baisero, David Klee, Dian Wang, Robert Platt, Christopher Amato

Incorporating inductive biases is a promising approach for tackling challenging robot learning domains with sample-efficient solutions. This paper identifies partially observable domains where symmetries can be a useful inductive bias for efficient learning. Specifically, by encoding the equivariance regarding specific group symmetries into the neural networks, our actor-critic reinforcement learning agents can reuse solutions in the past for related scenarios. Consequently, our equivariant agents outperform non-equivariant approaches significantly in terms of sample efficiency and final performance, demonstrated through experiments on a range of robotic tasks in simulation and real hardware.

Subjects: Robotics , Artificial Intelligence , Computer Vision and Pattern Recognition

Publish: 2024-08-26 15:07:01 UTC