arXiv reader

Particle-based algorithm for stochastic optimal control

Sebastian Reich

arXiv
2311.06906v4 · PDF · 27 February 2024
Categories
math.OC, cs.NA, math.NA
Licence
CC BY 4.0 The site shows the text.

Abstract

The solution to a stochastic optimal control problem can be determined by computing the value function from a discretization of the associated Hamilton-Jacobi-Bellman equation. Alternatively, the problem can be reformulated in terms of a pair of forward-backward SDEs, which makes Monte-Carlo techniques applicable. More recently, the problem has also been viewed from the perspective of forward and reverse time SDEs and their associated Fokker-Planck equations. This approach is closely related to techniques used in diffusion-based generative models. Forward and reverse time formulations express the value function as the ratio of two probability density functions; one stemming from a forward McKean-Vlasov SDE and another one from a reverse McKean-Vlasov SDE. In this paper, we extend this approach to a more general class of stochastic optimal control problems and combine it with ensemble Kalman filter type and diffusion map approximation techniques in order to obtain efficient and robust particle-based algorithms.

Items 6

Remark

Remark 2.1

8P4D

The transformed HJB equation (?) simplifies to

-\partial_t v_t^\ast = b \cdot \nabla_x v_t^\ast + \frac{1}{2} \Sigma : D_x^2 v_t^\ast - c v_t^\ast

for the special case G = \sigma and R = I and (?) becomes linear in v_t^\ast. This well-known fact has been exploited in the numerical work of [MO22]. Furthermore, the transformed HJB equation arising from the further simplification c(x)\equiv 0 leads to the standard backward Kolmogorov equation [Pavliotis2016].

Lemma

Lemma 3.1

ZZZN

Given the forward evolution equation (?) and the HJB equation (?), the probability density defined by (?) satisfies the reverse time evolution equation

-\partial_t \tilde \pi_t = -\nabla_x \cdot \left(\tilde \pi_t \tilde b_t \right) - \tilde \pi_t \left(c - \alpha_t + \frac{1}{2}\|\sigma^{\rm T} \nabla_x \log v_t^\ast \|^2 - \frac{1}{2} \|R G^{\rm T} \nabla_x \log v^\ast_t\|^2_R - \zeta_t\right)

with terminal condition \tilde \pi_T = Z_T^{-1}\exp(-f) \bar \pi_T, where \zeta_t is an appropriate normalization constant.

Proof of Lemma 3.1

Since (?) and

\nabla_x \cdot (\Sigma \nabla_x v_t^\ast) = \nabla_x v_t^\ast \cdot (\nabla_x \cdot \Sigma) + \Sigma : D_x^2 v_t^\ast,

and assuming that (?) holds, it follows that

\begin{align} -\nabla_x \cdot (\tilde \pi_t \tilde b_t) &= \nabla_x \cdot \left(\tilde \pi_t \left( \bar b_t + \frac{1}{2} \Sigma \nabla_x \log v_t^\ast \right)\right)\\ &= \frac{v_t^\ast}{Z_t} \nabla_x \cdot (\bar \pi_t \bar b_t ) + \frac{\bar \pi_t}{Z_t} (\Sigma \nabla_x v_t^\ast) \cdot \bar b_t + \frac{1}{2Z_t} \nabla_x \cdot (\bar \pi_t \Sigma \nabla_x v_t^\ast)\\ &= \frac{v_t^\ast}{Z_t} \nabla_x \cdot (\bar \pi_t \bar b_t ) + \frac{\bar \pi_t}{Z_t} \left( \nabla_x v_t^\ast \cdot \left( b -\frac{1}{2} \nabla_x \cdot \Sigma \right) + \frac{1}{2} \nabla_x \cdot ( \Sigma \nabla_x v_t^\ast) \right)\\ &= \frac{v_t^\ast}{Z_t} \nabla_x \cdot \left( \bar \pi_t \bar b_t \right) + \frac{\bar \pi_t}{Z_t} \left( b \cdot \nabla_x v_t^\ast + \frac{1}{2} \Sigma : D_x^2 v_t^\ast \right). \end{align}

Hence it holds indeed that

\partial_t \tilde \pi_t = \frac{v_t^\ast}{Z_t} \partial_t \bar \pi_t + \frac{\bar \pi_t}{Z_t} \partial_t v_t^\ast - \frac{\tilde \pi}{Z_t} \frac{{\rm d}Z_t}{{\rm d}t}

for the partial time derivatives given by (3.1), (?), and (?), respectively. Furthermore, \tilde \pi_T = v_T^\ast \bar \pi_T/Z_T at final time and, hence, (?) holds for all times t\in (0,T].

Remark

Remark 3.1

H7KZ

While [MRO20] and [MO22] form the basis for our work, we mention the alternative approach put forward in [JTMM22], where v_t^\ast is viewed directly as an unnormalized probability density. This approach leads to the interpretation of the HJB equation (?) in terms of a nonlinear Fokker–Planck equation which in turn can be approximated using interacting particles and EnKF-type approximations as in our work. Only deterministic control problems are considered in [JTMM22]. Furthermore, it is not obvious whether the value function v_t^\ast is always normalizable with respect to the Lebesque measure on \mathbb{R}^{d_x}.

Lemma

Lemma 3.2

KLZB

The two diffusion processes defined by (?) satisfy \bar X_t \sim \bar \pi_t and \tilde X_t \sim \tilde \pi_t, respectively. Given \tilde \pi_t and \bar \pi_t, the optimal control law u_t^\ast(x) is provided by

u_t^\ast(x) = R G(x)^{\rm T} \left( \nabla_x \log \tilde \pi_t(x) - \nabla_x \log \bar \pi_t(x) \right).
Proof of Lemma 3.2

The lemma follows immediately from writing down the associated (nonlinear) Fokker–Planck equations for the two McKean–Vlasov SDEs (?). The stated formula for the optimal control follows from (?).

Source and licence

The text on this page is from arXiv:2311.06906v4 by Sebastian Reich, under the licence CC BY 4.0. The corpus made these changes: it read the items out of the LaTeX source, it expanded the macros of the paper, it put each proof after its statement, and it converted the text to HTML. The formulas are the TeX of the paper, and KaTeX renders them. A formula or a reference can look different from the PDF. The page is published under CC BY 4.0.

The record of each item is in analysis/objects/2311/2311.06906.jsonl.