Search papers, labs, and topics across Lattice.
This paper introduces multi-step proximal policy improvement (MPI) as a novel refinement mechanism in offline reinforcement learning that allows for controlled policy updates beyond the behavior distribution while maintaining reliability. By conceptualizing policies as a probability manifold and interpreting various offline actor objectives through a single proximal policy improvement step, the authors enable sequential re-centered updates that enhance performance on established benchmarks. Experiments demonstrate that MPI significantly improves upon strong offline baselines like TD3+BC and IQL across multiple tasks, highlighting its effectiveness in navigating the trade-off between exploration and reliability in offline settings.
Sequential refinements using MPI can enhance offline RL performance beyond traditional behavior constraints, outperforming established baselines.
Offline reinforcement learning (RL) must reconcile two competing requirements: policy updates should stay near dataset-supported actions to keep value estimates reliable, yet meaningful gains often require moving beyond the behavior distribution. We develop a geometric view of offline actor updates by modeling policies as a probability manifold endowed with a chosen metric geometry. Under this lens, a broad class of offline actor objectives can be interpreted as a single proximal policy improvement step (SPI), i.e., an implicit discretization of a manifold gradient flow induced by a critic-defined energy. Building on this insight, we propose multi-step proximal policy improvement (MPI), a plug-in refinement mechanism that composes sequential re-centered proximal steps. MPI enables controlled policy improvement beyond dataset support while retaining proximal control at each refinement. The framework accommodates multiple policy geometries and admits practical instantiations for deterministic and diagonal-Gaussian policies. Experiments on D4RL benchmarks show that small numbers of MPI refinements improve strong offline baselines, including TD3+BC, ReBRAC, and IQL, on many tasks. Focused diagnostics further distinguish re-centered refinement from fixed-objective update scheduling and characterize limitations under critic error.
