Second-Order Smooth Planning with Optimal-Transport Bellman Smoothing
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
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks