Second-Order Smooth Planning with Optimal-Transport Bellman Smoothing

arXiv:2609.06484 · cs.LG, cs.AI · Submitted 2026-09-06 · Read on arXiv

cs.LG, cs.AI

Submitted: 2026-09-06

Updated: 2026-09-06

Comments: Published at the International Conference on Machine Learning (ICML 2026)

License: http://creativecommons.org/licenses/by/4.0/

The gist: Planning with a generative model aims to estimate the value of a state using as few simulator calls as possible.

Terminology

Abstract

Planning with a generative model aims to estimate the value of a state using as few simulator calls as possible. SmoothCruiser achieves problem-independent complexity O(epsilon-4) by exploiting the smoothness of the entropy-regularized Bellman backup, but its estimator is only first-order. We show that the sample-complexity exponent of SmoothCruiser-type planners is governed by the order β of the local Taylor remainder, giving oracle complexity O(epsilon-(2+2/(β-1))): the first-order case β=2 recovers SmoothCruiser, while a second-order/cubic remainder β=3 yields O(epsilon-3). We reach this regime with an optimal-transport-smoothed Bellman backup over action distributions, which has a closed form, a policy gradient, and a Lipschitz Hessian, and whose quadratic correction admits an unbiased cross-product estimator. The resulting SecondOrderSmoothCruiser achieves O(epsilon-3) oracle complexity for fixed OT parameters, and we relate the OT, entropy-regularized, and unregularized objectives through explicit regularization-bias bounds.

Related papers