English

Learning in latent spaces improves the predictive accuracy of deep neural operators

Machine Learning 2023-04-18 v1 Numerical Analysis Dynamical Systems Numerical Analysis

Abstract

Operator regression provides a powerful means of constructing discretization-invariant emulators for partial-differential equations (PDEs) describing physical systems. Neural operators specifically employ deep neural networks to approximate mappings between infinite-dimensional Banach spaces. As data-driven models, neural operators require the generation of labeled observations, which in cases of complex high-fidelity models result in high-dimensional datasets containing redundant and noisy features, which can hinder gradient-based optimization. Mapping these high-dimensional datasets to a low-dimensional latent space of salient features can make it easier to work with the data and also enhance learning. In this work, we investigate the latent deep operator network (L-DeepONet), an extension of standard DeepONet, which leverages latent representations of high-dimensional PDE input and output functions identified with suitable autoencoders. We illustrate that L-DeepONet outperforms the standard approach in terms of both accuracy and computational efficiency across diverse time-dependent PDEs, e.g., modeling the growth of fracture in brittle materials, convective fluid flows, and large-scale atmospheric flows exhibiting multiscale dynamical features.

Keywords

Cite

@article{arxiv.2304.07599,
  title  = {Learning in latent spaces improves the predictive accuracy of deep neural operators},
  author = {Katiana Kontolati and Somdatta Goswami and George Em Karniadakis and Michael D. Shields},
  journal= {arXiv preprint arXiv:2304.07599},
  year   = {2023}
}

Comments

22 pages, 12 figures