An Operator Learning Approach to Nonsmooth Optimal Control of Nonlinear PDEs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Song, Yongcun, Yuan, Xiaoming, Yue, Hangrui, Zeng, Tianyou
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913806859370496
author Song, Yongcun
Yuan, Xiaoming
Yue, Hangrui
Zeng, Tianyou
author_facet Song, Yongcun
Yuan, Xiaoming
Yue, Hangrui
Zeng, Tianyou
contents 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 efficiency 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 retraining of neural networks can be avoided and computational cost can be substantially alleviated. We affirmatively validate the efficiency of the primal-dual-based operator learning approach across a range of typical optimal control problems with nonlinear PDEs.
format Preprint
id arxiv_https___arxiv_org_abs_2409_14417
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An Operator Learning Approach to Nonsmooth Optimal Control of Nonlinear PDEs
Song, Yongcun
Yuan, Xiaoming
Yue, Hangrui
Zeng, Tianyou
Optimization and Control
49M41, 35Q93, 35Q90, 68T07, 65K05
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 efficiency 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 retraining of neural networks can be avoided and computational cost can be substantially alleviated. We affirmatively validate the efficiency of the primal-dual-based operator learning approach across a range of typical optimal control problems with nonlinear PDEs.
title An Operator Learning Approach to Nonsmooth Optimal Control of Nonlinear PDEs
topic Optimization and Control
49M41, 35Q93, 35Q90, 68T07, 65K05
url https://arxiv.org/abs/2409.14417