A successive approximation method in functional spaces for hierarchical optimal control problems and its application to learning

TL;DR

This paper introduces a hierarchical optimal control framework for high-dimensional nonlinear function learning, combining controllability and regularization objectives, and proposes a successive approximation algorithm with numerical validation.

Contribution

It establishes a novel connection between learning problems and hierarchical optimal control, and develops a nested successive approximation method for solving such problems.

Findings

Hierarchical control framework effectively balances generalization and regularization.

Proposed algorithm converges to optimal parameter estimates in nonlinear regression.

Numerical results demonstrate practical applicability of the method.

Abstract

We consider a class of learning problem of point estimation for modeling high-dimensional nonlinear functions, whose learning dynamics is guided by model training dataset, while the estimated parameter in due course provides an acceptable prediction accuracy on a different model validation dataset. Here, we establish an evidential connection between such a learning problem and a hierarchical optimal control problem that provides a framework how to account appropriately for both generalization and regularization at the optimization stage. In particular, we consider the following two objectives: (i) The first one is a controllability-type problem, i.e., generalization, which consists of guaranteeing the estimated parameter to reach a certain target set at some fixed final time, where such a target set is associated with model validation dataset. (ii) The second one is a regularization-type problem ensuring the estimated parameter trajectory to satisfy some regularization property over a certain finite time interval. First, we partition the control into two control strategies that are compatible with two abstract agents, namely, a leader, which is responsible for the controllability-type problem and that of a follower, which is associated with the regularization-type problem. Using the notion of Stackelberg's optimization, we provide conditions on the existence of admissible optimal controls for such a hierarchical optimal control problem under which the follower is required to respond optimally to the strategy of the leader, so as to achieve the overall objectives that ultimately leading to an optimal parameter estimate. Moreover, we provide a nested algorithm, arranged in a hierarchical structure-based on successive approximation methods, for solving the corresponding optimal control problem. Finally, we present some numerical results for a typical nonlinear regression problem.