Skip to content

Trajectory-Regularized Stochastic Optimal Control via KL Divergence

Jul 2026 · arXiv.org · Vol abs/2607.22201 · 0 citations · 20 references
Computer Science Engineering

TL;DR

TRSOC is introduced, which augments standard stochastic optimal control with a Kullback--Leibler (KL) divergence between controlled and reference trajectory distributions, and the regularization parameter induces a trade-off between performance-driven and reference-preserving behavior.

Abstract

We introduce trajectory-regularized stochastic optimal control (TRSOC), which augments standard stochastic optimal control (SOC) with a Kullback--Leibler (KL) divergence between controlled and reference trajectory distributions. Using Girsanov's theorem, the trajectory KL reduces to a quadratic drift mismatch penalty, yielding a modified running cost that preserves the dynamic programming (DP) structure. We derive the corresponding Hamilton--Jacobi--Bellman (HJB) equation and characterize the optimal policy. In the linear-quadratic (LQ) setting, the formulation admits a closed-form solution with an augmented control cost. Experiments show that the regularization parameter induces a trade-off between performance-driven and reference-preserving behavior, including cases with reference dynamics learned from offline data.

View source

Similar papers

Preprint Aug 2026

Data-Driven optimal control via Koopman operators and Hamilton-Jacobi-Bellman equations

This paper presents a data-driven stable manifold (DD-SM) method, which integrates Koopman operator representation learning with the geometric stable manifold approach to Hamilton-Jacobi-Bellman (HJB) equations, enabling end-to-end optimal feedback control synthesis from raw trajectory data without prior knowledge of s...

Guuyuan Chen, Gao-Sheng Zhu · 0 citations
Preprint Aug 2026

Path Integral Value Matching for Linear Quadratic Stochastic Optimal Control

Empirical results demonstrate that PI-VM matches SOTA precision with an order-of-magnitude efficiency gain in low-dimensional settings, while effectively mitigating mode collapse in high-dimensional scenarios, and offers a scalable solution for solving complex SOC problems.

Bangyan Liao, Cheng-Lei Yu, Yuchen Yang et al. · 0 citations
Preprint Jul 2026

Model-Free Q-Learning for Infinite-Horizon Stochastic Linear Quadratic Problems with Regime Switching

On-policy and off-policy Q-learning algorithms that learn the optimal controller solely from online state trajectory data are developed, specifically developing on-policy and off-policy Q-learning algorithms that learn the optimal controller solely from online state trajectory data.

Xinyu Zhang, Na Li, Xun Li et al. · 0 citations
Preprint Aug 2026

Wasserstein Policy Gradient for Entropy-Regularized Linear-Quadratic Control

Wasserstein policy gradient (WPG) reduces exactly to a finite-dimensional ODE for the feedback gain and action covariance and it is proved that this ODE is globally well posed and converges exponentially from every admissible initialization.

Zhao Zhu, Rui Gao, Shuang Li · 0 citations
Preprint Aug 2026

Mean-covariance turnpikes in Wasserstein distributionally robust linear-quadratic control

A hybrid policy combining time-independent affine feedback with finite-horizon steering over a terminal layer is constructed, proving that its worst-case cost gap decays exponentially with the terminal-layer length uniformly in the horizon, which helps reducing the computation cost for long-horizon robust controls.

Yan-Zhi Wu, Zhengbiao Ji · 0 citations

Related blog posts

Microsoft Research Blog Oct 12, 2026

Simulating student learning for better AI tutoring

StudentSim is a research system for evaluating and improving AI tutoring tools before they are tested with learners. It models how individuals may engage with tutors, giving feedback to improve tutor performance across subjects like chess, writing, and mathematics. The post Simulating student learning for better AI tutoring appeared first on Microsoft Research.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.