Exploratory HJB Equations and Their Convergence
Wenpin Tang, Yuming Paul Zhang, Xun Yu Zhou
Abstract
.We study the exploratory Hamilton–Jacobi–Bellman (HJB) equation arising from the entropy-regularized exploratory control problem, which was formulated by Wang, Zariphopoulou, and Zhou (J. Mach. Learn. Res., 21 (2020), 198) in the context of reinforcement learning in continuous time and space. We establish the well-posedness and regularity of the viscosity solution to the equation, as well as the convergence of the exploratory control problem to the classical stochastic control problem when the level of exploration decays to zero. We then apply the general results obtained to the exploratory temperature control problem, which was introduced by Gao, Xu, and Zhou (SIAM J. Control Optim., 60 (2022), pp. 1250–1268) to design an endogenous temperature schedule for simulated annealing in the context of nonconvex optimization. We derive an explicit rate of convergence for this problem as exploration diminishes to zero, and find that the stationary distribution of the optimally controlled process exists, which is however neither a Dirac mass on the global optimum nor a Gibbs measure.KeywordsHJB equationsstochastic controlpartial differential equationsreinforcement learningexploratory controlentropy regularizationsimulated annealingoverdamped Langevin equationMSC codes35F2160J6093E1593E20