Taylor approximation and variance reduction for PDE-constrained optimal control under uncertainty

Abstract

In this work we develop a scalable computational framework for the solution of PDE-constrained optimal control under high-dimensional uncertainty. Specifically, we consider a mean-variance formulation of the control objective and employ a Taylor expansion with respect to the uncertain parameter either to directly approximate the control objective or as a control variate for variance reduction. The expressions for the mean and variance of the Taylor approximation are known analytically, although their evaluation requires efficient computation of the trace of the (preconditioned) Hessian of the control objective. We propose to estimate this trace by solving a generalized eigenvalue problem using a randomized algorithm that only requires the action of the Hessian on a small number of random directions. Then, the computational work does not depend on the nominal dimension of the uncertain parameter, but depends only on the effective dimension, thus ensuring scalability to high-dimensional problems. Moreover, to increase the estimation accuracy of the mean and variance of the control objective by the Taylor approximation, we use it as a control variate for variance reduction, which results in considerable computational savings (several orders of magnitude) compared to a plain Monte Carlo method. We demonstrate the accuracy, efficiency, and scalability of the proposed computational method for two examples with high-dimensional uncertain parameters: subsurface flow in a porous medium modeled as an elliptic PDE, and turbulent jet flow modeled by the Reynolds-averaged Navier--Stokes equations coupled with a nonlinear advection-diffusion equation characterizing model uncertainty. In particular, for the latter more challenging example we show scalability of our algorithm up to one million parameters resulting from discretization of the uncertain parameter field.

References in corpus (7)

• Scalable and efficient algorithms for the propagation of uncertainty from data through inference to prediction for large-scale problems, with application to flow of the Antarctic ice sheet
• Optimal control with stochastic PDE constraints and uncertain controls
• Improved bounds on sample size for implicit matrix trace estimators
• Optimal design of acoustic metamaterial cloaks under uncertainty
• Hessian-based adaptive sparse quadrature for infinite-dimensional Bayesian inverse problems
• A-optimal encoding weights for nonlinear inverse problems, with applications to the Helmholtz inverse problem
• Sparse polynomial approximation for optimal control problems constrained by elliptic PDEs with lognormal random coefficients

Cited by in corpus (13)

• Optimal design of acoustic metamaterial cloaks under uncertainty
• Bayesian inference of heterogeneous epidemic models: Application to COVID-19 spread accounting for long-term care facilities
• A Stochastic Gradient Method with Mesh Refinement for PDE Constrained Optimization under Uncertainty
• A fast and scalable computational framework for large-scale and high-dimensional Bayesian optimal experimental design
• Parabolic PDE-constrained optimal control under uncertainty with entropic risk measure using quasi-Monte Carlo integration
• hIPPYlib: An Extensible Software Framework for Large-Scale Inverse Problems Governed by PDEs; Part I: Deterministic Inversion and Linearized Bayesian Inference
• Residual-Based Error Corrector Operator to Enhance Accuracy and Reliability of Neural Operator Surrogates of Nonlinear Variational Boundary-Value Problems
• Taylor approximation for chance constrained optimization problems governed by partial differential equations with high-dimensional random parameters
• Polynomial Selection in Spectral Graph Neural Networks: An Error-Sum of Function Slices Approach
• Optimality Conditions for Convex Stochastic Optimization Problems in Banach Spaces with Almost Sure State Constraints
• Robust optimal design of large-scale Bayesian nonlinear inverse problems
• Hessian-based sampling for high-dimensional model reduction
• Multimodal Information Gain in Bayesian Design of Experiments