ICML2026
Second-Order Smooth Planning with Optimal-Transport Bellman Smoothing
Tuan Dam
摘要
Planning with a generative model aims to estimate state values using minimal oracle calls. For entropy-regularized MDPs, SmoothCruiser exploits the smoothness of the Bellman operator to achieve sample complexity, but its first-order Taylor approximation limits the rate. We develop a curvature--complexity theory showing that if a Bellman aggregator has Taylor remainder of order , the optimal oracle complexity exponent is ---recovering for and predicting for . To achieve , we introduce an entropic optimal-transport regularizer over action distributions. The resulting OT-smoothed Bellman operator admits a closed-form expression, explicit gradient policy, and Lipschitz Hessian. We derive an unbiased estimator of the quadratic Taylor term via cross-product debiasing, enabling a second-order SmoothCruiser with complexity. We further propose gap-dependent variants and provide a complexity analysis and show advantage of our method.