TTRISK: Tensor Train Decomposition Algorithm for Risk Averse Optimization

TL;DR

This paper introduces TTRISK, a tensor train-based algorithm for high-dimensional risk-averse optimization problems involving differential equations, demonstrating its effectiveness through numerical experiments including PDE constraints and COVID-19 lockdown planning.

Contribution

The paper presents a novel tensor train decomposition algorithm for risk-averse optimization, incorporating adaptive smoothing and efficient preconditioning for large-scale problems.

Findings

Accurate CVaR optimization for high-dimensional systems.

Effective tensor train approximation reduces computational complexity.

Feasibility demonstrated on PDE constraints and COVID-19 planning.

Abstract

This article develops a new algorithm named TTRISK to solve high-dimensional risk-averse optimization problems governed by differential equations (ODEs and/or PDEs) under uncertainty. As an example, we focus on the so-called Conditional Value at Risk (CVaR), but the approach is equally applicable to other coherent risk measures. Both the full and reduced space formulations are considered. The algorithm is based on low rank tensor approximations of random fields discretized using stochastic collocation. To avoid non-smoothness of the objective function underpinning the CVaR, we propose an adaptive strategy to select the width parameter of the smoothed CVaR to balance the smoothing and tensor approximation errors. Moreover, unbiased Monte Carlo CVaR estimate can be computed by using the smoothed CVaR as a control variate. To accelerate the computations, we introduce an efficient preconditioner for the KKT system in the full space formulation.The numerical experiments demonstrate that the proposed method enables accurate CVaR optimization constrained by large-scale discretized systems. In particular, the first example consists of an elliptic PDE with random coefficients as constraints. The second example is motivated by a realistic application to devise a lockdown plan for United Kingdom under COVID-19. The results indicate that the risk-averse framework is feasible with the tensor approximations under tens of random variables.