Multi-fidelity wavelet neural operator surrogate model for time-independent and time-dependent reliability analysis

Abstract

Operator learning frameworks have recently emerged as an effective scientific machine learning tool for learning complex nonlinear operators of differential equations. Since neural operators learn an infinite-dimensional functional mapping, it is useful in applications requiring rapid prediction of solutions for a wide range of input functions. A task of a similar nature arises in many applications of uncertainty quantification, including reliability estimation and design under uncertainty, each of which demands thousands of samples subjected to a wide range of possible input conditions, an aspect to which neural operators are specialized. Although the neural operators are capable of learning complex nonlinear solution operators, they require an extensive amount of data for successful training. Unlike the applications in computer vision, the computational complexity of the numerical simulations and the cost of physical experiments contributing to the synthetic and real training data compromise the performance of the trained neural operator model, thereby directly impacting the accuracy of uncertainty quantification results. We aim to alleviate the data bottleneck by using multi-fidelity learning in neural operators, where a neural operator is trained by using a large amount of inexpensive low-fidelity data along with a small amount of expensive high-fidelity data. We propose the multi-fidelity wavelet neural operator, capable of learning solution operators from a multi-fidelity dataset, for efficient and effective data-driven reliability analysis of dynamical systems. We illustrate the performance of the proposed framework on bi-fidelity data simulated on coarse and refined grids for spatial and spatiotemporal systems.