Context-aware learning of hierarchies of low-fidelity models for multi-fidelity uncertainty quantification

TL;DR

This paper introduces a context-aware multi-fidelity Monte Carlo method that optimally balances training and sampling costs across hierarchies of low-fidelity models, significantly speeding up uncertainty quantification in complex simulations.

Contribution

It generalizes previous bi-fidelity methods to hierarchies of models, incorporating context to optimize variance reduction and computational efficiency.

Findings

Achieves up to 100x speedup in uncertainty quantification.

Reduces simulation runtime from 72 days to 4 hours.

Applicable to diverse low-fidelity models like sparse-grid and deep networks.

Abstract

Multi-fidelity Monte Carlo methods leverage low-fidelity and surrogate models for variance reduction to make tractable uncertainty quantification even when numerically simulating the physical systems of interest with high-fidelity models is computationally expensive. This work proposes a context-aware multi-fidelity Monte Carlo method that optimally balances the costs of training low-fidelity models with the costs of Monte Carlo sampling. It generalizes the previously developed context-aware bi-fidelity Monte Carlo method to hierarchies of multiple models and to more general types of low-fidelity models. When training low-fidelity models, the proposed approach takes into account the context in which the learned low-fidelity models will be used, namely for variance reduction in Monte Carlo estimation, which allows it to find optimal trade-offs between training and sampling to minimize upper bounds of the mean-squared errors of the estimators for given computational budgets. This is in stark contrast to traditional surrogate modeling and model reduction techniques that construct low-fidelity models with the primary goal of approximating well the high-fidelity model outputs and typically ignore the context in which the learned models will be used in upstream tasks. The proposed context-aware multi-fidelity Monte Carlo method applies to hierarchies of a wide range of types of low-fidelity models such as sparse-grid and deep-network models. Numerical experiments with the gyrokinetic simulation code \textsc{Gene} show speedups of up to two orders of magnitude compared to standard estimators when quantifying uncertainties in small-scale fluctuations in confined plasma in fusion reactors. This corresponds to a runtime reduction from 72 days to about four hours on one node of the Lonestar6 supercomputer at the Texas Advanced Computing Center.