June 09, 2025
We introduce a continuous policy-value iteration algorithm where the approximations of the value function of a stochastic control problem and the optimal control are simultaneously updated through Langevin-type dynamics. This framework applies to both the entropy-regularized relaxed control problems and the classical control problems, with infinite horizon. We establish policy improvement and demonstrate convergence to the optimal control under the monotonicity condition of the Hamiltonian. By utilizing Langevin-type stochastic differential equations for continuous updates along the policy iteration direction, our approach enables the use of distribution sampling and non-convex learning techniques in machine learning to optimize the value function and identify the optimal control simultaneously.
\(\mathbf{Keywords}\). Hamilton-Jacobi-Bellman equation; Reinforcement learning; Policy gradient ascent; Stochastic control; Relaxed control; Langevin dynamics.
2000 AMS Mathematics subject classification: 93E20, 93E35, 60H10
The standard approaches for solving stochastic optimal control problems rely on the dynamic programming principle (i.e., Hamilton-Jacobi-Bellman equations) bellman1954theory?, bellman1966dynamic? or the maximum principle boltyanski1960maximum?, both of which involve highly non-linear partial differential equations for general reward/cost functions and state dynamics, and closed-form solutions are rarely obtained. These equations can either be solved numerically, using methods ranging from the classical finite difference approach BS1991? to deep learning methods HJE2018?, or their solutions can be approximated in closed form GW2020?, accompanied by error analysis based on duality arguments.
More recently, inspired by the (stochastic) gradient descent method for minimizing a given non-convex function, policy iteration algorithms for stochastic control problems have gained significant attention puterman1979convergence?, puterman1981convergence?, bertsekas2015value?, jacka2017policy?, kerimkulov2020exponential?, kerimkulov2021modified?. In particular, in the relaxed control framework, where controls are randomized at every time and state FN1984?, ZHou1992? and entropy regularization is incorporated into the cost functional, policy improvements are seen through the lens of reinforcement learning (RL) wang2020reinforcement?. In the literature, reinforcement learning sutton1998reinforcement?, bertsekas2012dynamic?, szepesvari2022algorithms? and Q-learning bertsekas2011dynamic? are traditionally formulated in a discrete time setting. wang2020reinforcement? extends the discussion to a continuous-time, linear-quadratic setting with relaxed control framework. This advancement significantly enhances the applicability of RL to financial mathematics dai2023learning?, denkert2025control?, chau2024continuous? and stochastic control tran2024policy?, tang2024regret?, possamai2024policy?, dianetti2024exploratory?, zhao2023policy?, vsivska2024gradient?, and huang2022convergence? recently extends this approach to a broader class of control problems.
In most of the above literature, although state variables follow continuous-time dynamics, the policy iterations are still updated discretely along the iteration direction. Thus the mathematical argument for the convergence of the approximated value and policy to the optimal ones is highly delicate, once it is beyond the linear-quadratic case in wang2020reinforcement?, e.g., the approach adopted in huang2022convergence? can only deal with the case where the diffusion coefficients of the state dynamics are not controlled. Furthermore, at each step of the policy iteration one has to evaluate the associated value function, either by solving the associated PDE or approximating it by learning algorithms, posing significant challenges, particularly in high-dimensional problems. ma2024convergence2?, ma2024convergence? recently proposed a new policy improvement algorithm that, while being discrete in nature, removes the necessity of solving the value function at each iteration, thereby enhancing both the efficiency and applicability of these methods. Furthermore, gradient flow and actor-critic strategy are further studied in this direction jia2022policy?, zhou2024solving?, zhou2023policy?, while the gradient flow does not directly provide a gradient descent algorithm for policy improvements.
To the best of our knowledge, we are the first to introduce iteration-continuous dynamics for both value and policy approximations, alongside continuous-time dynamics for the state variable in the relaxed control framework, where both the drift and diffusion coefficients are controlled. Also, unlike the extant iteration methods for the relaxed control problem, our approach extends naturally to classical stochastic control problems, as discussed in Section 4. For relaxed control problems, our algorithm establishes a connection to distribution sampling problems through continuous-time mean-field Langevin dynamics chen2022uniform?, chizat2022mean?, hu2021mean?, bayraktar2024exponential?. In the case of classical stochastic control problems, our approach builds a bridge between solving these problems and applying non-convex optimization techniques inspired by annealing-based continuous-time Langevin dynamics MCMC algorithms kirkpatrick1983optimization?, vcerny1985thermodynamical?, geman1986diffusions?, gelfand1990sampling?, tang2021simulated?, feng2024fisher?. By incorporating our policy-continuous dynamics, we extend the policy improvement algorithm to policy gradient ascent/descent methods, which have the potential to incorporate stochastic gradient ascent/descent algorithms welling2011bayesian?, zhang2017hitting?, raginsky2017non?, dalalyan2020sampling? for high-dimensional problems. Compared to the continuous dynamics with gradient flow in vsivska2024gradient?, ZZH2025?, we only update the policy locally and infinitesimally at every step in the iteration, instead of its whole path at every point in time. Furthermore, we do not need to keep track of the dynamics of the state under the new strategy, nor do we use the stochastic maximum principle, thus avoiding the need to solve the forward and backward system of stochastic differential equations as in vsivska2024gradient?. We also point out that one of the key features of our continuous policy dynamics is its inhomogeneous nature, meaning that the objective function in the Langevin dynamics is not fixed, but varies along the iteration. It leads to an interesting mathematical problem on the convergence of general inhomogeneous diffusion, which we leave for future studies.
This paper is organized as follows. In Section 2, we define the relaxed control problem and its associated HJB equation. In Section 3, we propose continuous iteration dynamics for both the value function and the optimal control which informs the distribution strategy and show the policy improvement result. In Section 4, we propose the continuous iteration for the classical control problem and show the policy improvement under these dynamics. In Section 6, we prove the exponential convergence rate of the proposed continuous dynamics for both relaxed and classical control problems. An example is provided in Section 7.
Consider an infinite horizon relaxed control problem, where at every time \(t\geq 0\), the agent randomizes the strategy \(u\in\mathbb{R}^n\) according to a distribution density \(\pi_{t}\), which controls both the drift and the diffusion of the state variable \(X\in \mathbb{R}^d\), so that it follows the exploratory dynamics \[dX^\pi_t = \tilde{b}(X^\pi_t,\pi_t)\,dt + \tilde{\sigma}(X^\pi_t,\pi_t)\,dW_t, \quad X_0 = x,\label{eq:32X-dyn}\tag{1}\] where \(\tilde{b} (x,\pi) = \int_{\mathbb{R}^n} b(x,u)\pi(u)du\) and \(\tilde{\sigma}(x,\pi) = \left(\int_{\mathbb{R}^n} \sigma\sigma^{\top}(x,u)\pi(u)du\right)^{1/2}\), with \(b: \mathbb{R}^d\times \mathbb{R}^n \rightarrow \mathbb{R}^d\), \(\sigma: \mathbb{R}^d\times \mathbb{R}^n \rightarrow \mathbb{R}^{d\times k}\) satisfying Assumption 1 below, and \(W\) is a \(d\)-dimensional Brownian motion. The agent’s goal is to maximize the expected running payoff \(f:\mathbb{R}^d\times \mathbb{R}^n \rightarrow \mathbb{R}\), regularized by entropy of \(\pi\), i.e., \(V(x) = \sup_{\pi\in\mathcal{A}(x)} J(x,\pi)\), where \[\label{definition:J} J(x,\pi)=\mathbb{E}\!\left[\int_0^\infty e^{-\beta t}\!\left(\tilde{f}(X^\pi_t,\pi_t) -\lambda\!\int_{\mathbb{R}^n}\pi_t(u)\ln\pi_t(u)\,du\right)dt\right],\tag{2}\] with \(\tilde{f}(x,\pi) = \int_{\mathbb{R}^n} f(x,u)\pi(u)du\) the exploratory reward (cf. wang2020reinforcement? for the motivation of this model and its connection with reinforcement learning). The admissible set \(\mathcal{A}(x)\) of the relaxed control \(\pi\) is defined as follows.
Definition 1. For every initial state \(x\in\mathbb{R}^d\), the admissible control set \(\mathcal{A}(x)\) is the collection of \(\pi = \{\pi_t, t\geq 0\}\) such that
\(\pi_t \in \mathcal{P}(\mathbb{R}^n)\) for each \(t\geq 0\), where \(\mathcal{P}(\mathbb{R}^n)\) is the space of absolutely continuous probability distributions on \(\mathbb{R}^n\) (identified with their densities).
\(\mathbb{E}\left[\int_0^T \left(|\tilde{b}(X^\pi_t,\pi_t)|^2 + \mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top(X^\pi_t,\pi_t)]\right) dt \right]<\infty\) for each \(T > 0\).
\(J(x,\pi) < \infty\).
We also make the following assumptions on model parameters.
Assumption 1.
\(f\), \(b\) and \(\sigma\sigma^\top\) are bounded and \(C^2\) in both \(x\) and \(u\), and all the first-order and second-order derivatives are uniformly bounded;
\(\sigma\sigma^\top\) is uniformly positive definite, i.e.\(\sigma\sigma^\top(x,u)\succeq c\,I\) for some constant \(c>0\);
\(V(x) < \infty\).
Due to the Markovian and infinite horizon setting, the value function is time-independent and is expected to solve the corresponding Hamilton-Jacobi-Bellman (HJB) equation \[\begin{align} &-\beta v(x) + \sup_{\pi\in\mathcal{P}(\mathbb{R}^n)}\mathbb{E}_{u\sim\pi}\left[f(x,u) - \lambda \ln \pi(u) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u):v_{xx}(x)] + b^\top(x,u)v_{x}(x)\right]\nonumber\\ =& -\beta v(x) +\sup_{\pi\in\mathcal{P}(\mathbb{R}^n)} \int_{\mathbb{R}^n} \left(\mathbf{H}(x,u,v_x,v_{xx}) -\lambda \ln\pi(u)\right)\pi(u)du =0,\nonumber \end{align}\] where \(\mathbf{H}(x,u,v_x,v_{xx}) = f(x,u) + b^\top(x,u)v_x(x)+ \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u):v_{xx}(x)]\), and \(v_x\), \(v_{xx}\) are the gradient and Hessian of \(v\) with respect to \(x\), respectively. The first-order condition gives \[\hat{\pi}(x,u) = \frac{\exp\left(\frac{1}{\lambda}\mathbf{H}(x,u,v_x,v_{xx})\right)}{\int_{\mathbb{R}^n} \exp\left(\frac{1}{\lambda}\mathbf{H}(x,u',v_x,v_{xx})\right)du'}. \nonumber\] If the optimal value function \(\hat{v} = V\) is known, then the optimal distribution \(\hat{\pi}\) (by plugging \(\hat{v}\) in the above first order condition) is the invariant distribution of (for a given \(x\) and \(n\)-dimensional Brownian motion \(B\) independent of \(W\)) the following diffusion, \[\label{eq:u-dynamics} du^x_\tau = \nabla_u \mathbf{H} (x,u^x_\tau,\hat{v}_x,\hat{v}_{xx})d\tau+ \sqrt{2\lambda}dB_\tau,\tag{3}\] provided that \(\int_{\mathbb{R}^n} \exp\left(\frac{1}{\lambda}\mathbf{H}(x,u,\hat{v}_x,\hat{v}_{xx})\right)du <\infty\)geman1986diffusions?, CHS1987?, roberts1996exponential?. In what follows, the invariant distribution refers to the density of the invariant measure of the diffusion process; being invariant means that this density solves the stationary Fokker–Planck equation \(\mathcal{L}^{*} \hat{\pi} = 0\), where \(\mathcal{L}^{*}\) is the formal \(L^2\)-adjoint of the infinitesimal generator \(\mathcal{L}\) for 3 . However, for general model parameters, the HJB equation does not have a closed-form solution and the optimal value function \(\hat{v}\) is not known a priori. In this paper, instead of solving the HJB equation corresponding to the optimal solution directly TZZ2022?, we propose a method that combines reinforcement learning and stochastic gradient descent from deep learning techniques, in which the optimal policy and value function are learned simultaneously through coupled continuous-time dynamics. It avoids evaluating the value function corresponding to the updated policy at each step of the iteration, e.g., by solving the associated partial differential equation (PDE) or approximating it with a neural network, as proposed in wang2020reinforcement?, where the iteration is discrete in nature.
Given \(\pi^x_0 \in \mathcal{P}(\mathbb{R}^n)\) for each \(x\in \mathbb{R}^d\), consider \((v^0,\rho^0)\) as the starting point of the iteration algorithm, where \(\rho^0 = \{\rho^0_t = \pi_0, 0\leq t<\infty\}\) and \(v^0(x) = J(x,\rho^0)\). The policy iteration follows the diffusion \[du^x_\tau =\nabla_u \mathbf{H}(x,u^x_\tau,v^\tau_x(0,x),v^\tau_{xx}(0,x))d\tau + \sqrt{2\lambda}dB_\tau, \quad u_0 \sim \pi_0,\label{eq:DynU}\tag{4}\] where \(v^\tau\) is the result of the value iteration as described later in this section, and is motivated by the above observation that the optimal randomized strategy is the invariant distribution of 3 . Notice that if \(v_x\) and \(v_{xx}\) are not updated throughout the policy iteration, then given \(v\), as \(\tau\rightarrow \infty\), the distribution of \(u_\tau\) converges to \(\pi_{\infty} = \frac{\exp\left(\frac{1}{\lambda}\mathbf{H}(x,u,v_x,v_{xx})\right)}{\int_{\mathbb{R}^n} \exp\left(\frac{1}{\lambda}\mathbf{H}(x,u',v_x,v_{xx})\right)du'}\), which matches the policy iteration method proposed in wang2020reinforcement?, huang2022convergence?. Thus, their method is equivalent to only updating the value function at \(\tau = \infty\) and starting another sequence of policy iteration given the new value function, without updating it until the policy becomes stationary, which explains its “discrete” nature. From this perspective, it can be regarded as a special case of the method we propose in this paper, which updates the value function much more frequently, simultaneously with the policy. For each \(\tau\), \(v^\tau = J(x,\rho^\tau)\) for a potentially time-inhomogeneous strategy \(\rho^\tau\), and thus may be a function of time (we will show that the strategy will become time-homogeneous as \(\tau\rightarrow \infty\), as expected in the optimal case). We update the value function by updating the corresponding strategy \(\rho^{\tau}\), using the distribution of \(u_s\) for \(0\leq s\leq \tau\). Notice that starting from \(\pi^x_0\in \mathcal{P}(\mathbb{R}^n)\), the density function of \(u^x_\tau\) depend on \(x\), and thus should be denoted as \(\pi^x_\tau\). In the following, since we will evaluate this density function at different state, we denote it as \(\pi_\tau(x)\), to emphasize its dependence on \(x\), while omitting of the argument \(u\) for the ease of notation, unless we need to write the argument \(u\) explicitly, for example, in an integral with respect to the density function, to avoid ambiguity.
Definition 2. For each \(\tau\geq 0\), given \(\{\pi_s(x),0\leq s\leq \tau\}\) (the distribution of \(u_s\) for \(0\leq s\leq \tau\)) for every \(x \in \mathbb{R}^d\), define \(\boldsymbol{\rho}^\tau = \{\rho^\tau_t(X^\tau_t)\}_{0\leq t<\infty}\) as \[\rho^\tau_t(x) = \begin{cases} \pi_{\tau-t}(x), &0\leq t \leq \tau;\\ \pi_0(x), & t > \tau. \end{cases}\] Denote the corresponding state process as \(X^{\tau}\), satisfying \[dX^\tau_t = \tilde{b}(X^\tau_t,\rho^\tau_t(X^\tau_t))dt +\tilde{\sigma}(X^\tau_t,\rho^\tau_t(X^\tau_t))dW_t,\nonumber\] and the corresponding value function \[\begin{align} v^{\tau}(t,x) = \mathbb{E}\left[\int_t^\infty e^{-\beta(s-t)}\left(\tilde{f}(X^\tau_s,\rho^\tau_s(X^\tau_s)) - \lambda\int_{\mathbb{R}^n}\rho^\tau_s(X^\tau_s,u)\ln\rho^\tau_s(X^\tau_s,u)du\right)ds\,\Big|\,X^\tau_t = x\right],\label{definition:v-tau} \end{align}\qquad{(1)}\] i.e., \(v^\tau(0,x) = J(x,\rho^\tau)\).
By the above definition, as \(\tau\) increases and new distributions \(\pi_\tau\) are generated from the policy iteration 3 , we update the strategy used in generating the value function \(v^\tau\) from the immediate previous step by using the newest \(\pi_\tau\) at time \(t=0\), and pushing the previous sequence of strategies (as a function of time) into the future. As we show in Proposition 1, \(\boldsymbol{\rho}^\tau\) becomes better as \(\tau\) increases, in the sense that the policy improvement property holds. Furthermore, as the iteration proceeds, the initial suboptimal strategies are pushed further into the future and because of the positive discounting factor, become less relevant. Finally, Propositions 2 and 3 show that \(\pi_\tau\) converges to the optimal strategy, and \(v^\tau\) also becomes optimal. Also, at every \(\tau\), we only make infinitesimal changes to \(\boldsymbol{\rho}^\tau\), so the changes in the corresponding value \(v^\tau\) are infinitesimal, which allows us to describe \(v^\tau\) and its derivatives by simple continuous dynamics in Lemma 1. Notice that in the above policy iteration, we only use \(v^\tau(0,x)\) and its derivatives in \(x\), thus in the following discussion, we focus on the continuous dynamics which update \(v^\tau(0,x)\) along the \(\tau\)-direction instead of the \(t\)-direction. In particular, we drop the time argument “\(0\)” for the ease of notation (i.e., \(v^\tau(x)\) stands for \(v^\tau(0,x)\)) unless \(v^\tau(t,x)\) for general \(t\) is discussed. Similarly, while the distribution of \(u^x_\tau\) should be denoted as \(\pi_\tau(x;u)\), the index \(x\) and argument \(u\) are dropped in the following discussion if no ambiguity arises. Also in the following discussion, \(\tau\) always denotes the direction of algorithm iteration, i.e., we are updating the distribution of the control by letting \(u\) evolve according to a diffusion. In this dynamics, \(x\) (the state) is fixed, and in order to update the policy for all states, we are supposed to run this diffusion for every \(x\). Finally, for ease of notation, we follow the convention that for any process \(z\) in the following discussion, \(z_t = z_0\) for \(t<0\). As a sanity check, for a model with \(n=d=1\) and a cost function \(f\) independent of state \(x\), the optimization problem is reduced to a static problem to maximize \(f(u)\pi(u) - \lambda\pi(u)\ln\pi(u)\), and the optimal solution is \(\hat{\pi} = \frac{e^{f(u)/\lambda}}{\int_{\mathbb{R}} e^{f(u')/\lambda}du'}\) and the optimal value function is independent of \(x\). Also for any state-independent and time-homogeneous strategy \(\pi(u)\), the corresponding value \[\begin{align} v^\pi(x) = \int_0^\infty e^{-\beta t} \int_{\mathbb{R}}\left(f(u)\pi(u) - \lambda \pi(u)\ln\pi(u)\right)du\,dt, \nonumber \end{align}\] is independent of \(x\). If we start with such a distribution of \(u_0\), and follow the state-independent iteration \[\begin{align} du_\tau =& \nabla f(u_\tau)d\tau + \sqrt{2\lambda}dB_\tau,\nonumber \end{align}\] the invariant distribution of \(u_\tau\) is \(\pi^*(u) = \frac{e^{f(u)/\lambda}}{\int_{\mathbb{R}} e^{f(u')/\lambda}du'}\), which equals the optimal strategy. For more general cases, state-specific policy iterations are needed. We denote the Hamiltonian evaluated at the round \(\tau\) of the value iteration as \[\begin{align} \label{H95tau} \mathbf{H}^\tau :=\mathbf{H}^\tau(x,u)&=\mathbf{H}(x,u,v^\tau_x,v^\tau_{xx}) \nonumber \\ &= f(x,u) + b^\top(x,u)v^{\tau}_x(x)+ \frac{1}{2}\mathrm{Tr}[\sigma\sigma^{\top}(x,u):v^{\tau}_{xx}(x)], \end{align}\tag{5}\] and we make the following assumptions:
Assumption 2. \(\pi_0\), the initial distribution density of \(u\), is continuous in \(x\). For each \(\tau \geq 0\) and sufficiently small \(h>0\),
\(\mathbf{\rho}^\tau \in\mathcal{A}(x)\) and \(v^\tau \in C^{1,4}\).
\(\mathbb{E}\left[\int_0^{\tau +h} \| v^{\tau}_x(s,X^{\tau+h}_s)\|^2ds\right]< \infty\).
\(\mathbb{E}\left[\|v^{\tau-s+l}_x(X^{\tau+h}_s)\|^2 + \|v^{\tau-s+l}_{xx}(X^{\tau+h}_s)\|^2+\|\nabla_u\ln \pi_{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})\|^2\right]\) is bounded in \(0\leq l \leq h\), for every \(0\leq s\leq \tau+h\).
\(\mathbb{E}\left[\left|\int_{\mathbb{R}^n} \left(\mathbf{H}(\cdot,u,v^\tau_x,v^\tau_{xx}) - \lambda\ln\pi_{\tau+h-s}\right)\pi_{\tau+h-s}(x,u)du -\beta v^{\tau}\right|(X^{\tau+h}_s)\right]\) is bouned in \(0\leq s\leq h\).
Given \(x\), \(\int_{\mathbb{R}^n} \pi_\tau(x,u)\ln\pi_\tau(x,u)du\) is uniformly bounded in \(\tau\).
We first show the following policy improvement result.
Proposition 1. For \(0<\tau_1 < \tau_2\), \(v^{\tau_1}(t,x)\leq v^{\tau_2}(t,x)\).
Proof. We discuss the case of \(t =0\), while the following argument works for any \(t\geq 0\). By Itô’s formula, \(v^\tau\) satisfies (for \(h>0\)) \[\begin{align} &e^{-\beta t}v^{\tau}(t,X^{\tau+h}_t) - v^{\tau}(0,x)\nonumber\\ =& \int_0^{t} e^{-\beta s} \left(v^{\tau}_t -\beta v^{\tau} + \tilde{b}^\top(X^{\tau+h}_s,\pi_{\tau+h-s}(X^{\tau+h}_s))v^{\tau}_x(s,X^{\tau+h}_s)\right.\nonumber\\ &\left.+ \frac{1}{2}\mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top(X^{\tau+h}_s,\pi_{\tau+h-s}(X^{\tau+h}_s)):v^{\tau}_{xx}(s,X^{\tau+h}_s)]\right)ds\nonumber\\ &+ \int_0^{t} e^{-\beta s} \tilde{\sigma}^\top(X^{\tau+h}_s,\pi_{\tau+h-s}(X^{\tau+h}_s))v^{\tau}_x(s,X^{\tau+h}_s)dW_s.\nonumber \end{align}\] Since \(v^{\tau}_x(s,X^{\tau+h}_s)\) is square integrable for sufficiently small \(h\) by Assumption 2 and \(\tilde{\sigma}\) is bounded, the integral with respect to \(W\) is a martingale. Letting \(t=\tau+h\) and taking expectation on both sides of the above equation (in the following discussion, we summarize the arguments of all the functions inside an integral at the end of the integrand for the ease of notation, unless ambiguity arises), \[\begin{align} &v^{\tau}(0,x) \nonumber \\ =& \mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s} \left(-v^{\tau}_t +\beta v^{\tau} - \tilde{b}^\top(\cdot,\pi_{\tau+h-s})v^{\tau}_x - \frac{1}{2}\mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top(\cdot,\pi_{\tau+h-s}):v^{\tau}_{xx}]\right)(s,X^{\tau+h}_s)ds\right.\nonumber\\ &\left.+e^{-\beta(\tau+h)}v^{\tau}(\tau+h,X^{\tau+h}_{\tau+h})\right].\label{eq:v-tau-zero-1} \end{align}\tag{6}\] On the other hand, from the definition of \(v^{\tau}\) in Definition ?? , \[\begin{align} v^{\tau}(t,X^\tau_t) =& \mathbb{E}\left[\int_t^{t+h} e^{-\beta(s-t)}\left(\tilde{f}(\cdot,\pi_{\tau-s}) - \lambda\int_{\mathbb{R}^n}\pi_{\tau-s}\ln\pi_{\tau-s}(\cdot,u)du\right)(X^\tau_s)ds\right.\nonumber\\ +& \left. e^{-\beta h}v^\tau(t+h,X^\tau_{t+h})\,\Big|\,\mathcal{F}^W_t\right], \end{align}\] where \(\{\mathcal{F}^W_t\}_{t\geq 0}\) is the filtration generated by \(W\). Thus \[\begin{align} &e^{-\beta t}v^{\tau}(t,X^\tau_t) + \int_0^t e^{-\beta s}\left(\tilde{f}(\cdot,\pi_{\tau-s}) - \lambda\int_{\mathbb{R}^n}\pi_{\tau-s}\ln\pi_{\tau-s}(\cdot,u)du\right)(X^\tau_s)ds\nonumber \\ =& \mathbb{E}\left[\int_0^{t+h} e^{-\beta s}\left(\tilde{f}(\cdot,\pi_{\tau-s}) - \lambda\int_{\mathbb{R}^n}\pi_{\tau-s}\ln\pi_{\tau-s}(\cdot,u)du\right)(X^\tau_s)ds\right.\nonumber\\ &\left.+ e^{-\beta (t+h)}v^\tau(t+h,X^\tau_{t+h})\,\Big|\,\mathcal{F}^W_t\right]. \end{align}\] In other words, \[M_t = e^{-\beta t}v^{\tau}(t,X^\tau_t) + \int_0^t e^{-\beta s}\left(\tilde{f}(\cdot,\pi_{\tau-s}) - \lambda\int_{\mathbb{R}^n}\pi_{\tau-s}\ln\pi_{\tau-s}(\cdot,u)du\right)(X^\tau_s)ds,\] is a martingale. Based on the smoothness assumption on \(v^\tau\) in Assumption 2, by standard arguments, \(v^{\tau}(t,x)\) satisfies, \[\begin{align} &v^{\tau}_t(t,x) -\beta v^{\tau}(t,x) +\tilde{b}^\top(\cdot,\pi_{\tau-t})v^{\tau}_x(t,x) + \frac{1}{2}\mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top(\cdot,\pi_{\tau-t}):v^{\tau}_{xx}(t,x)] \nonumber\\ &+ \int_{\mathbb{R}^n}\left(f- \lambda \ln\pi_{\tau-t}\right)\pi_{\tau-t}(x,u)du = 0.\label{v-pde} \end{align}\tag{7}\] Substituting this into 6 , \[\begin{align} &v^{\tau}(0,x) \nonumber\\ =& \mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \left(\tilde{b}(\cdot,\pi_{\tau-s}) - \tilde{b}(\cdot,\pi_{\tau+h-s})\right)^\top v^{\tau}_x(s,X^{\tau+h}_s)ds\right]\nonumber\\ &+\frac{1}{2}\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \mathrm{Tr}[\left(\tilde{\sigma}\tilde{\sigma}^\top(\cdot,\pi_{\tau-s}) - \tilde{\sigma}\tilde{\sigma}^\top(\cdot,\pi_{\tau+h-s})\right):v^{\tau}_{xx}(s,X^{\tau+h}_s)]ds\right]\nonumber\\ &+ \mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\left( \int_{\mathbb{R}^n}\left(f(\cdot,u) - \lambda \ln\pi_{\tau-s}\right)\pi_{\tau-s}(X^{\tau+h}_s,u)du\right) ds\right]\nonumber\\ & + \mathbb{E}\left[e^{-\beta(\tau +h)}v^{\tau}(\tau +h,X^{\tau+h}_{\tau+h})\right].\label{eq:v-tau-zero-2} \end{align}\tag{8}\] Notice that \(\rho^\tau_t = \pi_0\) for \(t\geq \tau+h\), thus starting from \(t=\tau+h\) and initial state \(x\), adopting strategy \(\rho^\tau\), the agent obtains the value \[\begin{align} v^\tau(\tau+h,x) =& \mathbb{E}\left[\int_{\tau+h}^\infty e^{-\beta (t-\tau-h)}\int_{\mathbb{R}^n}\left(f(X^{\tau}_t,u)- \lambda \ln\rho^\tau_t(u)\right)\rho^\tau_t(u)du\,dt\,\Big|\,X^\tau_{\tau+h} = x\right]\\ =&\mathbb{E}\left[\int_{\tau+h}^\infty e^{-\beta (t-\tau-h)} \int_{\mathbb{R}^n}\left(f(X^\tau_t,u)- \lambda\ln\pi_0(u)\right)\pi_0(u)du\,dt\,\Big|\,X^\tau_{\tau+h}=x\right]\\ =& \mathbb{E}\left[\int_{0}^\infty e^{-\beta t}\int_{\mathbb{R}^n}\left(f(X^0_t,u)- \lambda \ln\pi_0(u)\right)\pi_0(u)du\,dt\,\Big|\,X^0_0 = x\right]\\ =&v^0(0,x). \end{align}\] A similar argument shows that \(v^0(0,x) = v^{\tau+h}(\tau+h,x)\). Thus \(v^{\tau}(\tau +h,X^{\tau+h}_{\tau+h}) = v^{\tau+h}(\tau +h,X^{\tau+h}_{\tau+h})\). Plugging this into 8 , also using the fact that \[\begin{align} &\mathbb{E}\left[e^{-\beta(\tau +h)}v^{\tau+h}(\tau +h,X^{\tau+h}_{\tau+h})\right] \nonumber\\ =& v^{\tau+h}(0,x) -\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\left(\int_{\mathbb{R}^n} \left(f - \lambda\ln\pi_{\tau+h-s}\right)\pi_{\tau+h-s}(X^{\tau+h}_s,u)du\right)ds \right], \end{align}\] the following holds \[\begin{align} v^\tau(0,x)=&v^{\tau+h}(0,x)+\mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\left(\tilde{f}\left(\cdot,\pi_{\tau-s}\right)-\tilde{f}\left(\cdot,\pi_{\tau+h-s}\right)\right)(X^{\tau+h}_s)ds\right]\nonumber\\ &+\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \left(\tilde{b}(\cdot,\pi_{\tau-s}) - \tilde{b}(\cdot,\pi_{\tau+h-s})\right)^\top v^{\tau-s}_x(X^{\tau+h}_s) ds\right]\nonumber\\ &+\frac{1}{2}\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \mathrm{Tr}[\left(\tilde{\sigma}\tilde{\sigma}^\top(\cdot,\pi_{\tau-s}) - \tilde{\sigma}\tilde{\sigma}^\top(\cdot,\pi_{\tau+h-s})\right):v^{\tau-s}_{xx}(X^{\tau+h}_s)]ds\right]\nonumber\\ &+ \mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\left( \int_{\mathbb{R}^n}\lambda\left(\pi_{\tau+h-s}\ln\pi_{\tau+h-s} - \pi_{\tau-s} \ln\pi_{\tau-s}\right)(X^{\tau+h}_s,u)du \right)ds\right], \end{align}\] where we also used fact that \(v^{\tau}(s,x) = v^{\tau-s}(0,x) = v^{\tau-s}(x)\) by our convention.
Since \(\pi_{\tau}\) is the distribution density function of \(u_\tau\), and with deterministic coefficients in 3 , \(u\) is independent of the Brownian motion \(W\), the above can be written as (with a slight abuse of notation where the expectation is taken in a larger probability space which supports the two independent Brownian motions \(B\) and \(W\)) \[\begin{align} \label{eq:v-difference} &v^{\tau}(0,x) - v^{\tau+h}(0,x)\nonumber\\ =&\mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\mathbb{E}\left[f(X^{\tau+h}_s,u_{\tau-s})-f(X^{\tau+h}_s,u_{\tau+h-s})\,\big|\,\mathcal{F}^W_s\right]ds\right]\nonumber\\ &+\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \mathbb{E}\left[\left(b(\cdot,u_{\tau-s}) - b(\cdot,u_{\tau+h-s})\right)^\top v^{\tau-s}_x(X^{\tau+h}_s)\,\big|\,\mathcal{F}^W_s\right] ds\right]\nonumber\\ &+\frac{1}{2}\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \mathbb{E}\left[\mathrm{Tr}[\left(\sigma\sigma^\top(\cdot,u_{\tau-s}) - \sigma\sigma^\top(\cdot,u_{\tau+h-s})\right):v^{\tau-s}_{xx}(X^{\tau+h}_s)]\,\big|\,\mathcal{F}^W_s\right]ds\right]\nonumber\\ &+\lambda \mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\mathbb{E}\left[\ln\pi_{\tau+h-s}(X^{\tau+h}_s,u_{\tau+h-s}) - \ln\pi_{\tau-s}(X^{\tau+h}_s,u_{\tau-s})\,\big|\,\mathcal{F}^W_s\right] ds\right]. \end{align}\tag{9}\] With the notation \(\mathbf{H}^\tau\) defined in 5 and denoting by \(\nabla_u\) the Jacobi matrix, applying Itô’s formula to \(f\) and \(b\) as a function of \(u_\tau\), which follows 4 , for each \(0\leq s\leq \tau+h\), \[\begin{align} &\mathbb{E}\left[f(X^{\tau+h}_s,u_{\tau-s})-f(X^{\tau+h}_s,u_{\tau+h-s})\,\big|\,\mathcal{F}^W_s\right] \nonumber\\ =&\mathbb{E}\left[- \int_0^{h}\left((\nabla_u f)^\top\nabla_u \mathbf{H}^{\tau-s+l} + \lambda\mathrm{Tr}[\nabla^2_{uu}f]\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^W_s\right]\tag{10}\\ &\mathbb{E}\left[\left(b(\cdot,u_{\tau-s}) - b(\cdot,u_{\tau+h-s})\right)^\top v^{\tau-s}_{x}(X^{\tau+h}_s)\,\big|\,\mathcal{F}^W_s\right]\nonumber\\ =& \mathbb{E}\left[-\int_0^{h}\left((\nabla_u b\,\nabla_u \mathbf{H}^{\tau-s+l})^\top v^{\tau-s}_{x} + \lambda\sum\limits_{i=1}^d\mathrm{Tr}[\nabla^2_{uu}b_i]v^{\tau-s}_{x_i}\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^W_s\right]\tag{11}\\ &\mathbb{E}\left[\mathrm{Tr}[\left(\sigma\sigma^\top(\cdot,u_{\tau-s}) - \sigma\sigma^\top(\cdot,u_{\tau+h-s})\right):v^{\tau-s}_{xx}(X^{\tau+h}_s)]\,\big|\,\mathcal{F}^W_s\right]\nonumber\\ =& \mathbb{E}\left[-\int_0^{h} \left(\nabla_u \mathrm{Tr}[\sigma\sigma^\top \,v^{\tau-s}_{xx}]^\top\nabla_u \mathbf{H}^{\tau-s+l} + \lambda\nabla^2_{uu}\mathrm{Tr}[\sigma\sigma^\top\, v^{\tau-s}_{xx}]\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^W_s\right]\tag{12}, \end{align}\] which follow from the fact that the integral with respect to Brownian motion \(B\) is a martingale by Assumption 1. Furthermore, \[\begin{align} &\lambda\mathbb{E}\left[\ln\pi_{\tau+h-s}(X^{\tau+h}_s,u_{\tau+h-s}) - \ln\pi_{\tau-s}(X^{\tau+h}_s,u_{\tau-s})\,\big|\,\mathcal{F}^W_s\right]\nonumber\\ =&\lambda \mathbb{E}\left[\int_0^{h}\left(\frac{\nabla_\tau \pi_{\tau-s+l}}{\pi_{\tau-s+l}} + \frac{(\nabla_u \pi_{\tau-s+l})^\top\nabla_u \mathbf{H}^{\tau-s+l}}{\pi_{\tau-s+l}}\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^{W}_s\right]\nonumber\\ +&\lambda^2 \mathbb{E}\left[\int_0^{h}\left(\frac{\mathrm{Tr}[\nabla^2_{uu} \pi_{\tau-s+l}]}{\pi_{\tau-s+l}}- \frac{(\nabla_u \pi_{\tau-s+l})^\top\nabla_u \pi_{\tau-s+l}}{\pi^2_{\tau-s+l}}\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^{W}_s\right]\\ =& \lambda\mathbb{E}\left[\int_0^{h} \left(-\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s+l}] + 2\lambda\frac{\mathrm{Tr}[\nabla^2_{uu} \pi_{\tau-s+l}]}{\pi_{\tau-s+l}} - \lambda\|\nabla_u\ln \pi_{\tau-s+l}\|^2\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^{W}_s\right]\\ =&\lambda\mathbb{E}\left[\int_0^{h} \left(-\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s+l}]- \lambda\|\nabla_u\ln \pi_{\tau-s+l}\|^2\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,\Big|\,\mathcal{F}^{W}_s\right], \end{align}\] where the second equation follows from the Fokker-Planck equation for \(\pi_{\tau-s+l}\), and the last equation holds because \[\mathbb{E}\left[\frac{\mathrm{Tr}[\nabla^2_{uu} \pi_{\tau-s+l}]}{\pi_{\tau-s+l}}(X^{\tau+h}_s,u_{\tau-s+l})\,\Big|\,\mathcal{F}^{W}_s\right] =\int_{\mathbb{R}^n}\mathrm{Tr}[\nabla^2_{uu} \pi_{\tau-s+l}](X^{\tau+h}_s,u)du = 0.\nonumber\] Plugging these calculations back into 9 , \[\begin{align} &v^{\tau}(0,x) - v^{\tau+h}(0,x)\nonumber\\ =&-\mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\int_0^{h} \left((\nabla_u f)^\top\nabla_u \mathbf{H}^{\tau-s+l} + \lambda\mathrm{Tr}[\nabla^2_{uu}f]\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,ds\right]\nonumber\\ &-\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \int_0^{h}\left((\nabla_u b^\top v^{\tau-s}_x)^\top\nabla_u \mathbf{H}^{\tau-s+l}+ \lambda\sum\limits_{i=1}^d\mathrm{Tr}[\nabla^2_{uu}b_i]v^{\tau-s}_{x_i}\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\, ds\right]\nonumber\\ &-\frac{1}{2}\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \int_0^{h}\left(\nabla_u \mathrm{Tr}[\sigma\sigma^\top\, :v^{\tau-s}_{xx}]^\top\nabla_u \mathbf{H}^{\tau-s+l} + \lambda\nabla^2_{uu}\mathrm{Tr}[\sigma\sigma^\top\, : v^{\tau-s}_{xx}]\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\, ds\right]\nonumber\\ &+\lambda \mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\int_0^{h}\left(-\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s+l}]- \lambda\|\nabla_u\ln \pi_{\tau-s+l}\|^2 \right)(X^{\tau+h}_s,u_{\tau-s+l})dl\, ds\right]\\ =& -\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h}(\nabla_u\mathbf{H}^{\tau-s})^\top\nabla_u \mathbf{H}^{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]\nonumber\\ &-\lambda\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h}\left(\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s+l}]+\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s}]\right)(X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]\nonumber\\ &-\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h} \lambda^2\|\nabla_u\ln \pi_{\tau-s+l}\|^2 (X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]. \end{align}\] Notice that (and the same applies to \(\mathbb{E}\left[\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s}](X^{\tau+h}_s,u_{\tau-s+l})\,\big|\,X^{\tau+h}_s\right]\)) \[\begin{align} &\mathbb{E}\left[\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s+l}](X^{\tau+h}_s,u_{\tau-s+l})\,\big|\,X^{\tau+h}_s\right] = \int_{\mathbb{R}^n}\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s+l}]\pi_{\tau-s+l}(X^{\tau+h}_s,u)du\nonumber\\ =&-\int_{\mathbb{R}^n}(\nabla_u \mathbf{H}^{\tau-s+l})^\top\nabla_u\pi_{\tau-s+l}(X^{\tau+h}_s,u)du\nonumber\\ =& -\mathbb{E}\left[(\nabla_u \mathbf{H}^{\tau-s+l})^\top\nabla_u\ln\pi_{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})\,\big|\,X^{\tau+h}_s\right],\label{eq:H-2nd} \end{align}\tag{13}\] where the last equation follows from the boundedness of the model parameters and that \(\pi_{\tau-s+l}\), as a probability density function, vanishes at \(\partial \mathbb{R}^n\). Thus by the dominated convergence theorem, the above implies that \[\begin{align} &\frac{1}{h}(v^{\tau}(0,x) - v^{\tau+h}(0,x)) \nonumber\\ =& -\frac{1}{h}\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h}(\nabla_u \mathbf{H}^{\tau-s})^\top\nabla_u \mathbf{H}^{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]\nonumber\\ &+\frac{\lambda}{h}\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h}\left(\nabla_u \mathbf{H}^{\tau-s+l}+\nabla_u \mathbf{H}^{\tau-s}\right)^\top\nabla_u\ln\pi_{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]\nonumber\\ &-\frac{1}{h}\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h} \lambda^2\|\nabla_u\ln \pi_{\tau-s+l}\|^2 (X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]\nonumber\\ &\xrightarrow{h\rightarrow 0} -\mathbb{E}\left[\int_0^{\tau} e^{-\beta s}\|\nabla_u \mathbf{H}^{\tau-s})\|^2(X^{\tau}_s,u_{\tau-s})ds \right]\nonumber\\ &+2\lambda\mathbb{E}\left[\int_0^{\tau} e^{-\beta s}(\nabla_u \mathbf{H}^{\tau-s})^\top\nabla_u\ln\pi_{\tau-s}(X^{\tau}_s,u_{\tau-s})ds \right]\nonumber\\ &-\mathbb{E}\left[\int_0^{\tau} e^{-\beta s}\lambda^2\|\nabla_u\ln \pi_{\tau-s}\|^2 (X^{\tau}_s,u_{\tau-s})ds \right]\nonumber\\ =& -\mathbb{E}\left[\int_0^{\tau} e^{-\beta s}\|\nabla_u \mathbf{H}^{\tau-s} - \lambda \nabla_u\ln \pi_{\tau-s}\|^2(X^{\tau}_s,u_{\tau-s})ds \right]\leq 0. \end{align}\] Therefore \(v^{\tau}(0,x)\) is non-decreasing in \(\tau\). ◻
Next we derive another representation of \(v^\tau\)’s dynamics (also of \(v^\tau_x\) and \(v^\tau_{xx}\)), which helps the convergence analysis in the next sections.
Lemma 1. Fixing the state variable \(x\) and regarding \(v^\tau,v^\tau_x,v^\tau_{xx}\) as functions of \(\tau\), they satisfy the following dynamics, \[\begin{align} dv^\tau(x) =&\left(\int_{\mathbb{R}^n} (\mathbf{H}^\tau - \lambda\ln \pi_{\tau})\pi_{\tau}(x,u)du -\beta v^{\tau}(x) \right)d\tau,\label{v-dynamics}\\ dv^\tau_x(x) =&\left(\nabla_x\left(\int_{\mathbb{R}^n} (\mathbf{H}^\tau- \lambda\ln \pi_\tau )\pi_\tau(x,u)du\right) -\beta v^\tau_x(x)\right)d\tau,\nonumber\\ dv^\tau_{xx}(x) =&\left(\nabla^2_{xx} \left(\int_{\mathbb{R}^n} (\mathbf{H}^\tau - \lambda\ln \pi_\tau )\pi_\tau(x,u)du\right) -\beta v^\tau_{xx}(x)\right)d\tau.\nonumber \end{align}\qquad{(2)}\]
Proof. For any \(h>0\), since \(v^{\tau+h}(h,x) = v^\tau(0,x)\), \[\begin{align} &v^{\tau+h}(0,x) - v^\tau(0,x)\nonumber\\ =&\mathbb{E}\left[\int_0^{h} e^{-\beta s} \left(\int_{\mathbb{R}^n}\left(f - \lambda\ln\pi_{\tau+h-s}\right)\pi_{\tau+h-s}(X^{\tau+h}_s,u)du\right) ds + e^{-\beta h}v^{\tau+h}\left(h,X^{\tau+h}_{h}\right) -v^\tau(0,x)\right]\\ =&\mathbb{E}\left[\int_0^{h} e^{-\beta s} \left(\int_{\mathbb{R}^n}\left(f - \lambda\ln\pi_{\tau+h-s}\right)\pi_{\tau+h-s}(X^{\tau+h}_s,u)du\right) ds + e^{-\beta h}v^{\tau}\left(0,X^{\tau+h}_{h}\right) -v^\tau(0,x)\right]\\ =&\mathbb{E}\left[\int_0^{h} e^{-\beta s} \left(\int_{\mathbb{R}^n}\left(f - \lambda\ln\pi_{\tau+h-s}\right)\pi_{\tau+h-s}(X^{\tau+h}_s,u)du\right) ds\right]\nonumber\\ & + \mathbb{E}\left[\int_0^h e^{-\beta s} \left(-\beta v^\tau + \tilde{b}^\top v^{\tau}_x + \frac{1}{2}\mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top:v^{\tau}_{xx}]\right)(X^{\tau+h}_s,\pi_{\tau+h-s})ds \right], \end{align}\] where the last equation follows from Itô’s formula, and the stochastic integral with respect to the Brownian motion \(W\) in expectation is \(0\) due to Assumption 2. Dividing by \(h\) on both sides of the above equations and letting \(h\rightarrow 0\), the dominated convergence theorem implies that \[\begin{align} dv^\tau(x) =&\left(\int_{\mathbb{R}^n} \left(f - \lambda\ln\pi_{\tau} + b^\top v^\tau_x + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top \,v_{xx}^\tau]\right)\pi_{\tau}(x,u)du -\beta v^{\tau} \right)d\tau\nonumber\\ =&\left(\int_{\mathbb{R}^n} \left(\mathbf{H}^\tau - \lambda\ln \pi_\tau\right)\pi_{\tau}(x,u)du -\beta v^{\tau} \right)d\tau. \label{eq:v-dynamics} \end{align}\tag{14}\] Notice that the integral on the right hand side is with respect to \(u\). As a function of \(x\), \(\mathbf{H}(x,u,v^\tau_x,v^\tau_{xx}) - \lambda\ln\pi_\tau\) is smooth and its derivatives are locally bounded. Thus by thedominated convergence theorem, switching the order of derivative with respect to \(x\) and the integral with respect to \(u\) does not change its value. Therefore taking derivative on both sides with respect to \(x\) yields the dynamics for \(v^\tau_x\) and \(v^\tau_{xx}\). ◻
The coupled dynamics 4 and ?? enable continuous iterations of both the policy, the value function and its derivatives, and we will discuss their well-poseness in Section 5. Notice that at every \(\tau \geq 0\), in order to update the policy and value function given the latest \(v^\tau\), \(v^\tau_x\) and \(v^\tau_{xx}\), we only need to estimate \(\mathbf{H}^\tau\), \(\nabla_x\mathbf{H}^\tau\) and \(\nabla_{xx} \mathbf{H}^\tau\), instead of the higher dimensional \(v^\tau_{xxx}\) and \(v^\tau_{xxxx}\). It is considerably simpler than the discrete iteration in wang2020reinforcement?, huang2022convergence?. In these methods, one has to calculate the value 2 corresponding to the latest policy iterates, approximate it by a neural network, or solve the PDE the value function satisfies.
Under the assumption that the value \(V(x)\) is finite, the above policy improvement result suggests that \(v^\tau\) converges in \(\tau\). The next lemma provides convergence on its \(x\)-derivatives.
Lemma 2. \((v^\tau(t,x),v^\tau_t(t,x),v_x^\tau(t,x),v_{xx}^\tau(t,x))\) converges (up to a subsequence) locally uniformly. The limit is independent of \(t\), which we denote as \((v^*(x),v^*_t(x) = 0,v_x^*(x),v_{xx}^*(x))\).
Proof. Since for given \(x\in\mathbb{R}^d\), \(v^\tau(t,x)\) is non-decreasing in \(\tau\) and is bounded above by the optimal \(V(x)\), there exists a limit \(v^*(t,x)\). Notice that for each \(\tau\), \(v^\tau(t,x)\) solves \[\begin{align} &v^{\tau}_t(t,x) -\beta v^{\tau}(t,x) +\tilde{b}^\top (x,\pi_{\tau-t})v^{\tau}_x(t,x) + \frac{1}{2}\mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top \,v^{\tau}_{xx}(t,x)] \nonumber\\ &+ \int_{\mathbb{R}^n}\left(f(x,u) - \lambda \ln\pi_{\tau-t}\right)\pi_{\tau-t}(u)du = 0,\nonumber \end{align}\] and by Proposition 1, \(v^0(t,x) \leq v^{\tau}(t,x) \leq v^*(t,x)\). Also notice that since \(\pi_0\) is continuous in \(x\), and \(\int_{\mathbb{R}^n}\pi_\tau\ln\pi_\tau(x,u)du\) is bounded uniformly in \(\tau\), the latter is locally bounded in \(x\), independent of \(\tau\). Then by Schauder’s interior estimates (c.f. krylov1996lectures?[Chapter 8], which gives an estimated of the norm of the derivatives of the solution to a PDE in the interior of a bounded domain) imply that \(v^\tau_t(t,x)\), \(v_x^\tau(t,x)\) and \(v_{xx}^\tau(t,x)\) are locally uniformly equicontinuous, and thus by the Arzelà-Ascoli theorem, \(v_x^\tau(t,x)\), \(v_t^\tau(t,x)\) and \(v_{xx}^\tau(t,x)\), each as a family of locally uniformaly bounded and equicontinuos functions, converge (up to a subsequence) locally uniformly to \(v_t^*(t,x)\), \(v_x^*(t,x)\) and \(v^*_{xx}(t,x)\). Furthermore, by definition \(v^\tau(t,x) = v^{\tau-t}(0,x)\) for \(\tau \geq t\). Thus for any \(t \geq 0\), \[v^*(t,x) = \lim_{\tau\rightarrow\infty}v^\tau(t,x) = \lim_{\tau\rightarrow\infty}v^{\tau -t}(0,x) = v^*(0,x).\] Therefore \(v^*\) is independent of \(t\), \(v^*_t(t,x)=0\) and we can drop the argument \(t\) in \(v^*\), \(v^*_x\) and \(v^*_{xx}\). ◻
To discuss the convergence of \(\pi_\tau\), we make the following assumptions on \(v^*\).
Assumption 3. Given every \(x\in\mathbb{R}^d\), (i) \(\int_{\mathbb{R}^n}e^{\mathbf{H}(x,u, v_x^*, v_{xx}^*)/\lambda}du < \infty\). (ii) \(\nabla_u \ln\pi_\tau\) is locally bounded as a function of \(u\), uniformly in \(\tau\).
Proposition 2. If Assumption 3 holds, then \(\pi^* = \lim\limits_{\tau\rightarrow\infty}\pi_\tau = \frac{e^{\mathbf{H}^*(x,u)/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u')/\lambda}du'}\), where \(\mathbf{H}^*(x,u) = \mathbf{H}(x,u,v^*_x,v^*_{xx})\).
Proof. With \(x\in\mathbb{R}^d\) given, since \(v^\tau (x)\) converges (cf. 14 ), \[\begin{align} &\int_{\mathbb{R}^n} \left(f - \lambda\ln\pi_{\tau} + b^\top v_x^\tau + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top:v_{xx}^\tau]\right)\pi_{\tau}(x,u)du -\beta v^{\tau}(x)\nonumber\\ =&\int_{\mathbb{R}^n} \left(\mathbf{H}^\tau- \lambda\ln\pi_{\tau}\right)\pi_{\tau}(x,u)du -\beta v^{\tau}, \end{align}\] converges to \(0\) as \(\tau\rightarrow\infty\). In other words, \[g^\tau(x) = \int_{\mathbb{R}^n} \left(\mathbf{H}^\tau- \lambda\ln\pi_{\tau}\right)\pi_{\tau}(x,u)du,\] converges to a constant \(\beta v^*(x)\). Notice that \(\pi_{\tau}\) is the distribution density function of \(u_\tau\), thus \[\begin{align} &g^{\tau}(x)- g^{\tau+h}(x)\nonumber\\ =&\mathbb{E}\left[f(\cdot,u_{\tau})-f(\cdot,u_{\tau+h})+ b^\top(\cdot,u_\tau)v^{\tau}_x -b^\top(\cdot,u_{\tau+h})v^{\tau+h}_x\right](x)\nonumber\\ &+ \mathbb{E}\left[\frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(\cdot,u_\tau):v^\tau_{xx} - \sigma\sigma^\top(\cdot,u_{\tau+h}):v^{\tau+h}_{xx}](x) + \lambda\left(\ln\pi_{\tau+h}(x,u_{\tau+h}) - \ln\pi_{\tau}(x,u_{\tau})\right)\right]\nonumber\\ =&\mathbb{E}\left[f(\cdot,u_{\tau})-f(\cdot,u_{\tau+h})+ (b(\cdot,u_\tau) - b(\cdot,u_{\tau+h}))^\top v^{\tau}_x + b^\top(\cdot,u_{\tau+h})(v^{\tau}_x-v^{\tau+h}_x) \right](x)\nonumber\\ &+ \mathbb{E}\left[\frac{1}{2}\mathrm{Tr}[(\sigma\sigma^\top(\cdot,u_\tau)-\sigma\sigma^\top(\cdot,u_{\tau+h})):v^\tau_{xx}] + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(\cdot,u_{\tau+h}):(v^\tau_{xx}-v^{\tau+h}_{xx})]\right](x)\nonumber\\ &+ \mathbb{E}\left[\lambda(\ln\pi_{\tau+h}(x,u_{\tau+h}) - \ln\pi_{\tau}(x,u_{\tau}))\right]. \end{align}\] Then by similar argument to the proof of Proposition 1 (cf. 9 ), \[dg^\tau = \mathbb{E}\left[\|\nabla_u \mathbf{H}^\tau - \lambda \nabla_u \ln\pi_{\tau}\|^2(x,u_\tau)\right]d\tau -\tilde{b}^\top(x,\pi_\tau)dv^\tau_x-\frac{1}{2}\mathrm{Tr}[\tilde{\sigma}\tilde{\sigma}^\top(x,\pi_\tau):dv^{\tau}_{xx}].\label{eq:g-tau}\tag{15}\] The fact that \(v^\tau_x\) and \(v^{\tau}_{xx}\) converge combined with the boundedness of \(\tilde{b}\) and \(\tilde{\sigma}\) implies that the last two terms converge to \(0\) as \(\tau\rightarrow \infty\). Since the diffusion coefficient for \(u\) is constant \(\sqrt{2\lambda} >0\), \(u_\tau\) is supported on \(\mathbb{R}^n\) for every \(\tau >0\). Then \(g^\tau\) being convergent implies that \(\nabla_u \ln\pi_{\tau}(x,u)\) converges to \(\nabla_u \mathbf{H}(x,u,v^*_x,v^*_{xx})/\lambda\). In other words, with \(h_\tau(x,u) = \nabla_u (\ln\pi_{\tau}-\mathbf{H}^\tau/\lambda)(x,u)\), \(\lim\limits_{\tau\rightarrow\infty}h_\tau(x,u) =0\). Furthermore, since \(v^\tau_x\) and \(v^\tau_{xx}\) converge and \(b\), \(\sigma\) and \(f\) are bounded by assumption, \(\nabla_u \mathbf{H}^\tau\) is locally bounded in \(u\), and the bounds are uniform in \(\tau\). By assumption on \(\nabla_u\ln\pi_\tau\), the same holds for \(h_\tau\).
Furthermore, since \(\pi_\tau\) is a probability density function, \[\pi_\tau(x,u) = \frac{e^{\mathbf{H}(x,u,v^\tau_x,v^\tau_{xx})/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}}{\int_{\mathbb{R}^n}e^{\mathbf{H}^\tau(x,u')/\lambda + \int_0^{u'_1}\cdots\int_0^{u'_n}h_\tau(x,r)dr_1\dots dr_n}du'}.\] Notice that by Fatou’s lemma, \[\begin{align} &\int_{\mathbb{R}^n}e^{\mathbf{H}^\tau(x,u')/\lambda + \int_0^{u'_1}\cdots\int_0^{u'_n}h_\tau(x,r)dr_1\dots dr_n}du'\geq \int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u)/\lambda + \liminf\limits_{\tau\rightarrow\infty}\int_0^{u'_1}\cdots\int_0^{u'_n}h_\tau(x,r)dr_1\dots dr_n}du\\ =&\int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u)/\lambda}du, \end{align}\] where the last equation follows from the dominated convergence theorem. The denominator of \(\pi_\tau\) above can be approximated similarly, and \[\limsup\limits_{\tau\rightarrow\infty}\pi_\tau(x,u) \leq \frac{\limsup\limits_{\tau\rightarrow\infty}e^{\mathbf{H}^\tau(x,u,v^\tau_x,v^\tau_{xx})/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}}{\liminf\limits_{\tau\rightarrow\infty}\int_{\mathbb{R}^n}e^{\mathbf{H}^\tau(x,u)/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}du} \leq \frac{e^{\mathbf{H}^*(x,u)/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u')/\lambda} du'}.\] On the other hand, following a similar argument, \(\liminf\limits_{\tau\rightarrow\infty}\pi_\tau(x,u) \geq \frac{e^{\mathbf{H}^*(x,u)/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u')/\lambda} du'}\), which concludes the proof. ◻
Proposition 3 (Optimality). Suppose \(\rho^* = \{\rho^*_t = \pi^*(X^*_t), t\geq 0)\} \in \mathcal{A}(x)\), where \(X^*\) is the solution to 1 under the policy \(\rho^*\). Then \(\rho^*\) is the optimal strategy for the relaxed control problem \(\sup\limits_{\pi}J(x,\pi)\). Furthermore, \[v^*(x) = \mathbb{E}\left[\int_0^\infty e^{-\beta t}\int_{\mathbb{R}^n}\left(f- \lambda \pi^*\right)\ln\pi^*(X^*_t,u)dudt|X^*_0 = x\right],\] and it is the corresponding value function.
Proof. In this proof we focus on showing that \(v^*\) is indeed the solution to the HJB equation of the relaxed control problem and that \(\pi^*\) is the associated maximizer of the Hamiltonian. The verification that \(\pi^*\) is the optimal strategy and \(v^*\) is the corresponding value follows standard arguments which we omit here. With \(\pi^*(x,u) = \frac{e^{\mathbf{H}^*(x,u)/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u')/\lambda} du'}\) and \(c = \int_{\mathbb{R}^n}e^{\mathbf{H}^*(x,u)/\lambda} du\), \[\begin{align} &\int_{\mathbb{R}^n} \left(f - \lambda\ln\pi^* + b^\top v_x^* +\frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top \,v_{xx}^*]\right)\pi^*(x,u)du -\beta v^* = \lambda\ln c - \beta v^*.\nonumber \end{align}\] On the other hand, since \(v^\tau\) converges, from the dynamics of \(v^\tau(x)\) in 14 , \[\lim_{\tau\rightarrow\infty}\left(\int_{\mathbb{R}^n} \left(f - \lambda\ln\pi_{\tau} + b^\top v_x^\tau+\frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top \,v^\tau_{xx}]\right)\pi_{\tau}(x,u)du -\beta v^{\tau}\right) = 0. \nonumber\] With \(h_\tau\) defined in the proof of Proposition 2, \(\pi_\tau(x,u) = \frac{e^{\mathbf{H}^\tau(x,u)/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}}{\int_{\mathbb{R}^n}e^{\mathbf{H}^\tau(x,u')/\lambda + \int_0^{u'_1}\cdots\int_0^{u'_n}h_\tau(x,r)dr_1\dots dr_n}du'}\), and \(\lim\limits_{\tau\rightarrow \infty}h_{\tau} = 0\). Thus the above equation implies that \(\lim\limits_{\tau\rightarrow\infty}\lambda \ln c_\tau -\beta v^* = 0\), where \[c_\tau = \int_{\mathbb{R}^n}e^{\mathbf{H}^\tau(x,u)/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}du = \frac{e^{\mathbf{H}^\tau(x,u)/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}}{\pi_\tau(x,u)}.\] Notice that \[\begin{align} \lim_{\tau\rightarrow\infty} c_\tau = \lim_{\tau\rightarrow\infty}\frac{e^{\mathbf{H}^\tau(x,u)/\lambda + \int_0^{u_1}\cdots\int_0^{u_n}h_\tau(x,r)dr_1\dots dr_n}}{\pi_\tau(x,u)} = \frac{e^{\mathbf{H}^*(x,u)/\lambda}}{\pi^*(x,u)} = c.\nonumber \end{align}\] Thus \(\beta v^* = \lim\limits_{\tau\rightarrow\infty}\lambda \ln c_\tau = \lambda\ln c\), and \[\int_{\mathbb{R}^n} \left(f- \lambda\ln\pi^* + b^\top v_x^*+\frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top:v_{xx}^*]\right)\pi^*(x,u)du -\beta v^* = 0.\nonumber\] Furthermore, since the maximizer of (which is pointwise concave in \(\pi(u)\)) \[-\beta v(x) +\sup_{\pi\in\mathcal{P}(\mathbb{R}^n)} \int_{\mathbb{R}^n} \left(\mathbf{H}(x,u,v_x,v_{xx}) -\lambda \ln\pi(u)\right)\pi(u)du, \nonumber\] is \(\hat{\pi} = \frac{e^{\mathbf{H}(x,u,v_x,v_{xx})/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}(x,u',v_x,v_{xx})/\lambda} du'}\), it implies that \[\sup_{\pi\in\mathcal{P}(\mathbb{R}^n)}\int_{\mathbb{R}^n} \left(\mathbf{H}(x,u,v^*_x,v^*_{xx}) -\lambda\ln \pi(u)\right)\pi(u)du -\beta v^*= 0. \nonumber\] Therefore, \(v^*\) solves the HJB equation, and \(\pi^*\) is the corresponding maximizer. ◻
In this section, we apply the approach similar to the policy and value iteration developed above to a classical infinite horizon stochastic control problem, where at every time \(t\geq 0\), the agent chooses the control \(u_t\), whereas the state variable \(X\) follows \[dX^u_t = b(X^u_t,u_t)dt +\sigma(X^u_t,u_t)dW_t, \quad X_0 = x.\] For the ease of notation, we drop the superscript \(u\) in the following discussion if no ambiguity arises. The agent’s goal is to maximize the cumulative discounted running payoff \(f(X_t,u_t)\), i.e., \(V(x) = \sup\limits_{u\in\mathcal{A}(x)} J(x,u)\), where \[J(x,u) = \mathbb{E}\left[\int_0^\infty e^{-\beta t}f(X_t,u_t)dt\right],\] and the admissible set of the control \(u\) is defined as
Definition 3. For any initial state \(x\in\mathbb{R}^d\), the admissible set \(\mathcal{A}(x)\) is a collection of strategies \(u = \{u_t \in \mathbb{R}^d, t\geq 0\}\), progressively measurebale to \(\{\mathcal{F}^W_t, t\geq 0\}\), and satisfy
\(\mathbb{E}\left[\int_0^T \left(|b(X_t,u_t)|^2 + \mathrm{Tr}[\sigma\sigma^\top(X_t,u_t)]\right) dt \right]<\infty\), for each \(T > 0\).
\(J(x,u) < \infty\).
Similar to the relaxed control problem, we assume that the optimal value is finite, i.e., the optimization problem is well-defined.
Assumption 4. \(V(x) < \infty\).
The HJB equation associated to this control problem is \[\begin{align} &-\beta v(x) + \sup_{u}\left(f(x,u) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u):v_{xx}(x)] + b^\top(x,u)v_x(x)\right)\\ =& -\beta v(x) +\sup_{u} \mathbf{H}(x,u,v_x,v_{xx}) =0, \end{align}\] where \(\mathbf{H}\) is defined in 5 . Assuming \(\mathbf{H}(x,u,v_x,v_{xx})\) is strictly concave in \(u\), there is a unique solution \(\hat{u}\) to the equation \(\nabla_u \mathbf{H}(x,u,v_x,v_{xx}) = 0\), which we denote as \(U(x,v_{x},v_{xx})\). If \(\hat{v} = V\), which is supposed to solve the HJB equation, then \(\hat{u} = U(x,\hat{v}_x,\hat{v}_{xx})\) is the optimal control. Without knowing concavity of \(\mathbf{H}\) and the optimal value \(\hat{v}\), we consider the following iteration of \(u\): for each state \(x\in\mathbb{R}^d\), given initial value \(u^x_0\), \[\label{eq:u-dynamics-c} du^x_\tau = \nabla_u \mathbf{H}(x,u^x_\tau,v^\tau_x(0,x),v^\tau_{xx}(0,x))d\tau+ \sqrt{2\lambda}dB_\tau,\tag{16}\] where \(v^\tau\) is the result of the value iteration. For each \(\tau\), \(v^\tau(0,x) = J(x,u^\tau)\), the value corresponding to a strategy \(u^\tau = \{u^\tau_t\}_{t\geq 0}\), which may be time-inhomogeneous. We update \(u^\tau\) in the following way (similar to the case of relaxed control, we will denote the strategy \(u^x_\tau\) as a function of state \(x\), \(u_\tau(x)\), to emphasize the dependence, and omit the argument \(x\) if no ambiguity arises):
Definition 4. For each \(\tau\geq 0\), given \(\{u_s(x),0\leq s\leq \tau\}\) for every \(x\), define \(\mathbf{u}^\tau = \{u^\tau_t(X^\tau_t)\}_{0\leq t<\infty}\) as \[\label{eq:u-reverse} u^\tau_t(x) = \begin{cases} u_{\tau-t}(x), &0\leq t \leq \tau;\\ u_0(x), & t > \tau. \end{cases}\qquad{(3)}\] where the corresponding state process \(X^{\tau}\) satisfies \[dX^\tau_t = b(X^\tau_t,u^\tau_t(X^\tau_t))dt +\sigma(X^\tau_t,u^\tau_t(X^\tau_t))dW_t,\] and the corresponding value function, given a realization of strategy \(u\) up to \(\tau\), is \[v^\tau(t,x) = \mathbb{E}\left[\int_t^\infty e^{-\beta(s-t)}f(X^\tau_s,u^\tau_s(X^\tau_s))ds\,\Big|\,X^\tau_t = x, \mathcal{F}^B_\tau\right].\label{definition:v-tau-c}\qquad{(4)}\]
Notice that at each \(\tau\), though strategy \(\mathbf{u}^\tau\) defined above is a strategy for an infinite horizon stochastic control problem,it only relies on \(u_s\), \(0\leq s\leq \tau\), i.e. the iteration results up to the current step, and not the future steps. Also, even though \(u_\tau\) is a diffusion process driven by Brownian motion \(B\), in the above definition of \(v^\tau\), given \(\mathcal{F}^B_\tau\), each \(u^\tau_t\) is a deterministic function of the state \(x\). Thus \(\mathbf{u}^\tau = \{u^\tau_t(X^\tau_t)\}_{0\leq t<\infty}\) is progressively measurable with respect to \(\{\mathcal{F}^W_t, t\geq 0\}\). We make the following assumptions on \(u^\tau\), \(X^\tau\) and \(v^\tau\).
Assumption 5. Given initial state \(X_0 = x\) and \(\tau \geq 0\), for each realization of \(\{u_s,0\leq s\leq \tau\}\),
\(\mathbf{u}^\tau \in\mathcal{A}(x)\) and \(v^\tau(t,x) \in C^{1,4}\).
\(\mathbb{E}\left[\int_0^{\tau +h} \|v^{\tau}_x(s,X^{\tau+h}_s)\|^2ds\right]< \infty\).
\(\mathbb{E}\left[\left(|v^{\tau-s+l}|+\|v^{\tau-s+l}_x\| + \|v^{\tau-s+l}_{xx}\|+\|\nabla_u\ln \pi_{\tau-s+l}\|^2\right)(X^{\tau+h}_s,u_{\tau-s+l})\right]\) is bounded in \(0\leq l \leq h\), for every \(0\leq s\leq \tau+h\).
In the following discussion, we are mainly interested in the updates of \(v^\tau(0,x)\) for a given \(x\), and for the ease of notation, we drop the time argument and denote it as \(v^\tau(x)\), if no ambiguity arises. Similarly to the relaxed control case, the value iteration for the classical control problem is also driven by continuous dynamics.
Lemma 3. The value function \(v^{\tau}(x)\), and its first and second order derivatives \(v^{\tau}_x\), \(v^{\tau}_{xx}\) satisfy the following dynamics, \[\begin{align} \label{eq:v-dynamics-c} dv^\tau(x) =& \left(-\beta v^\tau + f +b^{\top}v^\tau_x + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^{\top}:v^\tau_{xx}]\right)(x,u_\tau)d\tau;\\ dv^\tau_x(x) =& \left(-\beta v^\tau_x +\nabla_x\Big( f+b^{\top} v^\tau_x + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^{\top}:v^\tau_{xx}]\Big)\right)(x,u_\tau d\tau;\nonumber\\ dv^\tau_{xx}(x) =& \left(-\beta v^\tau_{xx}+\nabla^2_{xx}\Big( f(x,u_\tau) +b^{\top} v^\tau_x + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^{\top}:v^\tau_{xx}]\Big)\right)(x,u_\tau)d\tau.\nonumber \end{align}\qquad{(5)}\]
Proof. Following an argument similar to that for the relaxed control problem in Lemma 1, for any \(h>0\), since \(v^{\tau+h}(h,x) = v^\tau(0,x)\), \[\begin{align} &v^{\tau+h}(0,x) - v^\tau(0,x)\nonumber\\ =&\mathbb{E}\left[\int_0^{h} e^{-\beta s} f\, ds + e^{-\beta h}v^{\tau+h}\left(h,X^{\tau+h}_{h}\right) -v^\tau(0,x)\,\Big|\,X_0=x,\mathcal{F}^B_\tau\right] \\ =& \mathbb{E}\Big[\int_0^h e^{-\beta s} \Big(-\beta v^\tau + f+b^\top v^{\tau}_x + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top :v^{\tau}_{xx}]\Big)(X^{\tau+h}_s,u_{\tau+h-s}) ds \,\Big|\,X_0=x,\mathcal{F}^B_\tau\Big]. \end{align}\] The rest follows similarly to Lemma 1. ◻
Lemma 4. For \(\tau_1 < \tau_2\), \(\mathbb{E}[v^{\tau_1}(t,x)] \leq \mathbb{E}[v^{\tau_2}(t,x)]\), if one of the following conditions holds:
\(\lambda = 0\), or more generally
For every \(\tau\), \(\mathbb{E}\left[\|v^{\tau-s+l}_x(X^{\tau+h}_s)\| + \|v^{\tau-s+l}_{xx}(X^{\tau+h}_s)\|+\big|\int_{\mathbb{R}^n} \frac{1}{\pi_{\tau-s+l}}\|\nabla_u \pi_{\tau-s+l}\|^2(X^{\tau+h}_s,u)du\big|\right]\) is bounded in \((\tau-s)^-\leq l \leq h\), for every \(0\leq s\leq \tau+h\) and \(h\) sufficiently small, and \[\mathbb{E}\left[\int_0^\tau e^{-\beta s}\left(\|\nabla_u \mathbf{H}^{\tau-s} -\frac{\lambda}{2}\nabla_u \ln\pi_{\tau-s}\|^2 - \frac{\lambda^2}{4}\|\nabla_u\ln\pi_{\tau-s}\|^2 \right)(X^{\tau}_s,u_{\tau-s})ds \right]\geq 0,\] where \(\pi_{\tau-s+l}\) is the probability density function of \(u_{\tau-s+l}\) following 16 .
Proof. Similarly to the case of relaxed control, we only show the proof for \(t=0\), and other cases follow by the same argument. By Itô’s formula, \[\begin{align} &e^{-\beta t}v^{\tau}(t,X^{\tau+h}_t) - v^{\tau}(0,x)\nonumber\\ =& \int_0^{t} e^{-\beta s} \left(v^{\tau}_t(s,X^{\tau+h}_s) -\beta v^{\tau}(s,X^{\tau+h}_s) + b^\top(X^{\tau+h}_s,u_{\tau+h-s}) v^{\tau}_x(s,X^{\tau+h}_s)\right.\nonumber\\ &\left. + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(X^{\tau+h}_s,u_{\tau+h-s}):v^{\tau}_{xx}(s,X^{\tau+h}_s)]\right)ds+ \int_0^{t} e^{-\beta s} v^{\tau}_x(s,X^{\tau+h}_s)^\top \sigma(X^{\tau+h}_s,u_{\tau+h-s})dW_s\\ =& \int_0^{t} e^{-\beta s} \left(v^{\tau-s}_t -\beta v^{\tau-s} + b^\top v^{\tau-s}_x + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top:v^{\tau-s}_{xx}]\right)(X^{\tau+h}_s,u_{\tau+h-s})ds\nonumber\\ &+ \int_0^{t} e^{-\beta s} v^{\tau-s}_x{}^\top \sigma(X^{\tau+h}_s,u_{\tau+h-s})dW_s. \end{align}\] Under Assumption 5, the integral with respect to \(W\) is a martingale, and \[\begin{align} &\mathbb{E}\left[v^{\tau}(0,x)\right] \nonumber\\ =& \mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s} \left(-v^{\tau-s}_t +\beta v^{\tau-s} - b^\top v^{\tau-s}_x -\frac{1}{2} \mathrm{Tr}[\sigma\sigma^\top:v^{\tau-s}_{xx}]\right)(X^{\tau+h}_s,u_{\tau+h-s})ds\right]\nonumber\\ &+ \mathbb{E}\left[e^{-\beta(\tau+h)}v^{\tau}(\tau+h,X^{\tau+h}_{\tau+h})\right]. \end{align}\] On the other hand, by the definition of \(v^{\tau}\) and standard argument similar to that for 7 , \(v^{\tau}(t,x)\) satisfies, \[\begin{align} &v^{\tau}_t(t,x) -\beta v^{\tau}(t,x) +b^\top(x,u_{\tau-t})v^{\tau}_x(t,x) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u_{\tau-t}):v^{\tau}_{xx}(t,x)] + f(x,u_{\tau-t}) = 0. \end{align}\] Substituting this back into the last equation for \(v^\tau(0,x)\), \[\begin{align} &\mathbb{E}[v^{\tau}(0,x)] \nonumber\\ =& \mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \left(\left(b(X^{\tau+h}_s,u_{\tau-s}) - b(X^{\tau+h}_s,u_{\tau+h-s})\right)^\top v^{\tau-s}_x(X^{\tau+h}_s)\right) ds\right]\nonumber\\ &+ \frac{1}{2}\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \mathrm{Tr}[\left(\sigma\sigma^\top(X^{\tau+h}_s,u_{\tau-s}) - \sigma\sigma^\top(X^{\tau+h}_s,u_{\tau+h-s})\right):v^{\tau-s}_{xx}(X^{\tau+h}_s)] ds\right]\nonumber\\ &+ \mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s} f(X^{\tau+h}_s,u_{\tau-s}) ds\right] + \mathbb{E}\left[e^{-\beta(\tau +h)}v^{\tau}(\tau +h,X^{\tau+h}_{\tau+h})\right]\\ =&\mathbb{E}[v^{\tau+h}(0,x)]+\mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\left( f(X^{\tau+h}_s,u_{\tau-s})-f(X^{\tau+h}_s,u_{\tau+h-s})\right)ds\right]\nonumber\\ &+\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \left((b(X^{\tau+h}_s,u_{\tau-s}) - b(X^{\tau+h}_s,u_{\tau+h-s}))^\top v^{\tau-s}_x(X^{\tau+h}_s)\right) ds\right]\nonumber\\ &+ \frac{1}{2}\mathbb{E}\left[\int_0^{\tau +h} e^{-\beta s} \mathrm{Tr}\Big[\left(\sigma\sigma^\top(X^{\tau+h}_s,u_{\tau-s}) - \sigma\sigma^\top(X^{\tau+h}_s,u_{\tau+h-s})\right):v^{\tau-s}_{xx}(s,X^{\tau+h}_s)\Big] ds\right],\label{eq:v-difference-c} \end{align}\tag{17}\] where the last equation follows from a similar argument as that for 9 and the fact that by definition \(v^{\tau}(\tau +h,X^{\tau+h}_{\tau+h}) = v^{0}(0,X^{\tau+h}_{\tau+h}) = v^{\tau+h}(\tau +h,X^{\tau+h}_{\tau+h})\). Notice that by Itô’s formula, and the fact that the integral with respect to Brownian motion \(B\) is a martingale under Assumption 5, following similar calculations as in previous sections, for \(0\leq s\leq \tau+h\), 10 , 11 and 12 hold, and imply that \[\begin{align} &\mathbb{E}[v^{\tau}(0,x) - v^{\tau+h}(0,x)]\nonumber\\ =&-\mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\left(\int_0^{h}(\nabla_u\mathbf{H}^{\tau-s})^\top\nabla_u \mathbf{H}^{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})dl\right)ds\right]\nonumber\\ &-\lambda\mathbb{E}\left[\int_0^{\tau+h} e^{-\beta s}\int_0^{h}tr[\nabla^2_{uu} \mathbf{H}^{\tau-s}](X^{\tau+h}_s,u_{\tau-s+l})dl\,ds \right]. \end{align}\] Furthermore, 13 implies that, \[\mathbb{E}\left[\mathrm{Tr}[\nabla^2_{uu} \mathbf{H}^{\tau-s}](x,u_{\tau-s+l})\right] = -\mathbb{E}\left[(\nabla_u \mathbf{H}^{\tau-s})^\top\nabla_u\ln\pi_{\tau-s+l}(x,u_{\tau-s+l})\right],\] where \(\pi_{\tau-s+l}\) is the distribution density function of \(u_{\tau-s+l}\). Plugging this into \(\mathbb{E}[v^{\tau}(0,x) - v^{\tau+h}(0,x)]\) above, and dividing by \(h\) on both sides, \[\begin{align} &\frac{\mathbb{E}[v^{\tau+h}(0,x) - v^{\tau}(0,x)]}{h}\nonumber\\ =& \mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\left(\int_0^{h}\frac{1}{h}(\nabla_u\mathbf{H}^{\tau-s})^\top\nabla_u \mathbf{H}^{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})dl\right)ds\right]\nonumber\\ &-\lambda\mathbb{E}\left[\int_0^{\tau +h}e^{-\beta s}\int_0^{h} \frac{1}{h}(\nabla_u \mathbf{H}^{\tau-s})^\top\nabla_u\ln\pi_{\tau-s+l}(X^{\tau+h}_s,u_{\tau-s+l})dl\, ds\right]. \end{align}\] Condition (iii) of Assumption 5 guarantees that the dominated convergence theorem applies, and letting \(h\rightarrow 0\), \[\begin{align} \frac{d\mathbb{E}[v^\tau(0,x)]}{d\tau} =& \mathbb{E}\left[\int_0^{\tau}e^{-\beta s}\left(\|\nabla_u\mathbf{H}^{\tau-s})\|^2-\lambda(\nabla_u \mathbf{H}^{\tau-s})^\top\nabla_u\ln\pi_{\tau-s}\right)(X^{\tau}_s,u_{\tau-s})ds\right]\nonumber\\ =&\mathbb{E}\left[\int_0^{\tau} e^{-\beta s}\left(\|\nabla_u \mathbf{H}^{\tau-s} -\frac{\lambda}{2}\nabla_u\ln\pi_{\tau-s}\|^2 -\frac{\lambda^2}{4}\|\nabla_u \ln\pi_{\tau-s}\|^2\right)(X^{\tau}_s,u_{\tau-s})ds\right]\nonumber\\ &\geq 0, \end{align}\] which holds because of assumption (ii) of the lemma, and covers \(\lambda =0\) as a special case. Thus \(\mathbb{E}[v^{\tau}(0,x)]\) is non-decreasing in \(\tau\). ◻
In Tthe following discussion we focus on the case of \(\lambda = 0\), where the iterations of the value function deterministic.
Proposition 4. With \(\lambda = 0\), (i) \(v^\tau\) converges to \(v^*\), and (ii) If \[\nabla_u (\mathbf{H}(x,u^*,v^*_x(x),v^*_{xx}(x)))|_{|u|=\infty} \neq 0,\] then \(u_\tau\) converges to \(u^*\) and \(\nabla_u (\mathbf{H}(x,u^*,v^*_x(x),v^*_{xx}(x))) =0\). If in addition, \[u^*(x) = \mathop{\mathrm{arg\,max}}\limits_u \mathbf{H}(x,u,v^*_x(x),v^*_{xx}(x)),\] e.g., \(\mathbf{H}\) is concave in \(u\)3 and \(\hat{u} = \{u_t = u^*, t\geq 0\} \in \mathcal{A}(x)\), then \(\hat{u}\) is the optimal strategy for the stochastic control problem \(V(x) = \sup\limits_{u\in\mathcal{A}(x)} J(x,u)\).
Proof. If \(\lambda = 0\), then \(u_\tau\) and thus \(v^\tau\) are deterministic. Lemma 4 implies that \(v^\tau\) is non-decreasing in \(\tau\), and because the value function is finite, it converges to \(v^*\). Since for any \(\tau\), \(v^\tau(x) \in [v^0(x),v^*(x)]\), it is locally bounded. Then by the boundedness of \(f,b\) and \(\sigma\), following similar arguments as for Lemma 2, \((v^\tau_t,v^\tau,v^\tau_x,v^\tau_{xx})\) converge (up to a subsequence) locally uniformly to \((v^*_t=0,v^*,v^*_x,v^*_{xx})\).
Furthermore, since the monotone function \(v^\tau(x)\) converges, 16 implies that \[-\beta v^\tau(x) + f(x,u_\tau) + b^\top(x,u_\tau)v^\tau_x(x) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u_\tau):v^\tau_{xx}(x)],\] converges to \(0\). In other words, for every \(x\) \[g_\tau = f(x,u_\tau) + b^\top(x,u_\tau)v^\tau_x(x)+\frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u_\tau):v^\tau_{xx}(x)],\] converges to constant \(\beta v^*(x)\). By calculations similar to 15 , \[dg_\tau =\|\nabla_u \mathbf{H}^\tau(x,u_\tau)\|^2d\tau + b^\top(x,u_\tau)dv^\tau_x(x) +\frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u_\tau):dv^\tau_{xx}(x)].\] Since \(g_\tau, v^\tau_x(x)\) and \(v^\tau_{xx}(x)\) converge in \(\tau\), their derivative with respect to \(\tau\) should converge to \(0\), which implies that \(\nabla_u \mathbf{H}^\tau(x,u_\tau)\) converges to \(0\). Since \(\nabla_u(\mathbf{H}(x,u^*,v^*_x(x),v^*_{xx}(x)))|_{|u|=\infty} \neq 0\), \(\{u_\tau\}_{\tau\geq 0}\) is bounded, and thus up to a subsequence converges to \(u^*\), and by the continuity of \(\mathbf{H}\), \(\nabla_u (\mathbf{H}(x,u^*,v^*_x(x),v^*_{xx}(x))) =0\).
Finally, since \((v^\tau,v^\tau_x,v^\tau_{xx},u_\tau)\) converge, from ?? , \[\begin{align} 0 =& \lim_{\tau\rightarrow\infty}\left(-\beta v^\tau(x) + f(x,u_\tau) + b^\top(x,u_\tau) v^\tau_x(x)+ \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u_\tau):v^\tau_{xx}(x)]\right)\nonumber\\ =& -\beta v^*(x) + f(x,u^*) + b^\top(x,u^*)v^*_x(x) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u^*):v^*_{xx}(x)]. \end{align}\] If \(u^*(x) = \mathop{\mathrm{arg\,max}}\limits_u \mathbf{H}(x,u(x),v^*_x(x),v^*_{xx}(x))\), then \(\sup_{u}\left(-\beta v^*(x) + \mathbf{H}(x,u,v^*_x,v^*_{xx})\right) = 0\). Thus \(v^*\) satisfies the HJB equation corresponding to the control problem \(V(x) = \sup\limits_{u\in\mathcal{A}(x)} J(x,u)\), with \(u^*\) being the maximizer of the associated Hamiltonian. The verification of optimality follows standard arguments and is omitted here. ◻
In the case of non-concave Hamiltonian \(\mathbf{H}\), assuming that the policy iteration has a limit, we propose the following criterion to check for optimality, which involves a sequence of policy and value iterations. This criterion applies to any policy iteration algorithm (beyond 16 ) for the inhomogeneous Hamiltonian coupled with value iteration ?? .
Definition 5. Given initial values \(u_0\) and \(v_0(x) = J(x,u^0)\), where \(u^0 = \{u_t = u_0, t\geq 0\}\), denote as \(u^{(1)}_\tau\) and \(v^{(1)\tau}\) the policy and value following 16 \((\)or any other policy iteration algorithm\()\) and ?? , and denote as \(u^{(1)*}\) and \(v^{(1)*}\) the limit of these iterations. For \(i \geq 1\), if \(u^{(i)*}(x) \neq \mathop{\mathrm{arg\,max}}\limits_u \mathbf{H}(x,u,v^{(i)*}_x(x),v^{(i)*}_{xx}(x))\), then let \[u^{(i+1)}_0(x) = \mathop{\mathrm{arg\,max}}\limits_u \mathbf{H}(x,u,v^{(i)*}_x(x),v^{(i)*}_{xx}(x)),\] and otherwise let \(u^{(i+1)}_0(x) = u^{(i)*}\). Denote as \(u^{(i+1)}_\tau\) and \(v^{(i+1)\tau}\) the policy and value following 16 \((\)or any other policy iteration algorithm\()\) and ?? with initial values \(u^{(i+1)}_0(x)\) and \(v^{(i)*}\), and the limits are denoted as \(u^{(i+1)*}\) and \(v^{(i+1)*}\).
According to the above definition, if the limit of the policy iteration \(u^{(i)*}\) is the maximizer of the Hamiltonian corresponding to the limiting value \(v^{(i)*}\) for every \(x\) (so that it is the optimal control), then the iteration simply stops. If not, then the optimality fails, because we can choose an initial value \(u^{(i+1)}_0\) which gives a higher value of the same Hamiltonian, and restart the iteration. The next proposition shows the policy improvement holds in the above sequential iterations.
Proposition 5. For any \(i\geq 1\), if \(u^{(i+1)}_0 \neq u^{(i)*}\), then \(dv^{(i+1)\tau}(x)|_{\tau = 0} >0\). In particular, if the policy iteration follows 16 with \(\lambda= 0\), then \(v^{(i+1)\tau_j} > v^{(i)\tau_i}\), where \(\tau_i,\tau_j\in[0,\infty]\).
Proof. Since \(v^{(i+1)0} = v^{(i)*}\), thus \[\begin{align} &dv^{(i+1)\tau}(x)|_{\tau = 0} \nonumber\\ =& \left(-\beta v^{(i)*}(x) + f(x,u^{(i+1)}_0) + b^\top(x,u^{(i+1)}_0) v^{(i)*}_x(x) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u^{(i+1)}_0):v^{(i)*}_{xx}(x)]\right)d\tau. \end{align}\] By Definition 5, \[\mathbf{H}(x,u^{(i+1)}_0(x),v^{(i)*}_x(x),v^{(i)*}_{xx}(x)) > \mathbf{H}(x,u^{(i)*}(x),v^{(i)*}_x(x),v^{(i)*}_{xx}(x)).\] Thus, \[dv^{(i+1)\tau}(x)|_{\tau = 0} > \left(-\beta v^{(i)*}(x) + f(x,u^{(i)*}) + b^\top(x,u^{(i)*})v^{(i)*}_x(x) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u^{(i)*}):v^{(i)*}_{xx}]\right)d\tau.\] On the other hand, since \((u^{(i)*}, v^{(i)*})\) is the limit of the \(i\)-th round of iteration, \[-\beta v^{(i)*}(x) + f(x,u^{(i)*}) + b^\top(x,u^{(i)*})v^{(i)*}_x(x) + \frac{1}{2}\mathrm{Tr}[\sigma\sigma^\top(x,u^{(i)*}):v^{(i)*}_{xx}] = 0,\] which implies that \(dv^{(i+1)\tau}(x)|_{\tau = 0} >0\). If the policy iteration follows 16 with \(\lambda=0\), Lemma 4 implies that \(v^{(i)\tau_1} \leq v^{(i)\tau_2}\) for any \(i\geq 1\) and \(\tau_1 < \tau_2 \leq \infty\). Thus in order to prove the claim, it suffices to check \(dv^{(i+1)\tau}>0\) at \(\tau = 0\) as above, because that is the only instance where we manually choose the drift, in particular the control \(u\) in the Hamiltonian \(\mathbf{H}\), of the value iteration, which otherwise should be the current iterates for \(u\), according to 16 . ◻
In this section, we focus on the well-posedness of the policy dynamics 4 coupled with the value function \(v^{\tau}\) introduced in ?? for the relaxed control problem. Notice that the dynamics of \(v^\tau\) in ?? is not a definition, but a result derived in Lemma 1. \(v^\tau\) is defined in ?? , as the value of the stochastic control problem corresponding to the policy \(\rho^\tau\). Thus instead of the well-posedness of 24 , which we use to analyze the convergence rate of our iteration method, we focus on the couple dynamics 4 ??
By its definition in ?? , the value \(v^{\tau}\) depends on the policy \(\rho^{\tau}\) through the entropy-regularized source \(\bar f_{\rho}=\tilde{f}(\cdot,\rho)-\lambda\mathcal{H}_{\rho}\), where \(\mathcal{H}_{\rho}(x)=\int_{\mathbb{R}^{n}}\rho(x,u)\ln\rho(x,u)\,du\) denotes the differential entropy of the policy density. As we show in Lemma 7(b) below, the coefficient maps \(\rho\mapsto(\bar a_{\rho},\bar b_{\rho},\bar f_{\rho})\) are Lipschitz in the policy at the level of \(C^{0}_{b}\); however, the Schauder theory underlying our value estimates requires control of the coefficients in the Hölder space \(C^{0,\alpha}_{b}\), and at that level the dependence on the policy — obtained by interpolation — is only Hölder-continuous with exponent \(\theta<1\). A contraction argument is therefore unavailable, and we construct solutions by a compactness (Schauder fixed point) argument instead. This in turn requires the coefficients to be Hölder-regular in the state variable, which motivates the following function spaces and assumptions.
Definition 6 (Hölder space). For \(\alpha\in(0,1]\), let \(C^{0,\alpha}_{b}(\mathbb{R}^{d})\) be the Banach space of bounded, \(\alpha\)-Hölder continuous functions \(g:\mathbb{R}^{d}\to\mathbb{R}\), equipped with the norm \[\|g\|_{C^{0,\alpha}_{b}}:=\|g\|_{\infty}+[g]_{\alpha}, \qquad [g]_{\alpha}:=\sup_{x\neq x'}\frac{|g(x)-g(x')|}{|x-x'|^{\alpha}} .\] For vector- or matrix-valued maps the norm is applied componentwise, and a function \(h(x,u)\) is said to be \(C^{0,\alpha}_{b}\) in \(x\), uniformly in \(u\), if \(\sup_{u}\|h(\cdot,u)\|_{C^{0,\alpha}_{b}}<\infty\).
Assumption 6. (i) For some \(\alpha'\in(0,1)\), the coefficients \(f,b,\sigma\) are \(C^{0,\alpha'}_{b}\) in \(x\), uniformly in \(u\). Throughout this section we fix \(\alpha\in(0,\alpha')\) and set \(\theta:=1-\alpha/\alpha'\in(0,1)\).
(ii) There exist \(c_{0}>0\), \(r_{0}\ge0\) and a nondecreasing function \(\Lambda\) such that, for
all \(x\in\mathbb{R}^{d}\), \(|u|\ge r_{0}\), and all \((p,q)\), \[\big\langle\nabla_{u}\mathbf{H}(x,u,p,q),\,u\big\rangle
\le -c_{0}|u|^{2}+\Lambda\big(|p|+|q|\big).\]
Next, we introduce the density class \(\mathcal{D}_{M}\) on which the fixed point will be constructed in the following argument for existence. Such weighted density classes with two-sided (Gaussian) bounds are of Aronson type, see Aronson1967?, FabesStroock1986?; we refer to BogachevKRS2015? for a comprehensive treatment of density regularity for Fokker–Planck–Kolmogorov equations. From now on, we denote by \(\mathcal{P}_{2}(\mathbb{R}^{n})\) the space of Borel probability measures with finite second moment, equipped with the \(2\)-Wasserstein distance \[\begin{align} \mathcal{W}_{2}(\mu,\nu) =\Big(\inf_{\gamma\in\Pi(\mu,\nu)} \int_{\mathbb{R}^{n}\times\mathbb{R}^{n}}|u-u'|^{2}\,\gamma(du,du')\Big)^{1/2}, \end{align}\] where \(\Pi(\mu,\nu)\) denotes the set of couplings of \(\mu,\nu\); see Villani2009?, AmbrosioGigliSavare2008?.
Definition 7. Fix \(0<\alpha<\alpha'<1\), \(\delta>0\), \(M\ge1\), \(\kappa>0\), and write \(\langle u\rangle^{2}=1+|u|^{2}\), \(\langle x\rangle^{2}=1+|x|^{2}\), \(\|g\|_{L^{1}_{2}}:=\int\langle u\rangle^{2}|g(u)|\,du\). Let \(\mathcal{D}_{M}\) be the set of policy fields \(x\mapsto\rho(x,\cdot)\) from \(\mathbb{R}^{d}\) to \(\mathcal{P}_{2}(\mathbb{R}^{n})\) such that each \(\rho(x,\cdot)\) is absolutely continuous with density \(\rho(x,u)\) satisfying, for all \(x,x'\in\mathbb{R}^{d}\), \[\begin{align} &\text{(i)}\;\int_{\mathbb{R}^{n}}\langle u\rangle^{2+\delta}\rho(x,u)\,du\le M, \qquad \text{(ii)}\;|\ln\rho(x,u)|\le M\langle u\rangle^{2},\\ &\text{(iii)}\;\|\rho(x,\cdot)-\rho(x',\cdot)\|_{L^{1}_{2}}\le M|x-x'|^{\alpha'}, \qquad \text{(iv)}\;\|\rho(x,\cdot)\|_{H^{1}(\mathbb{R}^{n})}\le M, \end{align}\] where \(H^{1}(\mathbb{R}^{n})\) is the usual Sobolev space, \(\|g\|_{H^{1}}^{2}=\|g\|_{L^{2}}^{2}+\|\nabla g\|_{L^{2}}^{2}\). We equip \(\mathcal{D}_{M}\) with the weighted* metric \[\label{eq:dkappa} \mathsf d_{\kappa}(\rho,\rho') :=\sup_{x\in\mathbb{R}^{d}}\langle x\rangle^{-\kappa} \,\|\rho(x,\cdot)-\rho'(x,\cdot)\|_{L^{1}_{2}} .\tag{18}\] *
Lemma 5. For every \(\kappa>0\), the set \(\mathcal{D}_{M}\) is convex and \((\mathcal{D}_{M},\mathsf d_{\kappa})\) is compact.
Proof. The convexity is immediate: (i), (iii) and (iv) from Definition 7 are stable under convex combinations, and for (ii) note that a mixture of \(\rho\) and \(\rho'\) is bounded above by \(e^{M\langle u\rangle^{2}}\), while the lower bound follows from concavity of the logarithm. As for the compactness. Let \((\rho_{k})\subset\mathcal{D}_{M}\). By (i), \(\|\rho_{k}(x,\cdot)-\rho_{j}(x,\cdot)\|_{L^{1}_{2}}\le 2M\) for every \(x\), so the region \(\{|x|>R\}\) contributes at most \(2M\langle R\rangle^{-\kappa}\) to \(\mathsf d_{\kappa}\). It therefore suffices to extract a subsequence converging uniformly on each ball \(B_{R}\) and conclude by a diagonal argument. Fix \(R\), for \(x\in B_{R}\) the densities \(\rho_{k}(x,\cdot)\) lie in a fixed relatively compact subset of \(L^{1}_{2}\): the \(H^{1}\) bound (iv) gives equicontinuity under translations in \(u\), the moment bound (i) prevents mass from escaping to \(|u|=\infty\) in the weighted norm, and the Fréchet-Riesz-Kolmogorov theorem applies, see hancheolsen2010kolmogorov? for a standard reference. Since (iii) provides equicontinuity in \(x\), Arzelà–Ascoli yields a subsequence converging uniformly on \(B_{R}\), as required. It remains to check that the limit \(\rho\) belongs to \(\mathcal{D}_{M}\). Passing to a further subsequence, \(\rho_{k}(x,\cdot)\to\rho(x,\cdot)\) a.e., so (i) and (ii) follow from Fatou’s lemma, (iii) from convergence of the \(L^{1}_{2}\)-norms, and (iv) from weak lower semicontinuity of the \(H^{1}\)-norm. ◻
The next lemma shows that on \(\mathcal{D}_{M}\) the weighted metric \(\mathsf d_{\kappa}\) and the transport metric \[\label{def:weighted32W} \mathcal{W}_{2,\kappa}(\rho,\rho') :=\sup_{x}\langle x\rangle^{-\kappa} \mathcal{W}_{2}\big(\rho(x,\cdot),\rho'(x,\cdot)\big)\tag{19}\] are Hölder equivalent.
Lemma 6. Under conditions (i)–(iv) of Definition 7, there exist a constant \(C_{M}>0\) and an exponent \(\gamma=\dfrac{2\delta}{\,n+8+4\delta\,}\in(0,1)\) such that, for all \(\rho,\rho'\in\mathcal{D}_{M}\), \[\label{eq:metric-equiv} \mathcal{W}_{2,\kappa}(\rho,\rho')\le \sqrt{2}\,\mathsf d_{\kappa}(\rho,\rho')^{1/2}, \qquad \mathsf d_{\kappa}(\rho,\rho')\le C_{M}\,\mathcal{W}_{2,\kappa}(\rho,\rho')^{\gamma}.\qquad{(6)}\] In particular \(\mathsf d_{\kappa}\) and \(\mathcal{W}_{2,\kappa}\) induce the same topology on \(\mathcal{D}_{M}\).
Proof. We first prove the pointwise (unweighted) inequalities \[\label{eq:metric-equiv-ptw} \mathcal{W}_{2}(\mu,\nu)\le\sqrt{2}\,\|\mu-\nu\|_{L^{1}_{2}}^{1/2}, \qquad \|\mu-\nu\|_{L^{1}_{2}}\le C_{M}\,\mathcal{W}_{2}(\mu,\nu)^{\gamma},\tag{20}\] for \(\mu=\rho(x,\cdot)\), \(\nu=\rho'(x,\cdot)\) at a fixed \(x\); all constants are uniform in \(x\) and may depend on \(M,n,\delta\).
Forward bound. Let \(\eta=\mu\wedge\nu\) be the overlap, and \(\mu_{1}=(\mu-\nu)_{+}\), \(\nu_{1}=(\nu-\mu)_{+}\) the positive and negative parts of \(\mu-\nu\), respectively. Since \(\mu,\nu\) are both probability measures, \(\mu_{1}\) and \(\nu_{1}\) have equal mass \(m=\tfrac12\int_{\mathbb{R}^{n}}|\mu-\nu|\,du\), and \(\int_{\mathbb{R}^{n}}\eta\,du=1-m\). If \(m=0\) then \(\mu=\nu\). Otherwise \(\gamma_{0}=(\mathrm{id},\mathrm{id})_{\#}\eta+\tfrac1m\mu_{1}\otimes\nu_{1}\) has marginals \(\mu,\nu\), and by the definition of \(\mathcal{W}_{2}\), \[\begin{align} \mathcal{W}_{2}^{2}(\mu,\nu) &\le\int_{\mathbb{R}^{n}}\int_{\mathbb{R}^{n}}|u-v|^{2}\,d\gamma_{0} =\frac{1}{m}\int_{\mathbb{R}^{n}}\int_{\mathbb{R}^{n}}|u-v|^{2}\,\mu_{1}(du)\nu_{1}(dv)\\ &\le\frac{2}{m}\int_{\mathbb{R}^{n}}\int_{\mathbb{R}^{n}} \big(|u|^{2}+|v|^{2}\big)\mu_{1}(du)\nu_{1}(dv) =2\!\int_{\mathbb{R}^{n}}|u|^{2}\,(\mu_{1}+\nu_{1})\,du . \end{align}\] Since \(\mu_{1}+\nu_{1}=|\mu-\nu|\), \[\mathcal{W}_{2}^{2}(\mu,\nu)\le 2\!\int_{\mathbb{R}^{n}}|u|^{2}|\mu-\nu|\,du \le 2\!\int_{\mathbb{R}^{n}}\langle u\rangle^{2}|\mu-\nu|\,du=2\,\|\mu-\nu\|_{L^{1}_{2}},\] which is the first inequality in 20 .
Reverse bound. Set \(W=\mathcal{W}_{2}(\mu,\nu)\). If \(W=0\) then \(\mu=\nu\), and since \(\|\mu-\nu\|_{L^{1}_{2}}\le 2M\) by (i), the case \(W>1\) holds with \(C_{M}=2M\). Thus we focus on \(0<W\le1\). The difficulty is that \(\|\mu-\nu\|_{L^{1}_{2}}\) is a strong distance while \(W\) is a weak one: two densities can be \(\mathcal{W}_{2}\)-close yet far apart in \(L^{1}\), either by spreading mass to large \(|u|\), where the weight \(\langle u\rangle^{2}\) amplifies it, or by oscillating rapidly in \(u\). We control the first effect by the moment bound (i) and the second by the Sobolev bound (iv), through a truncation radius \(R\ge1\) and a mollification scale \(t>0\) fixed at the end. We now separate the bound as \(\|\mu-\nu\|_{L^{1}_{2}}=I_{R}+II_{R}\) with \(I_{R}=\int_{|u|\le R}\langle u\rangle^{2}|\mu-\nu|\) and \(II_{R}=\int_{|u|>R}\langle u\rangle^{2}|\mu-\nu|\). On \(\{|u|>R\}\), one has \(\langle u\rangle^{2}\le R^{-\delta}\langle u\rangle^{2+\delta}\), so (i) gives \[II_{R}\le R^{-\delta}\!\int_{\mathbb{R}^{n}}\langle u\rangle^{2+\delta}(\mu+\nu)\,du \le 2M\,R^{-\delta}.\] On the ball, \(\langle u\rangle^{2}\le 2R^{2}\), so \(I_{R}\le 2R^{2}\|\mu-\nu\|_{L^{1}(B_{R})}\), and it remains to bound the unweighted mass \(\|\mu-\nu\|_{L^{1}(B_{R})}\) by \(W\). Mollifying with the heat semigroup \(P_{t}=e^{t\Delta_{u}}\) in the control variable — a purely auxiliary smoothing over length scale \(\sqrt t\), with \(t\) unrelated to the relaxation time — we write \[\|\mu-\nu\|_{L^{1}(B_{R})} \le\|P_{t}(\mu-\nu)\|_{L^{1}}+\|(I-P_{t})(\mu-\nu)\|_{L^{1}(B_{R})},\] where the smoothed part is controlled by transport and the remainder by the \(H^{1}\) bound. We use two standard heat-semigroup facts for \(P_{t}=e^{t\Delta}\) with kernel \(p_{t}\): \(\|\nabla p_{t}\|_{L^{1}}\le C_{n}t^{-1/2}\), and \(1-e^{-t|\xi|^{2}}\le\sqrt{t}\,|\xi|\), whence \(\|(I-P_{t})g\|_{L^{2}}\le\sqrt{t}\,\|\nabla g\|_{L^{2}}\). Indeed, by \(L^{1}\)–\(L^{\infty}\) duality, self-adjointness of \(P_{t}\), the bound \(\mathrm{Lip}(P_{t}\phi)\le\|\nabla p_{t}\|_{L^{1}}\le C_{n}t^{-1/2}\) for \(\|\phi\|_{\infty}\le1\), the Kantorovich–Rubinstein duality, and \(\mathcal{W}_{1}\le\mathcal{W}_{2}\), \[\|P_{t}(\mu-\nu)\|_{L^{1}} =\sup_{\|\phi\|_{\infty}\le1}\int_{\mathbb{R}^{n}}(P_{t}\phi)\,d(\mu-\nu) \le C_{n}t^{-1/2}\,\mathcal{W}_{1}(\mu,\nu)\le C_{n}t^{-1/2}W;\] and by Cauchy–Schwarz, \(\|(I-P_{t})g\|_{L^{2}}\le\sqrt t\,\|\nabla g\|_{L^{2}}\), and assumption (iv) for \(\mu\) and \(\nu\), \[\|(I-P_{t})(\mu-\nu)\|_{L^{1}(B_{R})} \le|B_{R}|^{1/2}\|(I-P_{t})(\mu-\nu)\|_{L^{2}} \le C_{n,M}R^{n/2}\sqrt t .\] The two terms balance at \(t=WR^{-n/2}\), giving \[\|\mu-\nu\|_{L^{1}(B_{R})} \le C_{n,M}\big(t^{-1/2}W+R^{n/2}\sqrt t\big)\Big|_{t=WR^{-n/2}} =C_{n,M}\,R^{n/4}W^{1/2}.\] Combining bulk and tail, \(\|\mu-\nu\|_{L^{1}_{2}}\le C_{n,M}R^{2+n/4}W^{1/2}+2MR^{-\delta}\), which for \(0<W\le1\) is optimized at \(R=W^{-1/(2(2+n/4+\delta))}\ge1\) and yields the second inequality in 20 with \[\gamma=\frac{\delta}{2\big(2+\tfrac n4+\delta\big)} =\frac{2\delta}{\,n+8+4\delta\,}\in(0,1).\]
In the end, we show the inequalities for the weighted norm. Since \(\langle x\rangle\ge1\) and the exponents \(1/2\), \(\gamma\) lie in \((0,1]\), we have \(\langle x\rangle^{-\kappa}\le \langle x\rangle^{-\kappa\beta'}\) for any \(\beta'\in(0,1]\). Multiplying the pointwise inequalities 20 by \(\langle x\rangle^{-\kappa}\) therefore gives \(\langle x\rangle^{-\kappa} W\le\sqrt2\, (\langle x\rangle^{-\kappa}\|\mu-\nu\|_{L^{1}_{2}})^{1/2}\) and \(\langle x\rangle^{-\kappa}\|\mu-\nu\|_{L^{1}_{2}}\le C_{M}(\langle x\rangle^{-\kappa} W)^{\gamma}\) (for the case \(W>1\) use \(\|\mu-\nu\|_{L^{1}_{2}}\le 2M\) and \(\langle x\rangle^{-\kappa}\le \langle x\rangle^{-\kappa*\gamma}\) for \(\langle x\rangle^{-\kappa}\in[0,1]\)). Taking suprema over \(x\) yields ?? . ◻
Recall our relaxed dynamics defined in 1 , and for \(\rho\in\mathcal{D}_{M}\), in the notation \(\tilde{b}(x,\pi)=\int b(x,u)\pi(u)\,du\) and \(\tilde{\sigma}(x,\pi)=\big(\int\sigma\sigma^{\top}(x,u)\pi(u)\,du\big)^{1/2}\), we denote \[\bar b_{\rho}(x)=\tilde{b}(x,\rho(x,\cdot)),\qquad \bar a_{\rho}(x)=\tilde{\sigma}\tilde{\sigma}^{\top}(x,\rho(x,\cdot)) =\int\sigma\sigma^{\top}(x,u)\rho(x,u)\,du .\] Similarly, for the averaged cost \(\tilde{f}(x,\rho)=\int f(x,u)\rho(x,u)\,du\), we denote \[\begin{align} \bar f_{\rho}=\tilde{f}(\cdot,\rho)-\lambda\mathcal{H}_{\rho}, \qquad\text{with}\qquad \mathcal{H}_{\rho}(x)=\int\rho(x,u)\ln\rho(x,u)\,du . \end{align}\] The corresponding generator associated with 1 and \(\rho\in\mathcal{D}_{M}\) is denoted as \(\bar{\mathcal{L}}_{\rho}=\tfrac12\operatorname{tr}(\bar a_{\rho}\nabla^{2}) +\bar b_{\rho}^{\top}\nabla\). By Assumption 1, \(\bar a_{\rho}\succeq\theta_{0}I_{d}\) for some constant \(\theta_{0}>0\), so \(\tilde{\sigma}(\cdot,\rho)\) is well-defined and uniformly non-degenerate. We then have the following entropy regularity estimates.
Lemma 7. Let Assumptions 1, 6 hold and \(\lambda>0\). Then:
for \(\rho\in\mathcal{D}_{M}\), \(\|\mathcal{H}_{\rho}\|_{C^{0,\alpha'}_{b}}\le C_{M}\), hence \(\bar a_{\rho},\bar b_{\rho},\bar f_{\rho}\in C^{0,\alpha'}_{b}\) with norm \(\le C_{M}\) and \(\bar a_{\rho}\succeq\theta_{0}I_{d}\);
\(\rho\mapsto(\bar a_{\rho},\bar b_{\rho},\mathcal{H}_{\rho},\bar f_{\rho})\) are Lipschitz pointwise in \(x\): for all \(\rho,\rho'\in\mathcal{D}_{M}\) and all \(x\), \[\label{eq:coeff-ptw-lip} |\bar f_{\rho}(x)-\bar f_{\rho'}(x)| +|\bar a_{\rho}(x)-\bar a_{\rho'}(x)| +|\bar b_{\rho}(x)-\bar b_{\rho'}(x)| \le C_{M}\,\|\rho(x,\cdot)-\rho'(x,\cdot)\|_{L^{1}_{2}} .\qquad{(7)}\] In particular, they are \(\mathsf d_{\kappa}\)-Lipschitz into the \(\langle x\rangle^{-\kappa}\)-weighted supremum norm, \(\|\langle\cdot\rangle^{-\kappa}(\bar f_{\rho}-\bar f_{\rho'})\|_{\infty} \le C_{M}\,\mathsf d_{\kappa}(\rho,\rho')\). Moreover, with \(\mathsf d_{\infty}(\rho,\rho'):=\sup_{x}\|\rho(x,\cdot)-\rho'(x,\cdot)\|_{L^{1}_{2}}\), interpolation with (a) gives the \(\theta\)-Hölder bound \(\|\bar f_{\rho}-\bar f_{\rho'}\|_{C^{0,\alpha}_{b}} \le C_{M}\,\mathsf d_{\infty}(\rho,\rho')^{\theta}\);
if \(\pi_0\in\mathcal{D}_{M_{0}}\) and, for each \(x\), \(u^{x}_{\tau}\) solves ?? \[du^x_\tau =G(\tau, x,u^x_\tau)d\tau + \sqrt{2\lambda}dB_\tau, \quad u_0 \sim \pi_0,\label{eq:DynU-G}\qquad{(8)}\] with a drift \(G(\tau,x,u)\) bounded in \(C^{0,\alpha'}_{b}\) (in \(x\)) and Lipschitz with linear growth (in \(u\)), uniformly in \((\tau,x)\), then for every \(T>0\), \(\mathrm{Law}(u^{\cdot}_{\tau})\in\mathcal{D}_{M}\) for all \(\tau\le T\), with \(M=M(T)\).
Proof. (a). We first bound the entropy term. Since \(\langle u\rangle^{2}\le\langle u\rangle^{2+\delta}\), condition (i) of Definition 7 gives the second-moment bound \(\int\langle u\rangle^{2}\rho(x,u)\,du\le M\), and combined with the two-sided bound (ii), \(|\ln\rho|\le M\langle u\rangle^{2}\), \[|\mathcal{H}_{\rho}(x)|=\Big|\int\rho\ln\rho\,du\Big| \le\int\rho\,|\ln\rho|\,du \le M\!\int\langle u\rangle^{2}\rho\,du\le M^{2}.\] For the \(x\)-modulus, apply the mean-value identity \(t\ln t-t'\ln t'=(\ln\xi+1)(t-t')\) with \(\xi\) between \(t=\rho(x,u)\) and \(t'=\rho(x',u)\); by (ii), \(|\ln\xi+1|\le M\langle u\rangle^{2}+1\), so (iii) yields \[|\mathcal{H}_{\rho}(x)-\mathcal{H}_{\rho}(x')| \le\int(M\langle u\rangle^{2}+1)\,|\rho(x,\cdot)-\rho(x',\cdot)|\,du \le(M+1)\,\|\rho(x,\cdot)-\rho(x',\cdot)\|_{L^{1}_{2}} \le C_{M}|x-x'|^{\alpha'}.\] Hence \(\mathcal{H}_{\rho}\in C^{0,\alpha'}_{b}\) with norm \(\le C_{M}\). The same computation, with the \(C^{0,\alpha'}_{b}\)-in-\(x\), bounded-in-\(u\) maps \(\sigma\sigma^{\top},b,f\) (Assumption 6) in place of \(\ln\rho+1\), gives \(\bar a_{\rho},\bar b_{\rho},\tilde{f}(\cdot,\rho)\in C^{0,\alpha'}_{b}\) with norm \(\le C_{M}\), and therefore \(\bar f_{\rho}=\tilde{f}(\cdot,\rho)-\lambda\mathcal{H}_{\rho}\in C^{0,\alpha'}_{b}\). Finally, averaging \(\sigma\sigma^{\top}\succeq\theta_{0}I_{d}\) against \(\rho\) preserves the inequality, so \(\bar a_{\rho}\succeq\theta_{0}I_{d}\).
(b). Fix \(\rho,\rho'\in\mathcal{D}_{M}\) and \(x\in\mathbb{R}^{d}\). The same bracketing as in (a), now comparing \(\rho\) and \(\rho'\) at the common point \(x\), gives \(|\mathcal{H}_{\rho}(x)-\mathcal{H}_{\rho'}(x)| \le(M+1)\|\rho(x,\cdot)-\rho'(x,\cdot)\|_{L^{1}_{2}}\), and likewise for \(\bar a,\bar b,\tilde{f}(\cdot,\rho)\), which implies ?? . Multiplying by \(\langle x\rangle^{-\kappa}\) and taking the supremum over \(x\) gives the weighted Lipschitz bound. For the Hölder upgrade, set \(g=\bar f_{\rho}-\bar f_{\rho'}\), which by (a) is bounded in \(C^{0,\alpha'}_{b}\) by \(2C_{M}\) while \(\|g\|_{C^{0}_{b}}\le C_{M}\mathsf d_{\infty}(\rho,\rho')\) by ?? . The interpolation inequality \(\|g\|_{C^{0,\alpha}_{b}}\le C\,\|g\|_{C^{0}_{b}}^{1-\alpha/\alpha'} \|g\|_{C^{0,\alpha'}_{b}}^{\alpha/\alpha'}\) krylov1996lectures? then gives \(\|g\|_{C^{0,\alpha}_{b}}\le C_{M}\mathsf d_{\infty}(\rho,\rho')^{\theta}\).
(c). The marginal \(\pi_{\tau}(x,\cdot)=\mathrm{Law}(u^{x}_{\tau})\) solves the Fokker–Planck equation \[\partial_{\tau}\pi^{\tau} =\lambda\Delta_{u}\pi^{\tau}-\mathrm{div}_{u}\big(G\,\pi^{\tau}\big),\] which is uniformly parabolic in \(u\) (since \(\lambda>0\)), with a drift \(G\) of linear growth in \(u\) and bounded in \(C^{0,\alpha'}_{b}\) in \(x\). We check that the four defining conditions of \(\mathcal{D}_{M}\) are preserved on \([0,T]\), starting from \(\pi_0 = \mathrm{Law}(u_{0})\in\mathcal{D}_{M_{0}}\). The two-sided bound (ii) propagates by Aronson’s Gaussian estimates for non-degenerate parabolic equations Aronson1967?, Sheu1991?. The \((2+\delta)\)-moment (i) follows from Itô’s formula applied to \(\langle u^{x}_{\tau}\rangle^{2+\delta}\) and Grönwall, using the linear growth of \(G\). The spatial modulus (iii) follows from the weighted-\(L^{1}\) estimate for the equation satisfied by the difference \(\rho^{\tau}(x,\cdot)-\rho^{\tau}(x',\cdot)\), whose source is controlled by the \(C^{0,\alpha'}_{b}\) regularity of \(G\). The Sobolev bound (iv) follows from the standard parabolic energy estimate, which preserves — and for \(\tau>0\) improves — \(\|\pi_{\tau}(x,\cdot)\|_{H^{1}}\), with constants depending only on \(\|G\|_{C^{0,\alpha'}_{b}}\) and \(T\). Hence \(\pi_{\tau}\in\mathcal{D}_{M}\) for all \(\tau\le T\), with \(M=M(T,M_{0})\). In particular, under Assumption 6 (ii), the dissipative condition of \(\nabla_u\mathbf{H}\) caps its moments uniformly, independent of \(M_0\), and thus \(M=M(T)\). ◻
We now turn to the value functional. For a policy flow \(\boldsymbol{\rho}=(\rho_{s})_{s\ge0}\) we write \(v[\boldsymbol{\rho}](t,x)\) for the value defined by ?? .
Lemma 8 (Value functional). Let Assumptions 1, 6 hold and \(\beta>0\). For a measurable policy flow \(\boldsymbol{\rho}=(\rho_{s})_{s\ge0}\) with \(\rho_{s}\in\mathcal{D}_{M}\) for all \(s\), the value \(v[\boldsymbol{\rho}](t,x)\) defined by ?? is the unique bounded classical solution of \((\)see 7 \()\) \[\label{eq:backward-relaxed} \partial_{t}v+\bar{\mathcal{L}}_{\rho_{t}}v-\beta v+\bar f_{\rho_{t}}=0, \qquad \bar{\mathcal{L}}_{\rho_{t}} =\tfrac12\operatorname{tr}(\bar a_{\rho_{t}}\nabla^{2}) +\bar b_{\rho_{t}}^{\top}\nabla,\qquad{(9)}\] and satisfies the \(t\)-uniform bound \[\label{eq:v-bound-par} \sup_{t\ge0}\big\|v[\boldsymbol{\rho}](t,\cdot)\big\|_{C^{2,\alpha'}_{b}} \le C_{M}.\qquad{(10)}\] Moreover, the value depends continuously on the policy in the weighted topology: for flows \(\boldsymbol{\rho},\boldsymbol{\rho}'\) valued in \(\mathcal{D}_{M}\), writing \(\bar{\mathsf d}_{\kappa} :=\sup_{s\ge0}\mathsf d_{\kappa}(\rho_{s},\rho'_{s})\), \[\label{eq:v-cont-weighted} \sup_{t\ge0}\big\|\langle\cdot\rangle^{-\kappa} \big(v[\boldsymbol{\rho}](t,\cdot)-v[\boldsymbol{\rho}'](t,\cdot)\big)\big\|_{\infty} \le C_{M,\beta,\kappa}\,\bar{\mathsf d}_{\kappa},\qquad{(11)}\] and, with \(\lambda_{*}=\frac{2+\alpha}{2+\alpha'}\in(0,1)\), the locally uniform \(C^{2,\alpha}\) estimate \[\label{eq:v-cont-local} \sup_{t\ge0}\;\sup_{x\in\mathbb{R}^{d}}\;\langle x\rangle^{-\kappa} \big\|v[\boldsymbol{\rho}](t,\cdot)-v[\boldsymbol{\rho}'](t,\cdot) \big\|_{C^{2,\alpha}(B_{1}(x))} \le C_{M,\beta,\kappa}\,\bar{\mathsf d}_{\kappa}^{\,1-\lambda_{*}} .\qquad{(12)}\]
Proof. Proof of ?? . \(v^\tau\) defined in ?? is the Feynman–Kac representation of ??; boundedness of the running cost (Lemma 7(a)) and \(\beta>0\) make it the unique bounded classical solution, by the maximum principle for parabolic equations friedman1964parabolic?. By Assumption 1 and Lemma 7(a), ?? is uniformly parabolic with coefficients bounded in \(C^{0,\alpha'}_{b}\) uniformly in \(t\). The global parabolic Schauder estimate krylov1996lectures?, combined with the exponential damping furnished by \(-\beta\), gives \(\sup_{t}\|v(t,\cdot)\|_{C^{2,\alpha'}_{b}}\le C_{M}\) (see also lunardi1995analytic?).
Proof of ?? . The difference \(w:=v[\boldsymbol{\rho}]-v[\boldsymbol{\rho}']\) solves ?? (with policy \(\boldsymbol{\rho}\)) with the additional source \[g_{t}=(\bar f_{\rho_{t}}-\bar f_{\rho'_{t}}) +(\bar{\mathcal{L}}_{\rho_{t}}-\bar{\mathcal{L}}_{\rho'_{t}}) v[\boldsymbol{\rho}'] .\] By the pointwise Lipschitz bound ?? of Lemma 7(b) and the uniform bound ?? applied to \(v[\boldsymbol{\rho}']\), \[\label{eq:source-ptw} |g_{t}(x)| \le C_{M}\big(1+\|v[\boldsymbol{\rho}'](t,\cdot)\|_{C^{2}_{b}}\big) \|\rho_{t}(x,\cdot)-\rho'_{t}(x,\cdot)\|_{L^{1}_{2}} \le C_{M}\,\langle x\rangle^{\kappa}\,\bar{\mathsf d}_{\kappa} \qquad\text{for all }(t,x).\tag{21}\] Let \(X^{t,x}\) denote the diffusion generated by \(\bar{\mathcal{L}}_{\rho_{\cdot}}\) started at \(x\) at time \(t\). Since \(\bar b_{\rho},\bar a_{\rho}\) are bounded uniformly over \(\mathcal{D}_{M}\) (Lemma 7(a)), the Burkholder–Davis–Gundy inequality gives the polynomial moment estimate \[\label{eq:X-moment} \mathbb{E}\big[\langle X^{t,x}_{t+s}\rangle^{\kappa}\big] \le C_{\kappa}\big(\langle x\rangle^{\kappa}+1+s^{\kappa}\big), \qquad s\ge0 .\tag{22}\] The Feynman–Kac representation of \(w\), \(w(t,x)=\mathbb{E}\int_{0}^{\infty}e^{-\beta s}g_{t+s}(X^{t,x}_{t+s})\,ds\), together with 21 –22 , yields \[|w(t,x)| \le C_{M}\,\bar{\mathsf d}_{\kappa} \int_{0}^{\infty}e^{-\beta s}\, \mathbb{E}\big[\langle X^{t,x}_{t+s}\rangle^{\kappa}\big]\,ds \le C_{M,\beta,\kappa}\,\langle x\rangle^{\kappa}\,\bar{\mathsf d}_{\kappa},\] since the polynomial growth in \(s\) in 22 is integrable against \(e^{-\beta s}\) for every \(\beta>0\). This implies ?? .
Proof of ?? . Fix \(x\). By the interpolation inequality on balls krylov1996lectures?, \[\|w(t,\cdot)\|_{C^{2,\alpha}(B_{1}(x))} \le C\,\|w(t,\cdot)\|_{C^{0}(B_{2}(x))}^{1-\lambda_{*}} \|w(t,\cdot)\|_{C^{2,\alpha'}(B_{2}(x))}^{\lambda_{*}}, \qquad \lambda_{*}=\tfrac{2+\alpha}{2+\alpha'} .\] On \(B_{2}(x)\) one has \(\langle y\rangle^{\kappa}\le C\langle x\rangle^{\kappa}\), the pointwise form of ?? , \(|w(t,y)|\le C_{M,\beta,\kappa}\langle y\rangle^{\kappa}\bar{\mathsf d}_{\kappa}\), yields \(\|w(t,\cdot)\|_{C^{0}(B_{2}(x))} \le C_{M,\beta,\kappa}\,\langle x\rangle^{\kappa}\,\bar{\mathsf d}_{\kappa}\), and ?? bounds the second factor by \((2C_{M})^{\lambda_{*}}\). Hence \(\|w(t,\cdot)\|_{C^{2,\alpha}(B_{1}(x))} \le C\,\langle x\rangle^{\kappa(1-\lambda_{*})} \bar{\mathsf d}_{\kappa}^{\,1-\lambda_{*}}\), and since \(\langle x\rangle^{-\kappa}\le\langle x\rangle^{-\kappa(1-\lambda_{*})}\), ?? follows. ◻
We now state and prove the main result of this section. Since the value depends on the policy only Hölder-continuously at the level of the Hölder norms (Lemma 8), the policy-update map cannot be expected to be a contraction. We therefore obtain a fixed point from Schauder’s theorem.
Proposition 6 (Global existence). Let Assumptions 1, 6 hold, \(\beta,\lambda>0\), and \(\mathrm{Law}(u_{0})\in\mathcal{D}_{M_{0}}\). Then the coupled system 4 –?? admits a solution \((v^{\tau},u^{x}_{\tau})_{\tau\ge0}\) with \(\rho_{\tau} = Law(u^x_\tau)\in\mathcal{D}_{M}\) and \(\sup_{\tau\ge0}\|v^{\tau}\|_{C^{2,\alpha'}_{b}}\le L<\infty\).
Proof. The proof has five steps: we (1) define a convex set of policy flows \(\mathcal{K}_{T}\) and a map \(\Phi\) on it whose fixed points are exactly the solutions of the coupled system on \([0,T]\); (2) show \(\Phi(\mathcal{K}_{T})\subseteq\mathcal{K}_{T}\); (3) show \(\mathcal{K}_{T}\) is compact; (4) show \(\Phi\) is continuous; (5) apply Schauder’s theorem and extend the solution to \([0,\infty)\) by concatenation.
Step 1 (The fixed-point set and the map \(\Phi\)). Fix \(T>0\) and let \(M=M(T,M_{0})\) be the constant from Lemma 7(c). With a constant \(C_{T}\) determined in Step 2, define \[\mathcal{K}_{T}:=\Big\{\boldsymbol{\rho}=(\rho_{\tau})_{\tau\in[0,T]} \in C([0,T];\mathcal{D}_{M}):\; \mathsf d_{\kappa}(\rho_{\tau},\rho_{\tau'}) \le C_{T}|\tau-\tau'|^{1/2}\;\;\forall\,\tau,\tau'\Big\},\] which is nonempty (it contains constant-in-\(\tau\) flows) and convex, since \(\mathcal{D}_{M}\) is convex (Lemma 5) and the time modulus is stable under convex combinations. Given \(\boldsymbol{\rho}\in\mathcal{K}_{T}\), define \(\Phi(\boldsymbol{\rho})\) in two stages:
Value stage. Define \(\bar{\boldsymbol{\rho}} = \{\bar\rho_t\}_{t\geq 0}\) by \[\bar\rho_t = \begin{cases} \rho_{T-t}, &0\leq t \leq T;\\ \rho_0, & t > T. \end{cases}\] Let \(v^{\tau}:=v[\bar{\boldsymbol{\rho}}](0,\cdot)\). By Lemma 8, \(\sup_{\tau\le T}\|v^{\tau}\|_{C^{2,\alpha'}_{b}}\le C_{M}\), and by ?? the map \(\boldsymbol{\rho}\mapsto(\nabla v^{\tau},\nabla^{2}v^{\tau})\) is continuous from \((\mathcal{K}_{T},\sup_{\tau}\mathsf d_{\kappa})\) into \(C([0,T];C^{1,\alpha}(B_{R}))\) for every \(R>0\).
Policy stage. With this \(v^{\tau}\), the drift \(G(\tau,x,u)=\nabla_{u}\mathbf{H}\big(x,u,\nabla v^{\tau}(x), \nabla^{2}v^{\tau}(x)\big)\) in ?? (which is same as 4 ) is Lipschitz with linear growth in \(u\) and bounded in \(C^{0,\alpha'}_{b}\) in \(x\), uniformly in \(\tau\le T\), by the bound in (a). Solve 4 and denote \(\Phi(\boldsymbol{\rho})_{\tau}(x,\cdot):=\mathrm{Law}(u^{x}_{\tau})\).
By construction, \(\boldsymbol{\rho}^{\star}=\Phi(\boldsymbol{\rho}^{\star})\) if and only if \((v^{\tau},(u^{x}_{\tau})_{x})_{\tau\le T}\) solves the coupled system 4 –?? on \([0,T]\).
Step 2 (Invariance: \(\Phi(\mathcal{K}_{T})\subseteq\mathcal{K}_{T}\)). \(\Phi(\boldsymbol{\rho})_{\tau}\in\mathcal{D}_{M}\) for all \(\tau\le T\) is exactly Lemma 7(c), since the drift \(G\) satisfies its hypotheses by Step 1(b). For the time modulus, write \(\tilde{\rho}_{\tau}:=\Phi(\boldsymbol{\rho})_{\tau}\) and recall from Lemma 7(c) that, for each \(x\), \(\tilde{\rho}_{\tau}(x,\cdot)\) solves the Fokker–Planck equation \(\partial_{\tau}\tilde{\rho}=\lambda\Delta_{u}\tilde{\rho} -\mathrm{div}_{u}(G\tilde{\rho})\). Integrating in time and taking the \(L^{1}_{2}\)-norm, for each fixed \(x\), \[\|\tilde{\rho}_{\tau}(x,\cdot)-\tilde{\rho}_{\tau'}(x,\cdot)\|_{L^{1}_{2}} \le\int_{\tau'}^{\tau} \big\|\lambda\Delta_{u}\tilde{\rho}_{s} -\mathrm{div}_{u}(G\tilde{\rho}_{s})\big\|_{L^{1}_{2}}\,ds \le C_{T}'\,|\tau-\tau'|^{1/2},\] since \(\mathrm{div}_{u}G\) is bounded and \(|G|\le C\langle u\rangle\), the moment bound (i) gives \(\|(\mathrm{div}_{u}G)\tilde{\rho}_{s}\|_{L^{1}_{2}}\le C_{T}\). On the other hand, since the diffusion coefficient is a constant, the parabolic estimate for \(\|\Delta_{u}\tilde{\rho}^{s}\|_{L^{1}_{2}}\) degenerates like \(s^{-1/2}\) near \(s=0\). Integrating it directly leads to an upper bound which is multiple of \(|\tau-\tau'|^{1/2}\). Since the weight satisfies \(\langle x\rangle^{-\kappa}\le1\), and \(|\tau-\tau'|\le T^{1/2}|\tau-\tau'|^{1/2}\) on \([0,T]\), taking \(\sup_{x}\) gives \(\mathsf d_{\kappa}(\tilde{\rho}^{\tau},\tilde{\rho}^{\tau'}) \le C_{T}|\tau-\tau'|^{1/2}\) with \(C_{T}:=T^{1/2}C_{T}'\). Hence \(\Phi(\boldsymbol{\rho})\in\mathcal{K}_{T}\).
Step 3 (Compactness of \(\mathcal{K}_{T}\)). By Lemma 5, \((\mathcal{D}_{M},\mathsf d_{\kappa})\) is compact, and \(\mathcal{K}_{T}\subset C([0,T];\mathcal{D}_{M})\) is equicontinuous in \(\tau\) by the uniform \(\tfrac12\)-Hölder time modulus. The Arzelà–Ascoli theorem therefore shows that \(\mathcal{K}_{T}\) is compact in the metric \(\sup_{\tau\le T}\mathsf d_{\kappa}\), and closedness of the modulus condition under this convergence is immediate.
Step 4 (Continuity of \(\Phi\)). Let \(\boldsymbol{\rho}^{k}\to\boldsymbol{\rho} \in \mathcal{K}_T\) in \(\sup_{\tau}\mathsf d_{\kappa}\). By ?? , \(\nabla v[\boldsymbol{\rho}^{k}],\nabla^{2}v[\boldsymbol{\rho}^{k}] \to\nabla v[\boldsymbol{\rho}],\nabla^{2}v[\boldsymbol{\rho}]\) uniformly on \([0,T]\times B_{R}\) for every \(R>0\). Hence, for each fixed \(x\), the frozen-\(x\) drifts \(G^{k}(\cdot,x,\cdot)\to G(\cdot,x,\cdot)\) locally uniformly, with Lipschitz and growth constants uniform in \((k,x)\). Standard stability of SDEs with respect to convergence of coefficients then gives \[\sup_{\tau\le T}\;\mathcal{W}_{2}\big(\Phi(\boldsymbol{\rho}^{k})^{\tau}(x,\cdot), \Phi(\boldsymbol{\rho})^{\tau}(x,\cdot)\big)\longrightarrow0, \qquad\text{uniformly for }x\in B_{R},\;\forall R>0 .\] Moreover \(\mathcal{W}_{2}\) is bounded on \(\mathcal{D}_{M}\): by the moment bound (i), \(\mathcal{W}_{2}(\mu,\nu)^{2}\le2\int|u|^{2}(\mu+\nu)\le4M\). Splitting \(\sup_{x}=\sup_{|x|\le R}+\sup_{|x|>R}\) and using \(\langle x\rangle^{-\kappa}\le\langle R\rangle^{-\kappa}\) on the tail, we conclude \(\sup_{\tau}\mathcal{W}_{2,\kappa}\big(\Phi(\boldsymbol{\rho}^{k})^{\tau}, \Phi(\boldsymbol{\rho})^{\tau}\big)\to0\). Since all laws involved lie in \(\mathcal{D}_{M}\) by Step 2, the Hölder equivalence of \(\mathcal{W}_{2,\kappa}\) and \(\mathsf d_{\kappa}\) on \(\mathcal{D}_{M}\) (Lemma 6) yields \(\Phi(\boldsymbol{\rho}^{k})\to\Phi(\boldsymbol{\rho})\) in \(\sup_{\tau}\mathsf d_{\kappa}\), i.e.\(\Phi\) is continuous on \(\mathcal{K}_{T}\).
Step 5 (Fixed point and global extension). By Steps 2–4, \(\Phi\) is a continuous self-map of the nonempty convex and compact set \(\mathcal{K}_{T}\), so Schauder’s fixed point theorem yields \(\boldsymbol{\rho}^{\star}=\Phi(\boldsymbol{\rho}^{\star})\), i.e.a solution of the coupled system on \([0,T]\), with \(\rho^*_{\tau}\in\mathcal{D}_{M}\) and \(\sup_{\tau\le T}\|v^{\tau}\|_{C^{2,\alpha'}_{b}}\le L:=C_{M}\) by Lemma 8. Since the constant \(L\) does not depend on \(T\), we may iterate: for each \(k\ge 1\), the terminal law \(\mathrm{Law}(u^{x}_{kT})\in\mathcal{D}_{M}\) serves as initial datum for the next window \([kT,(k+1)T]\), and the solutions concatenate into a solution on \([0,\infty)\). ◻
In this section, we discuss the convergence rate of the continuous policy-value iteration dynamics proposed in the previous sections under monotonicity conditions on the Hamiltonians.
We impose the following monotonicity condition for the value function and the corresponding Hamiltonian in the relaxed control setting as follows. Recall that \[\begin{align} \label{notation:32hamiltonian} \mathbf{H}^\tau :=\mathbf{H}^\tau(x,u)&=\mathbf{H}(x,u,v^\tau_x,v^\tau_{xx}) \nonumber \\ &= f(x,u) + b^\top(x,u)v^{\tau}_x(x)+ \frac{1}{2}\mathrm{Tr}[\sigma\sigma^{\top}(x,u):v^{\tau}_{xx}(x)]. \end{align}\tag{23}\]
Assumption 7 (Monotonicity Condition). For each \(x\in\mathbb{R}^d\), and for all \(\tau\ge 0\), we assume (omitting the argument \(x\) for simplicity) \[\begin{align} (\text{MC I}):\quad &\mathbb{E}\Big[\Big((\mathbf{H}^\tau- \lambda \ln \pi_\tau)(u_\tau)-(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)(\tilde{u}_\tau)\Big)\Big] (v^\tau-\tilde{v}^\tau)\le 0;\\ (\text{MC II}):\quad&\Big( \nabla_x \mathbb{E}\Big[\Big((\mathbf{H}^\tau- \lambda \ln \pi_\tau )(u_\tau)-(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)(\tilde{u}_\tau)\Big)\Big]\Big)^\top (v^{\tau}_x-\tilde{v}^{\tau}_x )\le 0;\\ (\text{MC III}):\quad& \mathrm{Tr}\Big( \nabla^2_{xx}\mathbb{E}\Big[(\mathbf{H}^\tau- \lambda \ln \pi_\tau )(u_\tau)-(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)(\tilde{u}_\tau)\Big]:(v^{\tau}_{xx}-\tilde{v}^{\tau}_{xx})\Big) \le 0, \end{align}\] where \((\pi_\tau,v^\tau,\mathbf{H}^\tau)\) and \((\tilde{\pi}_\tau,\tilde{v}^\tau,\tilde{\mathbf{H}}^\tau)\) are the result of policy and value iteration for the relaxed control problem, and the corresponding Hamiltonian at each \(\tau\), i.e., \[\tilde{\mathbf{H}}^\tau:= \mathbf{H}(\tau, x, u, \tilde{v}^{\tau}_x, \tilde{v}^{\tau}_{xx}) = f(x,u) + b^\top(x,u)\tilde{v}^{\tau}_x(x)+ \frac{1}{2}\mathrm{Tr}[\sigma \sigma^{\top}(x, u):\tilde{v}^{\tau}_{xx}(x)],\] starting from initial value \((u_0,v^0)\) and \((\tilde{u}_0,\tilde{v}^0)\), following the dynamics defined in Lemma 1, respectively.
In the following, we show the convergence of the probability distribution of control to its Gibbs invariant distribution for the following policy and value continuous dynamics arising from Lemma 1, \[\begin{align} \label{couple32relax32dynamics} \begin{cases} dv^\tau =&\left[\int_{\mathbb{R}^n} (\mathbf{H}(x,u,v^{\tau}_x,v^\tau_{xx}) - \lambda\ln \pi_{\tau})\pi_{\tau}(x,u)du -\beta v^{\tau} \right]d\tau,\\ dv^\tau_x =&\left[\nabla_x\int_{\mathbb{R}^n} (\mathbf{H}(x,u,v^{\tau}_x,v^\tau_{xx})- \lambda\ln \pi_\tau(x,u) )\pi_\tau(x,u)du-\beta v^\tau_x\right]d\tau,\\ dv^\tau_{xx} =&\left[\nabla^2_{xx}\int_{\mathbb{R}^n} (\mathbf{H}(x,u,v^{\tau}_x,v^\tau_{xx}) - \lambda\ln \pi_\tau(x,u) )\pi_\tau(x,u)du -\beta v^\tau_{xx}\right]d\tau,\\ du^x_\tau =&\nabla_u \mathbf{H} (x,u^x_\tau, v^\tau_x, v^\tau_{xx})d\tau+ \sqrt{2\lambda}dB_\tau. \end{cases} \end{align}\tag{24}\] In the relaxed control framework, the distribution of the optimal stochastic control is the Gibbs invariant measure of the Langevin dynamics for \(u_\tau^x\) as described in 3 with the optimal value function in the Hamiltonian. In the following, we show the convergence of the value function under the Monotonicity condition, which can simultaneously guarantee the exponential convergence of the control process \(u_\tau^x\) to its limiting Gibbs distribution.
Other iteration methods can also be regarded as continuous in the iteration direction; see, for example, vsivska2024gradient? for a regularized control problem with measure-valued control process, and ZZH2025? for classical control problems and mean-field games. The main differences between our iteration dynamics 24 (and likewise 28 for classical control problems) and these existing methods can be summarized as follows:
The update of the control process and the corresponding value in vsivska2024gradient?, ZZH2025? is based on the linear function derivative of the value with respect to the control process. It means that at every round of iteration, the control is updated at every physical time \(t\) based on the gradient of the value function from the last round. In contrast, in 24 , we only keep track of a single distribution density function generated by the Langevin dynamics of \(u^x_{\tau}\), i.e. \(\pi_\tau\) (instead of a process of measure-valued control) and use it to update the strategy at \(t=0\). The strategy for \(t>0\) is updated by merely pushing the strategies used in the previous iteration into the future of physical time frame (c.f. Definition 2). This is the main innovation of our iteration method.
The value \(v^\tau\) at each iteration round \(\tau\) is decomposed into two parts. The first is the running reward \(\tilde{f}(x,\pi_\tau)\) accrued over \([0,\Delta t]\), using in particular the most recently generated strategy \(\pi_\tau\) at \(t=0\). The second is the aggregated discounted reward over \(t>\Delta t\). By construction of the policy update, this second part is precisely the value from the previous iteration round, \(v^{\tau-\Delta t}\), evaluated at \(X^{\tau}_{\Delta t}\) and suitably discounted. Letting \(\Delta t \rightarrow 0\) and applying Itô’s formula, we obtain the dynamics of \(v^\tau\) in 24 . This dynamics requires only \(v^\tau\) and the strategy \(\pi_\tau\)—the latest iterates of the value and the policy, respectively: the policy supplies the running reward \(\tilde{f}(x,\pi_\tau)\) at \(t=0\), while the value of the new strategy for \(t>0\) is embedded in the value from the previous step. For this reason, the coupled dynamics does not require the dynamics of \(X\).
Another major difference between our results and those of vsivska2024gradient? is that the latter relies on Pontryagin’s maximum principle, with the iteration defined through the adjoint process of the stochastic control problem. Consequently, in addition to the forward dynamics of \(X\), this method requires solving a backward stochastic differential equation at every iteration round. In contrast, our iteration is purely forward in nature, which eases the analysis. Finally, it is worth pointing out that our iteration is designed for the infinite-horizon stochastic control problem and cannot be directly extended to the finite-horizon case considered in vsivska2024gradient?. Conversely, it is also unclear to us how the backward component of their method could be applied to the infinite-horizon problem.
Lemma 9. Under Assumptions 1 and 7, the dynamics in Lemma 1 satisfy the following estimates, \[\begin{align} \|v^{\tau}(x)-\tilde{v}^{\tau}(x)\|^2&\le e^{-2\beta \tau} \|v^0(x)-\tilde{v}^0(x)\|^2;\\ \|v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)\|^2&\le e^{-2\beta \tau} \|v^0_x(x)-\tilde{v}^0_x(x)\|^2;\\ \|v^{\tau}_{xx}(x)-\tilde{v}^{\tau}_{xx}(x)\|^2&\le e^{-2\beta \tau} \|v^0_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2. \end{align}\]
Proof. The difference between the coupled processes \(v^{\tau}\) (and \(v^{\tau}_x\), \(v^\tau_{xx}\) respectively\()\) and \(\tilde{v}^{\tau}\) (and \(\tilde{v}^{\tau}_x\), \(\tilde{v}^\tau_{xx}\) respectively\()\) follows the dynamics below (omitting the argument \(x\) for simplicity), \[\begin{align} d(v^{\tau}-\tilde{v}^{\tau})=&-\beta(v^\tau-\tilde{v}^\tau)d\tau \\ &+\Big[ \int_{\mathbb{R}^n} (\mathbf{H}^\tau- \lambda \ln\pi_\tau)\pi_\tau(x,u)du -\int_{\mathbb{R}^n}(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)\tilde{\pi}_\tau(x,\tilde{u})d\tilde{u} \Big]d\tau,\\ d(v^{\tau}_x-\tilde{v}^{\tau}_x)=&-\beta(v^\tau_x-\tilde{v}_x^\tau) d\tau\\ &+ \nabla_x \Big[\int_{\mathbb{R}^n}(\mathbf{H}^\tau- \lambda \ln \pi_\tau)\pi_\tau(x,u)\,du -\int_{\mathbb{R}^n}(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)\tilde{\pi}_\tau(x,\tilde{u})\,d\tilde{u} \Big]d\tau ,\\ d(v^{\tau}_{xx}-\tilde{v}^{\tau}_{xx})=& -\beta(v^\tau_{xx}-\tilde{v}_{xx}^\tau)d\tau\\ &+\nabla^2_{xx}\Big[ \int_{\mathbb{R}^n} (\mathbf{H}^\tau- \lambda \ln \pi_\tau)\pi_\tau(x,u)\,du -\int_{\mathbb{R}^n}(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)\tilde{\pi}_\tau(x,\tilde{u})\,d\tilde{u}\Big] d\tau . \end{align}\] We show the convergence for \(v^{\tau}_x-\tilde{v}^{\tau}_x\), and the estimates for \(\|v^{\tau}(x)-\tilde{v}^{\tau}(x)\|^2\) and \(\|v_{xx}^{\tau}(x)-\tilde{v}_{xx}^{\tau}(x)\|^2\) follow similarly. From its dynamics above, \[\begin{align} &\frac{d}{d\tau} \|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2\\ =&2\Big(\nabla_x\Big[ \int_{\mathbb{R}^n} (\mathbf{H}^\tau- \lambda \ln\pi_\tau)\pi_\tau(x,u)\,du -\int_{\mathbb{R}^n}(\tilde{\mathbf{H}}^\tau -\lambda\ln\tilde{\pi}_\tau)\tilde{\pi}_\tau(x,\tilde{u})\,d\tilde{u} \Big]\Big)^\top(v^{\tau}_x-\tilde{v}^{\tau}_x)\\ &-2\beta \|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2 \\ \le & -2\beta\|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2, \end{align}\] where the last inequality follows from the monotonicity condition in Assumption 7. This then implies \[\label{convergence:32v39} \|v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)\|^2\le e^{-2\beta \tau} \|v^0_x(x)-\tilde{v}^0_x(x)\|^2.\tag{25}\] ◻
Proposition 7. Let (MC II) and (MC III) Assumption 7 hold true, and assume for each given \(x\in\mathbb{R}^d\), there exist positive constants \(\overline{\varepsilon}(x)\) and \(\kappa(x)\neq \beta\), such that \[\begin{align} \label{eigen32assumption} \nabla^2_{uu}f(x,\cdot) +L\sum_{i=1}^n\nabla^2_{uu}b_i(x,\cdot)&+ L^2\sum_{j,k=1}^n\nabla^2_{uu}\sigma_{jk}(x,\cdot)+L\sum_{i,j,k=1}^n \nabla_u \sigma_{ik}(x,\cdot) (\nabla_{u}\sigma_{jk} (x,\cdot))^{\top}\nonumber \\ &+\overline{\varepsilon}(x)(n+n^2)\mathbf{I}_n\preceq -\kappa(x)\mathbf{I}_n, \end{align}\qquad{(13)}\] where \(L\) is the uniform (in \(\tau\)) bound for \(v^\tau_x\), \(v^\tau_{xx}\), \(\nabla_u b\), \(\sigma\) and \(\nabla_u \sigma\), i.e., \[\label{condition:32uniform32bound} \Big\{ \|v^{\tau}_x(x)\|^2, \|v^{\tau}_{xx}(x)\|^2_F, \sum_{i=1}^n\|\nabla_u b_i\|^2, \|\sigma\|_F^2, \sum_{i,j=1}^n\|\nabla_u \sigma_{ij}\|^2\Big\}\le L, \quad \text{for all}\quad \tau\ge 0,\qquad{(14)}\] where \(\|A\|_F^2:=\sum_{i,j=1}^n|A_{ij}|^2\) denotes the Frobenius norm for a matrix \(A\). Then \[\mathcal{W}_2^2(\pi_\tau,\tilde{\pi}_\tau)\le e^{-2\kappa(x)\tau} \mathbb{E}[\|u_0-\tilde{u}_0\|^2]+\frac{e^{-2\beta\tau }-e^{-2\kappa(x)\tau}}{2(\kappa(x)-\beta)} C_0,\] where for some \(\underline{\varepsilon}(x)>0\), \[\begin{align} \label{constant32C0} C_0:= \frac{1}{\underline{\varepsilon}(x)}L\left(\mathbb{E}[\|v^{0}_{x}(x)-\tilde{v}^{0}_{x}(x)\|^2]+L\mathbb{E}[\|v^{0}_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2]\right) \geq 0. \end{align}\qquad{(15)}\]
Proof. Motivated by synchronous coupling techniques for Langevin dynamics EGZ19?, Dalalyan2020?, we adapt this idea to the inhomogeneous Langevin dynamics governing the control process \(u_\tau\) and derive a convergence rate for the distribution of the randomized control in our policy iteration scheme under the monotonicity condition. Following the proof of Lemma 9, denote as \((v^{\tau}_x,u_{\tau})\) and \((\tilde{v}^{\tau}_x,\tilde{u}_{\tau})\) the two coupled processes driven by the same Brownian motion \((B_{\tau})_{\tau\ge 0}\). The difference \(u_{\tau}-\tilde{u}_{\tau}\) follows \[d(u_\tau-\tilde{u}_\tau)=[\nabla_u \mathbf{H}^\tau(u_\tau) -\nabla_{u} \tilde{\mathbf{H}}^\tau(\tilde{u}_\tau)]d\tau.\] Next, we show the convergence of the difference process \(u_\tau-\tilde{u}_\tau\). Observe that \[\begin{align} &\nabla_u \mathbf{H}^\tau(u_\tau) -\nabla_{u} \tilde{\mathbf{H}}^\tau(\tilde{u}_\tau)\\ =& \nabla_u f (x,u_\tau) -\nabla_{u} f(x,\tilde{u}_\tau)+ \sum_{i=1}^n\nabla_u b_i(x,u_\tau)v^{\tau}_{x,i}(x)-\sum_{i=1}^n\nabla_{u} b_i(x,\tilde{u}_\tau) \tilde{v}^{\tau}_{x,i}(x)\\ &+\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)\nabla_u\sigma_{jk}(x,u_\tau)v^{\tau}_{xx,ij}(x)-\sum_{i,j,k=1}^n\sigma_{ik}(x,\tilde{u}_\tau)\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)\tilde{v}^{\tau}_{xx, ij}(x)\\ =&\int_0^1\frac{d}{dr} \nabla_u f(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr\\ &+\sum_{i=1}^n\Big(v^{\tau}_{x,i}(x)\int_0^1 \frac{d}{dr}\nabla_u b_i(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr +\nabla_{u}b_i(x,\tilde{u}_\tau)[v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)]\Big)\\ &+\sum_{i,j,k=1}^n \Big(\sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\int_0^1 \frac{d}{dr}\nabla_u \sigma_{jk}(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr\Big.\\ &\Big.+\sigma_{ik}(x,u_\tau)\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)[v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}]\Big) \\ &+\sum_{i,j,k=1}^n\tilde{v}^{\tau}_{xx,ij}\nabla_u\sigma_{jk}(x,\tilde{u}_\tau)\int_0^1 \frac{d}{dr}\sigma_{ik}(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr \\ =&\int_0^1\langle u_\tau-\tilde{u}_\tau, \nabla^2_{uu} f(x,r u_\tau+(1-r)\tilde{u}_\tau ) \rangle dr\\ &+ \sum_{i=1}^nv^{\tau}_{x,i}(x) \int_0^1 \langle u_\tau-\tilde{u}_\tau, \nabla^2_{uu} b_i(x,r u_\tau+(1-r)\tilde{u}_\tau ) \rangle dr +\sum_{i=1}^n\nabla_{u}b_i(x,\tilde{u}_\tau)[v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)]\\ &+\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\int_0^1 \langle u_\tau-\tilde{u}_\tau, \nabla^2_{uu} \sigma_{jk}(x,r u_\tau+(1-r)\tilde{u}_\tau ) \rangle dr\\ &+\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)[v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}] \\ &+\sum_{i,j,k=1}^n \tilde{v}^{\tau}_{xx,ij}\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)\int_0^1 \langle u_\tau-\tilde{u}_\tau, \nabla_{u} \sigma_{ik}(x,r u_\tau+(1-r)\tilde{u}_\tau ) \rangle dr\\ =& \mathbf{U}^f_\tau (u_\tau-\tilde{u}_\tau) +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau (u_\tau-\tilde{u}_\tau)+\sum_{i=1}^n\nabla_{u}b_i(x,\tilde{u}_\tau)[v^{\tau}_{x,i}(x)-\tilde{v}^\tau_{x,i}(x)]\\ &+\sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau (u_\tau-\tilde{u}_\tau)\sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}+\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)[v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}]\\ &+\sum_{i,j,k=1}^n(\hat{\mathbf{U}}^{\sigma_{ik}}_\tau)^\top(u_\tau-\tilde{u}_\tau)\tilde{v}^{\tau}_{xx,ij}\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau), \end{align}\] where we denote \[\begin{align} \mathbf{U}^f_\tau :=& \int_0^1 \nabla^2_{uu} f(x, r u_\tau+(1-r)\tilde{u}_\tau ) dr,\nonumber\\ \mathbf{U}^{b_i}_\tau :=& \int_0^1 \nabla^2_{uu} b_i(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr,\quad \text{for}\quad i=1,\cdots,n,\nonumber\\ \mathbf{U}^{\sigma_{ij}}_\tau :=& \int_0^1 \nabla^2_{uu} \sigma_{ij}(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr, \quad \text{for}\quad i,j=1,\cdots,n,\nonumber\\ \hat{\mathbf{U}}^{\sigma_{ij}}_\tau :=& \int_0^1 \nabla_{u} \sigma_{ij}(x,r u_\tau+(1-r)\tilde{u}_\tau ) dr, \quad \text{for}\quad i,j=1,\cdots,n. \end{align}\] Plugging the above terms into the difference equation for the control process, we have \[\begin{align} &\frac{d}{d\tau} (u_\tau-\tilde{u}_\tau)\\ =&\mathbf{U}^f_\tau (u_\tau-\tilde{u}_\tau) +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau (u_\tau-\tilde{u}_\tau)+\sum_{i=1}^n\nabla_{ u}b_i(x,\tilde{u}_\tau)[v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)]\\ &+\sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau (u_\tau-\tilde{u}_\tau)\sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}+\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)\nabla_{ u}\sigma_{jk}(x,\tilde{u}_\tau)[v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}]\\ &+\sum_{i,j,k=1}^n(\hat{\mathbf{U}}^{\sigma_{ik}}_\tau)^\top(u_\tau-\tilde{u}_\tau)\tilde{v}^{\tau}_{xx,ij}\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)\\ =&\Big[\mathbf{U}^f_\tau +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau+ \sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau \sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\\ &+\sum_{i,j,k=1}^nv^{\tau}_{xx,ij}\hat{\mathbf{U}}^{\sigma_{ik}}_\tau(\nabla_{ u}\sigma_{jk}(x,\tilde{u}_\tau))^\top\Big] (u_\tau-\tilde{u}_\tau)\\ &+\sum_{i=1}^n\nabla_{u}b_i(x,\tilde{u}_\tau)[v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)]+\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)[v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}]. \end{align}\] Applying the above equation and Young’s inequality (\(2ab\le \varepsilon a^2+\frac{1}{\varepsilon}b^2\), for \(\varepsilon>0\)), \[\begin{align} \label{difference32u} &\frac{d}{d\tau} \|u_\tau-\tilde{u}_\tau\|^2\nonumber\\ =&2(u_\tau-\tilde{u}_\tau)^\top\Big[\mathbf{U}^f_\tau +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau+ \sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau \sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\nonumber \\ &\quad \quad \quad\quad \quad\quad +\sum_{i,j,k=1}^n\hat{\mathbf{U}}^{\sigma_{ik}}_\tau( \nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau))^\top v^{\tau}_{xx,ij}\Big] (u_\tau-\tilde{u}_\tau) \nonumber\\ &+2\sum_{i=1}^n(\nabla_{u}b_i(x,\tilde{u}_\tau))^\top[u_\tau-\tilde{u}_\tau][v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)]\nonumber\\ &+2\sum_{i,j,k=1}^n\sigma_{ik}(x,u_\tau)[v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}](\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau))^\top[u_\tau-\tilde{u}_\tau]\nonumber\\ \le &2(u_\tau-\tilde{u}_\tau)^\top\Big[\mathbf{U}^f_\tau +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau+ \sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau \sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\nonumber \\ &\quad \quad \quad\quad \quad\quad +\sum_{i,j,k=1}^n\hat{\mathbf{U}}^{\sigma_{ik}}_\tau( \nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau))^\top v^{\tau}_{xx,ij}\Big] (u_\tau-\tilde{u}_\tau) \nonumber\\ &+(\sum_{i=1}^n\varepsilon_{1i}+\sum_{i,j=1}^n\varepsilon_{2ij})\|u_\tau-\tilde{u}_\tau\|^2+ \sum_{i=1}^n\frac{1}{\varepsilon_{1i}}\| \nabla_{u}b_i(x,\tilde{u}_\tau)\|^2|v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)|^2\nonumber\\ &+\sum_{i,j=1}^n\frac{1}{\varepsilon_{2ij}}|\sigma_{ik}(x,u_\tau)|^2|v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}|^2\|\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)\|^2, \end{align}\tag{26}\] where \(\varepsilon_{1i}\)’s and \(\varepsilon_{2ij}\)’s are any positive constants, satisfying \(\max\left\{\varepsilon_{1i},\varepsilon_{2ij},1\leq i,j\leq n\right\} = \overline{\varepsilon}(x)\). Notice that by Assumption 1, \(\nabla_u b(x,u_\tau)\), \(\sigma\) and \(\nabla_u \sigma\) are bounded. Furthermore, Lemma 9 implies that \(v^\tau_x\) and \(v^\tau_{xx}\) converge, and thus are also bounded. Therefore there exists \(L>0\), such that the bound in ?? holds. Applying Lemma 9 and using the upper bound \(L\), the above inequality implies that \[\begin{align} & \frac{d}{d\tau} \|u_\tau-\tilde{u}_\tau\|^2\\ \le & 2(u_\tau-\tilde{u}_\tau)^\top\Big[\mathbf{U}^f_\tau +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau+ \sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau \sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\nonumber \\ &\quad \quad \quad\quad \quad\quad +\sum_{i,j,k=1}^n\hat{\mathbf{U}}^{\sigma_{ik}}_\tau( \nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau))^\top v^{\tau}_{xx,ij}+\overline{\varepsilon}(x)(n+n^2)\mathbf{I}_n\Big] (u_\tau-\tilde{u}_\tau) \nonumber\\ &+\frac{1}{\underline{\varepsilon}(x)}L\left(\|v^{\tau}_{x}(x)-\tilde{v}^{\tau}_{x}(x)\|^2+L\|v^{\tau}_{xx}(x)-\tilde{v}^{\tau}_{xx}(x)\|^2\right)\\ \le &2(u_\tau-\tilde{u}_\tau)^\top\Big[\mathbf{U}^f_\tau +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau+ \sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau \sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\nonumber \\ &\quad \quad \quad\quad \quad\quad +\sum_{i,j,k=1}^n\hat{\mathbf{U}}^{\sigma_{ik}}_\tau( \nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau))^\top v^{\tau}_{xx,ij}+\overline{\varepsilon}(x)(n+n^2)\mathbf{I}_n\Big] (u_\tau-\tilde{u}_\tau) \nonumber\\ &+\frac{1}{\underline{\varepsilon}(x)}Le^{-2\beta\tau}\left(\|v^{0}_{x}(x)-\tilde{v}^{0}_{x}(x)\|^2+L\|v^{0}_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2\right), \end{align}\] where \(\underline{\varepsilon}(x) = \min\left\{\varepsilon_{1i},\varepsilon_{2ij},1\leq i,j\leq n\right\}\). Then condition ?? implies that \[\begin{align} &\frac{d}{d\tau} \|u_\tau-\tilde{u}_\tau\|^2\\ \le& -2\kappa(x) \|u_\tau-\tilde{u}_\tau\|^2+\frac{1}{\underline{\varepsilon}(x)}Le^{-2\beta\tau}\left(\|v^{0}_{x}(x)-\tilde{v}^{0}_{x}(x)\|^2+L\|v^{0}_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2\right). \end{align}\] Applying Gronwall’s inequality, the following estimate holds \[\begin{align} \label{grownall32relax} \|u_\tau-\tilde{u}_\tau\|^2 \le& e^{-2\kappa(x)\tau} \|u_0-\tilde{u}_0\|^2\\ &+e^{-2\kappa(x)\tau}\Big[ \frac{1}{\underline{\varepsilon}(x)}L\Big(\|v^{0}_{x}(x)-\tilde{v}^{0}_{x}(x)\|^2+L\|v^{0}_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2\Big) \int_0^\tau e^{2(\kappa(x)-\beta)r}dr\Big]\\ \le& e^{-2\kappa(x)\tau} \|u_0-\tilde{u}_0\|^2\\ &+\frac{e^{-2\beta\tau}-e^{-2\kappa(x)\tau}}{2(\kappa(x)-\beta)}\frac{1}{\underline{\varepsilon}(x)}L\Big( \|v^{0}_{x}(x)-\tilde{v}^{0}_{x}(x)\|^2+L\|v^{0}_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2\Big) \\ =& e^{-2\kappa(x)\tau} \|u_0-\tilde{u}_0\|^2+\frac{e^{-2\beta\tau }-e^{-2\kappa(x)\tau}}{2(\kappa(x)-\beta)} C_0. \end{align}\tag{27}\] Taking expectation on both sides, we conclude that \[\mathcal{W}_2^2(\pi_\tau,\tilde{\pi}_\tau)\le e^{-2\kappa(x)\tau} \mathbb{E}[\|u_0-\tilde{u}_0\|^2]+\frac{e^{-2\beta\tau }-e^{-2\kappa(x)\tau}}{2(\kappa(x)-\beta)} C_0.\] ◻
Corollary 1. If \(\pi_0 = \pi^* = \frac{e^{\mathbf{H}(x,u,v^*_x,v^*_{xx})/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}(x,u,v^*_x,v^*_{xx})/\lambda} du}\), then \(\pi_{\tau} = \pi^*\). Together with Proposition 7, it implies that \(\pi_{\tau}\) converges exponentially.
Proof. If \(\pi_0 = \pi^*\), then \(v^0\) is independent of \(t\), and \[v^0 = \mathbb{E}\left[\int_0^\infty e^{-\beta t}\left(\tilde{f}(X^*_t,\pi^*(X^*_t))- \lambda \int_{\mathbb{R}^n}\pi^*\ln\pi^*(X^*_t,u)du\right)dt\right],\] where \(X^*\) is the state process under the strategy \(\pi_t = \pi^*\) for \(t \geq 0\). Since Proposition 3 shows that \(\pi^*\) is the optimal strategy and \(v^*\) is the corresponding value, \(v^0 = v^*\). Furthermore, by the policy improvement result in Proposition 1, \(v^\tau = v^*\) for \(\tau \geq 0\), thus the diffusion 3 that \(u_\tau\) follows becomes \[du^x_\tau = \nabla_u \mathbf{H} (x,u^x_\tau,v^*_x,v^*_{xx})d\tau+ \sqrt{2\lambda}dB_\tau,\] which has an invariant distribution \(\pi^*= \frac{e^{\mathbf{H}(x,u,v^*_x,v^*_{xx})/\lambda}}{\int_{\mathbb{R}^n}e^{\mathbf{H}(x,u',v^*_x,v^*_{xx})/\lambda} du'}\). Since \(\pi_0 = \pi^*\), it implies that \(\pi_{\tau} = \pi_0\). ◻
Remark 8. The convergence analysis for the above continuous dynamics 24 provides the theoretical guarantees for the discrete algorithm design, as the Langevin dynamic always serves as the “ground truth” for the corresponding discrete Markov chain. The newly proposed dynamics 24 build a bridge between the relaxed stochastic optimal control problem and the Langevin dynamics-based distribution sampling problem from machine learning welling2011bayesian?, chen2015convergence?, teh2016consistency?, dalalyan2020sampling?, deng2020non?. The discrete algorithm designs are left for future works.
Recall the Hamiltonian defined in 5 , which is inhomogeneous in the \(\tau\) direction since \(v^{\tau}_x\) and \(v^{\tau}_{xx}\) change simultaneously along \(\tau\). As an analogue to Assumption 7, we propose the following monotonicity condition for the classical control problem.
Assumption 8 (Monotonicity Condition). With the Hamiltonian \(\mathbf{H}^\tau\) defined in 5 , for each \(x\in\mathbb{R}^d\), we assume \[\begin{align} (\text{MC IV}):\quad & \mathbb{E}\Big[\Big(\mathbf{H}^\tau(u_\tau)-\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau)\Big)(v^\tau(x)-\tilde{v}^\tau(x))\Big]\le 0;\\ (\text{MC V}):\quad& \mathbb{E}\Big[\Big(\partial_x \mathbf{H}^\tau(u_\tau)-\partial_x\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau) \Big)^\top(v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)) \Big]\le 0;\\ (\text{MC VI}):\quad& \mathbb{E}\Big[\mathrm{Tr}\Big[(\partial_{xx} \mathbf{H}^\tau(u_\tau)-\partial_{xx}\tilde{\mathbf{H}}_\tau (\tilde{u}_\tau)):(v^{\tau}_{xx}(x)-\tilde{v}^{\tau}_{xx}(x))\Big] \Big]\le 0, \end{align}\] where \((u_\tau,v^\tau,\mathbf{H}^\tau)\) and \((\tilde{u}_\tau,\tilde{v}^\tau,\tilde{\mathbf{H}}^\tau)\) are the result of strategy and value iteration for the classical control problem.
Under the above assumption, we first show the convergence of the value function for the following strategy and value iteration-continuous dynamics arising from Lemma 3, \[\begin{align} \label{coupled32classic32dynamics} \begin{cases} dv^\tau &= \left(-\beta v^\tau(x) +\mathbf{H}(x,u_\tau^x,v_x^{\tau},v_{xx}^{\tau})\right)d\tau,\\ dv^\tau_x &=(\nabla_x \mathbf{H}(x,u_\tau^x,v^{\tau}_x,v^{\tau}_{xx}) -\beta v^\tau_x)d\tau,\\ dv^\tau_{xx} &=(\nabla^2_{xx}\mathbf{H}(x,u_\tau^x,v^{\tau}_x,v^{\tau}_{xx}) -\beta v^\tau_{xx})d\tau,\\ du^x_\tau &= \nabla_u \mathbf{H} (x,u^x_\tau, v^\tau_x, v^\tau_{xx})d\tau+ \sqrt{2\lambda}dB_\tau. \end{cases} \end{align}\tag{28}\]
Lemma 10. Under Assumption 8, for a pair of value process \((v^\tau, \tilde{v}^{\tau})\) \((\)and \((v^\tau_x, \tilde{v}^{\tau}_x)\), \((v^\tau_{xx}, \tilde{v}^{\tau}_{xx}\) \()\) respectively\()\) with their initial point \((v^0, \tilde{v}^0)\) \((\)and \((v_x^0, \tilde{v}^0_x)\), \((v_{xx}^0, \tilde{v}^0_{xx})\) respectively \()\) following the dynamics defined in 28 , the following convergence results hold: for any \(\beta>0\) such that Assumption 4 holds (the optimal value function is finite), \[\begin{align} \mathbb{E}[\|v^{\tau}(x)-\tilde{v}^{\tau}(x)\|^2]&\le e^{-2\beta \tau} \mathbb{E}[\|v^0(x)-\tilde{v}^0(x)\|^2];\\ \mathbb{E}[\|v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)\|^2]&\le e^{-2\beta \tau} \mathbb{E}[\|v^0_x(x)-\tilde{v}^0_x(x)\|^2];\\ \mathbb{E}[\|v^{\tau}_{xx}(x)-\tilde{v}^{\tau}_{xx}(x)\|^2]&\le e^{-2\beta \tau} \mathbb{E}[\|v^0_{xx}(x)-\tilde{v}^0_{xx}(x)\|^2]. \end{align}\]
Proof. Similar to the proof of Lemma 9, the difference \(v_x^{\tau} - \tilde{v}^\tau_x\) follows \[\label{eq:v-dynamics-c-convergence} d[v_x^\tau(x)-\tilde{v}_x^\tau(x)] = \left(-\beta [v_x^\tau(x)-\tilde{v}_x^\tau(x)] + \nabla_x \mathbf{H}^\tau(u_\tau) -\nabla_x\tilde{\mathbf{H}}^\tau (\tilde{u}_\tau)\right)d\tau.\tag{29}\] Thus \[\begin{align} \frac{d}{d\tau} \|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2 =&2\Big[ \nabla_x \mathbf{H}^\tau(u_\tau) -\nabla_x \tilde{\mathbf{H}}^\tau(\tilde{u}_\tau) \Big]^\top(v^{\tau}_x-\tilde{v}^{\tau}_x) -2\beta\|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2 \\ =& 2\Big[\nabla_x(\mathbf{H}^\tau(u_\tau)-\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau))^\top(v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)) \Big] -2\beta\|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2. \end{align}\] Taking expectation on both sides, \[\begin{align} \frac{d}{d\tau} \mathbb{E}[\|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2] =& 2\mathbb{E}\Big[\nabla_x(\mathbf{H}^\tau(u_\tau)-\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau))^\top(v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)) \Big] -2\beta\mathbb{E} [\|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2]\\ \le & -2\beta \mathbb{E}[\|v^{\tau}_x-\tilde{v}^{\tau}_x\|^2], \end{align}\] where the last inequality follows from the monotonicity condition in Assumption 8. Then Gronwall’s inequality implies that \[\label{convergence:32v3932classic} \mathbb{E}[\|v^{\tau}_x(x)-\tilde{v}^{\tau}_x(x)\|^2]\le e^{-2\beta \tau}\mathbb{E}[ \|v^0_x(x)-\tilde{v}^0_x(x)\|^2].\tag{30}\] The estimates for \(\mathbb{E}[\|v^{\tau}(x)-\tilde{v}^{\tau}(x)\|^2]\) and \(\mathbb{E}[\|v^{\tau}_{xx}(x)-\tilde{v}^{\tau}_{xx}(x)\|^2]\) follow similar arguments. ◻
Proposition 9. Assume that condition (MC V) and (MC VI) in Assumption 8 and Condition ?? in Proposition 7 hold true, then \[\mathcal{W}_2^2(\pi_\tau,\tilde{\pi}_\tau)\le e^{-2\kappa(x)\tau} \mathbb{E}[\|u_0-\tilde{u}_0\|^2]+\frac{e^{-2\beta\tau }-e^{-2\kappa(x)\tau}}{2(\kappa(x)-\beta)} C_0,\] where \(C_0\) is the same as that defined in Proposition 7. \(\pi_\tau\) and \(\tilde{\pi}_\tau\) denote the probability distribution of the process \(u_\tau\) and \(\tilde{u}_\tau\) following the dynamics in 28 with same Brownian motion and initial conditions \(u_0\) and \(\tilde{u}_0\).
Proof. Following the proof of Proposition 7, and equation 26 , we have \[\begin{align} &\frac{d}{d\tau} \|u_\tau-\tilde{u}_\tau\|^2 \\ \le &2(u_\tau-\tilde{u}_\tau)^\top\Big[\mathbf{U}^f_\tau +\sum_{i=1}^nv^{\tau}_{x,i}(x) \mathbf{U}^{b_i}_\tau+ \sum_{i,j,k=1}^n\mathbf{U}^{\sigma_{jk}}_\tau \sigma_{ik}(x,u_\tau)v^{\tau}_{xx,ij}\\ &\quad\quad\quad\quad\quad\quad +\sum_{i,j,k=1}^n\hat{\mathbf{U}}^{\sigma_{ik}}_\tau\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau) v^{\tau}_{xx,ij}\Big] (u_\tau-\tilde{u}_\tau)\\ &+(\sum_{i=1}^n\varepsilon_{1i}+\sum_{i,j=1}^n\varepsilon_{2ij})\|u_\tau-\tilde{u}_\tau\|^2+ \sum_{i=1}^n\frac{1}{\varepsilon_{1i}}\| \nabla_{u}b_i(x,\tilde{u}_\tau)\|^2|v^{\tau}_{x,i}(x)-\tilde{v}^{\tau}_{x,i}(x)|^2\\ &+\sum_{i,j=1}^n\frac{1}{\varepsilon_{2ij}}|\sigma_{ik}(x,u_\tau)|^2|v^{\tau}_{xx,ij}-\tilde{v}^{\tau}_{xx,ij}|^2\|\nabla_{u}\sigma_{jk}(x,\tilde{u}_\tau)\|^2. \end{align}\] Then the claim of the proposition holds by taking expectation on both sides, following the proof for Proposition 7, and the estimates in Assumption 8. ◻
Remark 10. According to Lemma 10 and Proposition 9, for a given fixed temperature \(\lambda\), the optimal value and the optimal control converge simultaneously. In particular, Proposition 5 shows that value functions are improved along the policy dynamics as \(\tau\rightarrow \infty\) in the classical control framework with \(\lambda=0\). In a general model with non-cancave Hamiltonian, the coupled system is expected to move from one sub-optimal control to another improved sub-optimal control, which builds a bridge between the classical stochastic control problem and the non-convex optimization/learning problem gelfand1991recursive?, zhang2017hitting?, raginsky2017non?. The annealing-based continuous-time Langevin dynamics MCMC algorithms kirkpatrick1983optimization?, vcerny1985thermodynamical?, geman1986diffusions?, gelfand1990sampling?, tang2021simulated?, feng2024fisher? can be applied to our strategy and value iteration-continuous dynamics 28 by scheduling \(\lambda(\tau)\rightarrow 0\), as \(\tau\rightarrow \infty\). The convergence analysis of the annealed version of 28 with \(\lambda(\tau)\) deserves to be further explored in this new framework.
Remark 11. The newly proposed coupled continuous strategy-value iteration dynamics can be seen as policy-value inhomogeneous Langevin-type dynamics, which can be used to design Langevin dynamics-based Markov chain Monte Carlo (MCMC) algorithms to find the optimal policy or the corresponding Gibbs invariant distribution of the stochastic control. Using the stochastic gradient descent/ascent algorithms from Langevin MCMC welling2011bayesian?, teh2016consistency?, raginsky2017non?, our proposed continuous dynamics has the potential to be applied in higher dimensions as well. The dynamics of the control process \(u_\tau\) in both the relaxed-control formulation 24 and the classical-control formulation 28 follow an overdamped Langevin-type dynamics. One could also use the same Hamiltonian \(\mathbf{H}\) 23 as the potential function in Hamiltonian (underdamped) Langevin dynamics chen2015convergence?, ma2015complete?, cheng2018underdamped?, dalalyan2020sampling?, among many other related methods, to accelerate the convergence of the control. The exact numerical algorithm design with various generalizations and the corresponding discretization error analysis of the newly proposed continuous dynamics are left for future work.
Consider a relaxed control problem in a linear-quadratic setting with \(d=n=1\), in which \(b(x,u) = Ax + Bu\), \(\sigma(x,u) =Cx\) and \(f = - \left(\frac{M}{2}x^2+ \frac{N}{2}u^2 + Px + Qu\right)\), \(M \geq 0\), \(N >0\) and \(A<0\). Notice the model parameters do not satisfy the boundedness Assumption 1. However, as we show below, under our policy-value iterations, as long as we start the policy iteration from a Gaussian distribution, \(u_\tau\) stays Gaussian with bounded second moment and \(v^\tau\) is always quadratic in \(x\) for every \(\tau\). Furthermore, the linear dynamics of state \(X\) imply that it has finite first and second moments at any \(t\geq 0\). Finally, we also verify below that the relative entropies of \(\pi_\tau\) are uniformly bounded in \(\tau\). These conditions help us verify Assumptions 2 and 3 and use the dominated convergence theorem in all the proofs in Section 3, and all the results in that section, in particular the policy improvement and convergence results, still hold while we omit the proofs. Since \(f\) is strictly concave in \(u\), and both \(b\) and \(\sigma\) are linear in \(u\), the second-order derivatives of \(b\) and \(\sigma\) with respect to \(u\) vanish. Hence, \[\nabla^2_{uu} f(x,u) + L \nabla^2_{uu} b_i(x,u) + L^2 \nabla^2_{uu} \sigma(x,u) + L \big(\nabla_u \sigma(x,\cdot)\big)^2 = -N < 0.\] It follows that ?? holds whenever \(2\bar\varepsilon < N\), with \(\bar\varepsilon\) chosen independently of \(x\). . Thus in the following we focus on verifying the monotonicity conditions (MC II) and (MC III) proposed in Section 6.1, which guarantee the exponential convergence of the policy iteration. The coupled dynamics 24 for this model are reduced to (omitting the superscript and argument \(x\) for the ease of notation) \[\begin{align} \begin{cases} du_\tau =&\left(-Nu_\tau+Bv^\tau_x-Q\right)d\tau + \sqrt{2\lambda}dB_\tau,\\ dv^\tau =&\left(\frac{C^2v^\tau_{xx}-M}{2}x^2+ (Av^\tau_x-P)x +(Bv^\tau_x-Q)\mathbb{E}[u_\tau] -\frac{N}{2}\mathbb{E}[u^2_\tau] \right.\nonumber\\ &\left.- \lambda\int_{\mathbb{R}}\pi_{\tau}\ln \pi_{\tau}(u)du -\beta v^{\tau} \right)d\tau,\\ dv^\tau_x =&\left((C^2v^\tau_{xx}-M + Av^\tau_{xx})x + \frac{C^2v^\tau_{xxx}x^2}{2}+ Av^\tau_x-P \right.\nonumber\\ &\left.+Bv^\tau_{xx}\mathbb{E}[u_\tau] + (Bv^\tau_x -Q)\partial_x \mathbb{E}[u_\tau]- \frac{N}{2}\partial_x\mathbb{E}[u^2_\tau]\right.\nonumber\\ &\left.- \lambda\partial_x\int_{\mathbb{R}}\pi_{\tau}\ln \pi_{\tau}(u)du -\beta v^{\tau}_x \right)d\tau,\\ dv^\tau_{xx} =&\left((2C^2+A)v^\tau_{xxx} x +C^2v^\tau_{xx}-M + 2Av^\tau_{xx} + \frac{C^2v^\tau_{xxxx}x^2}{2} \right.\nonumber\\ &\left.+Bv^\tau_{xxx} \mathbb{E}[u_\tau] + 2Bv^\tau_{xx}\partial_x \mathbb{E}[u_\tau]+(Bv^\tau_x -Q)\partial_{xx} \mathbb{E}[u_\tau]\right.\nonumber\\ &\left.-\frac{N}{2}\partial_{xx}\mathbb{E}[u^2_\tau]- \lambda\partial_{xx}\int_{\mathbb{R}}\pi_{\tau}\ln \pi_{\tau}(u)du -\beta v^{\tau}_{xx} \right)d\tau. \end{cases} \end{align}\] Thus \(u\) is an Ornstein-Uhlenbeck process with inhomogeneous coefficients. In particular, \[u_\tau = e^{-N\tau}u_0 + \int_0^\tau e^{-N(\tau-s)}(Bv^s_x-Q)ds + \sqrt{2\lambda}\int_0^\tau e^{-(\tau-s)N}dB_s,\nonumber\] which follows a Gaussian distribution given \(u_0\) is Gaussian, and \[\begin{align} \mathbb{E}[u_\tau] =& e^{-N\tau}\mathbb{E}[u_0] + e^{-N\tau}\int_0^\tau e^{Ns}(Bv^s_x-Q)ds\tag{31},\\ \mathsf{Var}(u_\tau) =& e^{-2N\tau}\mathsf{Var}(u_0) + \frac{\lambda(1-e^{-2N\tau})}{N},\tag{32}\\ -\int_{\mathbb{R}}\pi_\tau(u)\ln\pi_\tau(u)du =& \frac{\ln\left(2e\pi\left(e^{-N\tau}\mathsf{Var}(u_0) + \frac{\lambda(1-e^{-2N\tau})}{N}\right)\right)}{2}.\tag{33} \end{align}\]
Lemma 11. If \(\mathbb{E}[u_0]\) is linear in \(x\), \(\mathsf{Var}(u_0)\) is independent of \(x\), and for \(\tau \geq 0\), \(v^s\) is a quadratic function for every \(0\leq s\leq \tau\), then \(\frac{dv^\tau_{xx}}{d\tau}\) is independent of \(x\).
Proof. Since \(v^\tau\) is quadratic in \(x\) for all \(0\leq s\leq \tau\), \(dv^\tau_{xx}\) reduces to \[\begin{align} dv^\tau_{xx} =&\left((C^2+2A)v^\tau_{xx}-M +2Bv^\tau_{xx}\partial_x \mathbb{E}[u_\tau]+(Bv^\tau_x -Q)\partial_{xx} \mathbb{E}[u_\tau] - \frac{N}{2}\partial_{xx}\mathbb{E}[u^2_\tau]\right.\nonumber\\ &\left. - \lambda\partial_{xx}\int_{\mathbb{R}}\pi_{\tau}\ln \pi_{\tau}(u)du -\beta v^{\tau}_{xx} \right)d\tau.\nonumber \end{align}\] Furthermore, since \(v^s_{xx}\) is constant for \(0\leq s\leq \tau\) and \(\mathsf{Var}(u_0)\) is independent of \(x\), 31 , 32 and 33 imply that \(\mathbb{E}[u_\tau]\) is linear in \(x\), and \(\mathsf{Var}(u_\tau)\) and \(\int_{\mathbb{R}}\pi_{\tau}\ln \pi_{\tau}(u)du\) are independent of \(x\). Therefore, \(\mathbb{E}[u_\tau^2]\) is quadratic in \(x\) and \[\begin{align} \frac{dv^\tau_{xx}}{d\tau} =&(C^2+2A)v^\tau_{xx}-M +2Bv^\tau_{xx}\partial_x \mathbb{E}[u_\tau] - \frac{N}{2}\partial_{xx}\mathbb{E}[u^2_\tau] -\beta v^{\tau}_{xx},\nonumber \end{align}\] is independent of \(x\). ◻
From the above lemma, if \(u_0\) is Gaussian, with mean linear and variance independent of \(x\) (which we expect to be satisfied by the optimal solution), then \(v^0\) is quadratic in \(x\) (cf. the proof of Theorem 4.2 in wang2020continuous?), and the same holds for all \(\tau\geq 0\), and \(\mathsf{Var}(u_\tau)\) is independent of \(x\). Then \(v^\tau\) can be written as \[\begin{align} \label{v32tau32decomposition} \begin{cases} v^\tau &= \frac{a_2^\tau}{2}x^2 + a^\tau_1 x + a^\tau_0,\\ v^\tau_x &= a^\tau_2 x + a^\tau_1,\\ v^\tau_{xx} &= a^\tau_2. \end{cases} \end{align}\tag{34}\] Letting \(\mathbb{E}[u_0] = \mu\) and \(\mathsf{Var}(u_0) = \sigma^2\), then \[\begin{align} \partial_x \mathbb{E}[u_\tau^2] =& \frac{\partial \mathbb{E}[u_\tau]^2}{\partial x} = 2\mathbb{E}[u_\tau]\partial_x \mathbb{E}[u_\tau],\quad \partial_{xx} \mathbb{E}[u_\tau^2]= 2\left(\partial_x \mathbb{E}[u_\tau]\right)^2,\\ \mathbb{E}[u_\tau] =& e^{-N\tau}\mu + \int_0^\tau e^{N(s-\tau)}(Ba^s_2 x + Ba^s_1-Q)ds\nonumber\\ =& BI^\tau_2x +e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q,\\ \partial_x\mathbb{E}[u_\tau]=& BI^\tau_2, \end{align}\] where \(I^\tau_1 = \int_0^\tau e^{N(s-\tau)}a^s_1ds\) and \(I^\tau_2 = \int_0^\tau e^{N(s-\tau)}a^s_2ds\). Thus \[\begin{align} \mathbb{E}[\mathbf{H}^\tau(u_\tau)]=& - \left(\frac{M}{2}x^2 + Q\mathbb{E}[u_\tau] + \frac{N}{2}\mathbb{E}[u^2_{\tau}] + Px\right) + (Ax+B\mathbb{E}[u_\tau])v^\tau_x+\frac{C^2x^2}{2}v^\tau_{xx}\nonumber\\ =&\left(- \frac{M + NB^2(I^\tau_2)^2 - C^2a^\tau_2}{2} +\left(A+B^2I^\tau_2\right)a^\tau_2\right) x^2 \nonumber\\ &-\left(Ne^{-N\tau}\mu + BNI^\tau_1 + e^{-N\tau}Q\right)BI^\tau_2x\nonumber\\ &+\left(\left(A+B^2I^\tau_2\right)a^\tau_1 + B\left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right)a^\tau_2-P\right)x\nonumber\\ &- Q\left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right)+ B\left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right)a^\tau_1\nonumber\\ &- \frac{N}{2}\left(e^{-2N\tau}\sigma^2 + \frac{\lambda(1-e^{-2N\tau})}{N} + \left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right)^2\right),\nonumber\\ \partial_{x}\mathbb{E}[\mathbf{H}^\tau(u_\tau)]=& - \left(Mx + Q\partial_x\mathbb{E}[u_\tau] + \frac{N}{2}\partial_x\mathbb{E}[u^2_{\tau}] + P\right)\nonumber\\ &+ \left(A+B\partial_x\mathbb{E}[u_\tau]\right)v^\tau_x+((A+C^2)x+B\mathbb{E}[u_\tau])v^\tau_{xx}\nonumber\\ =&\left(-M -NB^2(I^\tau_2)^2 + \left(2A+2B^2I^\tau_2+C^2\right)a^\tau_2\right)x\nonumber\\ &-\left(Ne^{-N\tau}\mu + BNI^\tau_1 +e^{-N\tau}Q\right)BI^\tau_2\nonumber\\ &+\left(A+B^2I^\tau_2\right)a^\tau_1 + B\left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right)a^\tau_2-P, \nonumber\\ \partial_{xx}\mathbb{E}[\mathbf{H}^\tau(u_\tau)]=& - \left(M + \frac{N}{2}\partial_{xx}\mathbb{E}[u^2_{\tau}]\right)+ 2\left(A+B\partial_x\mathbb{E}[u_\tau]\right)v^\tau_{xx}\nonumber\\ =&-M -NB^2(I^\tau_2)^2 + \left(2A+2B^2I^\tau_2+C^2\right)a^\tau_2\nonumber\\ \frac{dv^\tau_{xx}}{d\tau} =& \frac{da^\tau_2}{d\tau}=\left(C^2+2A + 2B^2I^\tau_2 -\beta\right)a^\tau_2-M- NB^2(I^\tau_2)^2.\label{vxx-LQ} \end{align}\tag{35}\]
Lemma 12. With \(u_0\sim \mathcal{N}(\mu,\sigma^2)\) and definition 34 , for \(\tau=0\), we have, \[\begin{align} a^0_1 = -\frac{BMC^2\mu}{(A+C^2)(\beta-2A-C^2)(\beta - A)}-\frac{P}{\beta-A},\quad a^0_2 =-\frac{M}{\beta-2A-C^2}. \end{align}\]
Proof. With the strategy \(\rho^0_t = \pi_0\) for every \(t\geq 0\) and initial value \(X_0 = x\), \[\begin{align} dX_t = \left(AX_t +B\mu\right)dt +CX_tdW_t.\nonumber \end{align}\] Then \(X_t = xe^{(A-\frac{1}{2}C^2)t +CW_t} + B\mu\int_0^te^{(A-\frac{1}{2}C^2)(t-s) +C(W_t-W_s)}ds\), and \[\begin{align} v^0 =& \mathbb{E}\left[\int_0^\infty e^{-\beta t}\left(-\frac{M}{2}X_t^2 - \frac{N}{2}\left(\mu^2+\sigma^2\right) -PX_t-Q\mu\right)dt\right]\nonumber\\ =&-\frac{M}{\beta-2A-C^2}x^2-\left(\frac{BMC^2\mu}{(A+C^2)(\beta-2A-C^2)(\beta - A)}+\frac{P}{\beta-A}\right)x + K,\nonumber \end{align}\] for some constant \(K\). ◻
Lemma 13. Denote the limit of 34 as \(v^*\). The limit coefficients are given as follows: \(v^*_{xx}=a^*_2 := \lim\limits_{\tau\rightarrow\infty} a^\tau_2 = -\frac{2M}{\beta-C^2-2A + \sqrt{(\beta-C^2-2A)^2+\frac{4MB^2}{N}}}\) and \(a^*_1 = \lim\limits_{\tau\rightarrow\infty} a^\tau_1 = \frac{BQa^*_2+NP}{N(A-\beta) + B^2a^*_2}\).
Proof. By 35 , \(I_2^\tau\) and \(a^\tau_2\) satisfy the following coupled equations \[\begin{align} \frac{da^\tau_2}{d\tau}=&\left(C^2+2A + 2B^2I^\tau_2 -\beta\right)a^\tau_2-M- NB^2(I^\tau_2)^2,\tag{36}\\ \frac{dI^\tau_2}{d\tau} =& -NI^\tau_2 + a^\tau_2\tag{37}. \end{align}\] Since \(a^\tau_2=v^\tau_{xx}\) converges to \(a^*_2\) and thus \(\lim\limits_{\tau\rightarrow\infty} I^\tau_2=I^*_2 = a^*_2/N\). The limit can be found by checking the equilibrium point \((a^*_2,I^*_2)\) of the above equations, i.e., \(\frac{da^\tau_2}{d\tau}=\frac{dI^\tau_2}{d\tau}=0\) at \(\tau = \infty\), \[\begin{align} \left(C^2+2A + 2B^2I^*_2 -\beta\right)a^*_2-M - NB^2(I^*_2)^2 =& 0,\\ -NI^*_2 + a^*_2 =& 0. \end{align}\] Since \(\mathbb{E}[u_\tau]\) is linear in \(x\), and thus \(\tilde{f}(x,\pi_\tau)\) is concave in \(x\), and so is \(v^\tau\). Thus we choose the negative root of the quadratic function of \(a^*_2\) above, and \(a^*_2 = -\frac{2M}{\beta-C^2-2A + \sqrt{(\beta-C^2-2A)^2+\frac{4MB^2}{N}}}\). Next, since \(v^\tau_x\) converges, \(a^\tau_1\) converges to a constant \(a^*_1\), and \(I^\tau_1 = \int_0^\tau e^{N(s-\tau)}a^s_1ds\) converges to \(I^*_1 = a^*_1/N\). Its dynamics is \[\begin{align} \frac{da^\tau_1}{d\tau}=\frac{dv^\tau_x}{d\tau} -\frac{da^\tau_2}{d\tau}x =&\left(A-\beta\right)a^\tau_1-P+(Ba^\tau_1- Q)BI^\tau_2\nonumber\\ &+\left(Ba^\tau_2- NBI^\tau_2\right)\left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right),\nonumber \end{align}\] and the right hand side should converge to \(0\) at \(\tau = \infty\). It implies that \[\begin{align} (A-\beta)a^*_1 - P + Ba^*_2(Ba^*_1-Q)/N = 0.\nonumber \end{align}\] Thus, we have \(a^*_1 = \frac{BQa^*_2+NP}{N(A-\beta) + B^2a^*_2}\). ◻
Proposition 12. For the policy-value iterations which start from a Gaussian distribution with mean and variance independent of \(x\), if \(N>|A|\) and both are sufficiently large, then the monotonicity conditions (MC II) and (MC III) are satisfied.
Proof. Suppose the initial values \(u\sim \mathcal{N}(\mu,\sigma^2)\) and \(\tilde{u}\sim \mathcal{N}(\tilde{\mu},\tilde{\sigma}^2)\). Lemma 12 implies that \(a^0_2 = \tilde{a}^0_2\), and since both \(a^\tau_2\) and \(\tilde{a}^\tau_2\) satisfy the differential equations 36 and 37 , \(a^\tau_2 = \tilde{a}^\tau_2\) and \(I^\tau_2 = \tilde{I}^\tau_2\). Thus monotonicity condition (MC III) is satisfied. Next, since \(\mathbb{E}[\mathbf{H}^\tau(u_\tau)]\) is quadratic in \(x\), \(\partial_x\mathbb{E}[\mathbf{H}^\tau(u_\tau)] = \partial_{xx}\mathbb{E}[\mathbf{H}^\tau(u_\tau)]x + C_\tau\), where \[\begin{align} C_\tau = &-\left(Ne^{-N\tau}\mu + BNI^\tau_1 +e^{-N\tau}Q\right)BI^\tau_2\nonumber\\ &+\left(A+B^2I^\tau_2\right)a^\tau_1 + B\left(e^{-N\tau}\mu + BI^\tau_1 - \frac{1-e^{-N\tau}}{N}Q\right)a^\tau_2-P. \nonumber \end{align}\] and \(a^\tau_1\) follows \[\begin{align} \frac{da^\tau_1}{d\tau}=&\frac{dv^\tau_x}{d\tau} -\frac{da^\tau_2}{d\tau}x\nonumber\\ =&\left(A-\beta\right)a^\tau_1-P+(Ba^\tau_1- Q)BI^\tau_2+B\left(a^\tau_2- NI^\tau_2\right)\left(e^{-N\tau}\mu - \frac{1-e^{-N\tau}}{N}Q + BI^\tau_1\right). \nonumber \end{align}\] Thus (notice that \(\partial_{xx}\mathbb{E}[\mathbf{H}^\tau(u_\tau)] - \partial_{xx}\mathbb{E}[\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau)]=0\)) \[\begin{align} &\left(\partial_{x}\mathbb{E}[\mathbf{H}^\tau(u_\tau)] - \partial_{x}\mathbb{E}[\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau)]\right)(v^\tau_{x}-\tilde{v}^\tau_{x}) \nonumber\\ =&\left((\partial_{xx}\mathbb{E}[\mathbf{H}^\tau(u_\tau)] - \partial_{xx}\mathbb{E}[\tilde{\mathbf{H}}^\tau(\tilde{u}_\tau)])x + C_\tau-\tilde{C}_\tau\right)((a^\tau_{2}-\tilde{a}^\tau_{2})x + a^\tau_1-\tilde{a}^\tau_1)\nonumber \\ =&(C_\tau-\tilde{C}_\tau)(a^\tau_{1}-\tilde{a}^\tau_{1})\\ =&e^{-N\tau}B(a^\tau_2-NI^\tau_2)(\mu-\tilde{\mu})(a^\tau_1-\tilde{a}^\tau_1)\nonumber\\ &+ B^2\left(a^\tau_2-NI^\tau_2\right)\left(I^\tau_1-\tilde{I}^\tau_1\right)(a^\tau_1-\tilde{a}^\tau_1)+\left(A+ B^2 I^\tau_2\right)\left(a^\tau_1-\tilde{a}^\tau_1\right)^2.\label{eq:Ca} \end{align}\tag{38}\] Let \(Y_{\tau} = \frac{I^\tau_1 - \tilde{I}^\tau_1}{a^\tau_1 - \tilde{a}^\tau_1}\) and \(Z_{\tau} = e^{-N\tau}\frac{\mu - \tilde{\mu}}{a^\tau_1 - \tilde{a}^\tau_1}\). With \(Y_{0} = 0\) and \(Z_0 = -\frac{(A+C^2)(\beta-2A-C^2)(\beta - A)}{BMC^2}\), \[\begin{align} \frac{dY_{\tau}}{d\tau} =& -\left(N+A-\beta + B^2I^\tau_2\right)Y_\tau+1-B(a^\tau_2-NI^\tau_2)Z_\tau Y_{\tau} - B^2(a^\tau_2-NI^\tau_2)Y^2_{\tau},\tag{39}\\ \frac{dZ_\tau}{d\tau} =& -(N+A-\beta+B^2I^\tau_2)Z_\tau - B^2(a^\tau_2 - NI^\tau_2)Z_\tau Y_\tau+B(a^\tau_2 - NI^\tau_2)Z^2_\tau.\tag{40} \end{align}\] Since \(a^\tau_2\) converges to \(a^*_2\) and \(I^\tau_2\) converges to \(I^*_2 = a^*_2/N\), \(I^\tau_2\) is bounded, and the bound does not depend on initial distribution \(\pi_0\). For \(A<0\) and \(|A|\) sufficiently large, \(A +B^2 I^\tau_2<0\), and with sufficiently large \(N>|A-\beta|\), \(N+A-\beta+B^2I^\tau_2>0\). In 39 and 40 , the coefficients of \(Z_\tau Y_\tau\), \(Y^2_\tau\) and \(Z_\tau^2\) are all bounded, and the bounds are independent of the initial distribution \(\pi_0\) and \(N\). Then with sufficiently large \(M\) and \(N\), such that \(|Z_0|\) is sufficiently small, since \(Y_0 = 0\), the mean reverting term in both 39 and 40 dominates other terms. Thus \(Y_\tau\) and \(Z_\tau\) are bounded. Furthermore, since \(a^\tau_2 - NI^\tau_2\) converges to \(0\), \(\lim\limits_{\tau\rightarrow \infty}Y_\tau = \frac{1}{N+A-\beta + B^2a^*_2/N}\) and \(\lim\limits_{\tau\rightarrow \infty}Z_\tau = 0\). Therefore, 38 implies that \[\begin{align} &(C_\tau - \tilde{C}_\tau)(a^\tau_1-\tilde{a}^\tau_1) \nonumber\\ =&\left(e^{-N\tau}B(a^\tau_2-NI^\tau_2)Z_\tau+ B^2\left(a^\tau_2-NI^\tau_2\right)Y_\tau+ A+ B^2 I^\tau_2\right)\left(a^\tau_1-\tilde{a}^\tau_1\right)^2. \label{a32tau32difference} \end{align}\tag{41}\] Notice that Lemma 12 implies that \(|a^0_2|\) is decreasing in \(|A|\), and \(Z_0\) can be made small by choosing large \(M\) (e.g. \(M = |A|^4\)). For sufficiently large \(|A|\), the coefficient of \((a^\tau_1 - \tilde{a}^\tau_1)^2\) in 41 is negative. ◻
Department of Mathematics, Florida State University, Tallahassee, 32306; email: qfeng2@fsu.edu. This author is partially supported by the National Science Foundation under grant #DMS-2420029.↩︎
Department of Mathematical Sciences, Worcester Polytechnic Institute, Worcester, MA 01609 ; email: gwang2@wpi.edu. This author is partially supported by National Science Foundation grant #DMS - 2206282.↩︎
If \(f\) is concave and \(b\) and \(\sigma\) are linear in \(u\), then \(v(x,u)\) for any \(u\in\mathcal{A}(x)\) is concave in \(x\), and thus \(\mathbf{H}(x,\cdot,v_x,v_{xx})\) is concave in \(u\). See more details in Section 7.↩︎