Abstract:Optimal control problems with nonsmooth objectives and nonlinear partial differential equation (PDE) constraints are challenging, mainly because of the underlying nonsmooth and nonconvex structures and the demanding computational cost for solving multiple high-dimensional and ill-conditioned systems after mesh-based discretization. To mitigate these challenges numerically, we propose an operator learning approach in combination with an effective primal-dual optimization idea which can decouple the treatment of the control and state variables so that each of the resulting iterations only requires solving two PDEs. Our main purpose is to construct neural surrogate models for the involved PDEs by operator learning, allowing the solution of a PDE to be obtained with only a forward pass of the neural network. The resulting algorithmic framework offers a hybrid approach that combines the flexibility and generalization of operator learning with the model-based nature and structure-friendly efficiency of primal-dual-based algorithms. The primal-dual-based operator learning approach offers numerical methods that are mesh-free, easy to implement, and adaptable to various optimal control problems with nonlinear PDEs. It is notable that the neural surrogate models can be reused across iterations and parameter settings, hence computational cost can be substantially alleviated. We validate the effectiveness and efficiency of the primal-dual-based operator learning approach across a range of typical optimal control problems with nonlinear PDEs, including optimal control of stationary Burgers equations, sparse bilinear control of parabolic equations, and optimal control of semilinear parabolic equations.

A primal--dual algorithm as applied to optimal control problems

Chemical Process Optimization Approach Based on Primal-Dual Interior-Point Method

Optimal Control Problems with Incomplete and Different Integral Time Domains in the Objective and Constraints

An Operator Learning Approach to Nonsmooth Optimal Control of Nonlinear PDEs

A Primal-dual hybrid gradient method for solving optimal control problems and the corresponding Hamilton-Jacobi PDEs

Primal-Dual iLQR

Accelerated primal-dual methods with enlarged step sizes and operator learning for nonsmooth optimal control problems

A Unified Primal-Dual Algorithm Framework for Inequality Constrained Problems

A Primal-Dual Method for Optimal Control and Trajectory Generation in High-Dimensional Systems

Infeasible and Critically Feasible Optimal Control

Applying the canonical dual theory in optimal control problems

Optimal Control Duality and the Douglas-Rachford Algorithm

Primal-dual Interior Point QP-free Algorithm for Nonlinear Constrained Optimization

A feedback control approach to convex optimization with inequality constraints

An essentially decentralized interior point method for control

On the Differentiability of the Primal-Dual Interior-Point Method

A Mixed Variable Variational Method for Optimal Control Problems with Applications in Aerospace Control

Parallel D–D Type Domain Decomposition Algorithm for Optimal Control Problem Governed by Parabolic Partial Differential Equation

Primal-dual interior-point algorithm for linearly constrained convex optimization based on a parametric algebraic transformation

A PRIMAL-DUAL EXTERIOR POINT ALGORITHM FOR LINEAR PROGRAMMING PROBLEMS