Tailoring Reproducing Kernels for Optimal Control via Policy Iteration
Shengyuan Niu, Ali Bouland, Haoran Wang, Filippos Fotiadis, Andrew Kurdila, Andrea L'Afflitto, Sai Tej Paruchuri, Kyriakos G. Vamvoudakis
TL;DR
The paper develops a framework for tailoring reproducing kernel Hilbert spaces to nonlinear optimal control by embedding policy iteration in native spaces and deriving explicit, stepwise error bounds for both critic and actor updates. By constructing feature mappings that align RKHSs with the problem dynamics, it provides two Galerkin critic schemes (C1 with $Y=W$ and C2 with $Y=L^2(\Omega)$) and analyzes their error contributions, including sampling and approximation terms, with and without Bernstein inequalities. The work extends prior RKHS-based convergence analyses to include the actor update, yielding practical bounds that inform basis selection and conditioning. Numerical experiments on a nonlinear 2D system illustrate that smoother kernels improve final PI accuracy, though they can increase conditioning and computational cost, highlighting trade-offs between C1 and C2 and the potential for adaptive basis strategies. Overall, the approach bridges optimal control and statistical learning by providing provable, geometry-aware error bounds for kernel-based policy iteration in continuous-time settings.
Abstract
This paper presents a novel approach to formulating the actor-critic method for optimal control by casting policy iteration in reproducing kernel Hilbert spaces (RKHSs -- also known as native spaces). By tailoring the reproducing kernel and RKHS to the dynamics of the nonlinear optimal control problem, we leverage recent advancements in characterizing error bounds from statistical and machine learning theory. These approximations define a general strategy to select the bases of the actor-critic networks, and we formally guarantee for the first time that this basis selection procedure leads to closed-form error bounds for the individual steps of policy iteration. These bounds often have a geometric and computable form, making them potentially useful for a priori or a posteriori evaluation of candidate collections of scattered bases. Numerical studies subsequently provide qualitative evidence of the practical performance achieved for the full recursion using the algorithms and theory developed for the single-step error bounds.
