DPO: A Differential and Pointwise Control Approach to Reinforcement Learning
Minh Nguyen, Chandrajit Bajaj
TL;DR
The paper addresses sample-inefficiency and lack of physical consistency in reinforcement learning for scientific computing by reframing RL as a continuous-time control problem via a differential dual and Hamiltonian structure. It introduces Differential Policy Optimization (DPO), a stagewise, pointwise update rule that learns a local trajectory operator, promoting trajectory-consistent learning aligned with system dynamics. The authors establish pointwise convergence guarantees and a regret bound of $O(K^{5/6})$, and demonstrate empirically that DPO outperforms standard baselines on surface modeling, grid-based modeling, and molecular dynamics under low data. This approach integrates physics priors into RL through a differential dual, enabling more data-efficient learning in physics-constrained environments with broad potential impact in scientific computing. The work also provides a foundation for future extensions to adaptive discretization and broader domains beyond physics-informed control.
Abstract
Reinforcement learning (RL) in continuous state-action spaces remains challenging in scientific computing due to poor sample efficiency and lack of pathwise physical consistency. We introduce Differential Reinforcement Learning (Differential RL), a novel framework that reformulates RL from a continuous-time control perspective via a differential dual formulation. This induces a Hamiltonian structure that embeds physics priors and ensures consistent trajectories without requiring explicit constraints. To implement Differential RL, we develop Differential Policy Optimization (DPO), a pointwise, stage-wise algorithm that refines local movement operators along the trajectory for improved sample efficiency and dynamic alignment. We establish pointwise convergence guarantees, a property not available in standard RL, and derive a competitive theoretical regret bound of $O(K^{5/6})$. Empirically, DPO outperforms standard RL baselines on representative scientific computing tasks, including surface modeling, grid control, and molecular dynamics, under low-data and physics-constrained conditions.
