Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces PriPG-RL, a reinforcement learning framework for POMDPs where a planner agent with privileged state information guides a learning agent with only partial observations. They use an anytime-feasible MPC algorithm as the planner and propose Planner-to-Policy Soft Actor-Critic (P2P-SAC) to distill the planner's knowledge. Experiments in simulation and on a real quadruped robot demonstrate improved sample efficiency and policy performance compared to standard RL methods under partial observability.
Overcome partial observability in reinforcement learning by distilling knowledge from a privileged planner agent, achieving superior sample efficiency and real-world performance.
This paper addresses the problem of training a reinforcement learning (RL) policy under partial observability by exploiting a privileged, anytime-feasible planner agent available exclusively during training. We formalize this as a Partially Observable Markov Decision Process (POMDP) in which a planner agent with access to an approximate dynamical model and privileged state information guides a learning agent that observes only a lossy projection of the true state. To realize this framework, we introduce an anytime-feasible Model Predictive Control (MPC) algorithm that serves as the planner agent. For the learning agent, we propose Planner-to-Policy Soft Actor-Critic (P2P-SAC), a method that distills the planner agent's privileged knowledge to mitigate partial observability and thereby improve both sample efficiency and final policy performance. We support this framework with rigorous theoretical analysis. Finally, we validate our approach in simulation using NVIDIA Isaac Lab and successfully deploy it on a real-world Unitree Go2 quadruped navigating complex, obstacle-rich environments.