Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
FERPO addresses the issue of unreliable policy updates in continuous control RL by avoiding direct action differentiation of the critic, which is often trained on return prediction. The proposed algorithm optimizes policies using a critic-derived target action distribution regularized by entropy and KL divergence, fitting the actor via a forward-KL objective estimated with SNIS. This approach promotes exploration by covering multiple high-value modes and achieves competitive performance and sample efficiency on benchmark tasks.
A novel policy optimization method bypasses critic action gradients entirely, achieving faster, more robust updates and improved exploration in continuous control.
Several state-of-the-art methods for online reinforcement learning in continuous control improve policies using action gradients of a learned critic. However, critics are typically trained to predict returns, and accurate value predictions do not necessarily yield accurate action derivatives, potentially leading to unreliable policy updates. We propose Forward Entropy-Regularized Policy Optimization (FERPO), an on-policy maximum entropy reinforcement learning algorithm that performs policy improvement using critic values without differentiating the critic with respect to actions. FERPO derives an optimal target action distribution from a policy-improvement objective regularized by entropy and Kullback-Leibler (KL) divergence. We then fit the actor to this target by minimizing a forward-KL objective, estimated using self-normalized importance sampling (SNIS) with actions drawn from the rollout policy. By limiting the target distribution's deviation from the rollout policy, the KL regularization helps keep these importance weights well behaved. In contrast to reverse-KL objectives, which can favor a subset of the target distribution's modes, the forward-KL objective encourages coverage of multiple high-value modes and thereby promotes exploration. Experiments and ablations on MuJoCo Playground and ManiSkill show competitive performance and sample-efficiency gains. Computational benchmarks also demonstrate faster actor updates than Relative Entropy Pathwise Policy Optimization (REPPO).