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

by   Yongcun Song, et al.

We consider a general class of nonsmooth optimal control problems with partial differential equation (PDE) constraints, which are very challenging due to its nonsmooth objective functionals and the resulting high-dimensional and ill-conditioned systems after discretization. We focus on the application of a primal-dual method, with which different types of variables can be treated individually and thus its main computation at each iteration only requires solving two PDEs. Our target is to accelerate the primal-dual method with either larger step sizes or operator learning techniques. For the accelerated primal-dual method with larger step sizes, its convergence can be still proved rigorously while it numerically accelerates the original primal-dual method in a simple and universal way. For the operator learning acceleration, we construct deep neural network surrogate models for the involved PDEs. Once a neural operator is learned, solving a PDE requires only a forward pass of the neural network, and the computational cost is thus substantially reduced. The accelerated primal-dual method with operator learning is mesh-free, numerically efficient, and scalable to different types of PDEs. The acceleration effectiveness of these two techniques is promisingly validated by some preliminary numerical results.


page 1

page 2

page 3

page 4


A nonsmooth primal-dual method with simultaneous adaptive PDE constraint solver

We introduce an efficient first-order primal-dual method for the solutio...

A semigroup method for high dimensional elliptic PDEs and eigenvalue problems based on neural networks

In this paper, we propose a semigroup method for solving high-dimensiona...

Alternative design of DeepPDNet in the context of image restoration

This work designs an image restoration deep network relying on unfolded ...

A primal dual projection algorithm for efficient constraint preconditioning

We consider a linear iterative solver for large scale linearly constrain...

Toward Designing Intelligent PDEs for Computer Vision: An Optimal Control Approach

Many computer vision and image processing problems can be posed as solvi...

Accelerating Deep Unrolling Networks via Dimensionality Reduction

In this work we propose a new paradigm for designing efficient deep unro...

Accelerated PDE's for efficient solution of regularized inversion problems

We further develop a new framework, called PDE Acceleration, by applying...

Please sign up or login with your details

Forgot password? Click here to reset