BelNet : basis enhanced learning, a mesh-free neural operator

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

6 Scopus Citations
View graph of relations

Author(s)

Related Research Unit(s)

Detail(s)

Original languageEnglish
Article number20230043
Journal / PublicationProceedings of the Royal Society A: Mathematical, Physical and Engineering Sciences
Volume479
Issue number2276
Online published30 Aug 2023
Publication statusPublished - Aug 2023

Abstract

Operator learning trains a neural network to map functions to functions. An ideal operator learning framework should be mesh-free in the sense that the training does not require a particular choice of discretization for the input functions, allows for the input and output functions to be on different domains, and is able to have different grids between samples. We propose a mesh-free neural operator for solving parametric partial differential equations. The basis enhanced learning network (BelNet) projects the input function into a latent space and reconstructs the output functions. In particular, we construct part of the network to learn the 'basis' functions in the training process. This generalized the networks proposed in Chen & Chen (Chen and Chen 1995 IEEE Trans. Neural Netw. 49, 911-917. (doi:10.1109/72.392253) and 6, 904-910. (doi:10.1109/IJCNN.1993.716815)) to account for differences in input and output meshes. Through several challenging high-contrast and multiscale problems, we show that our approach outperforms other operator learning methods for these tasks and allows for more freedom in the sampling and/or discretization process. © 2023 The Author(s) Published by the Royal Society. All rights reserved.

Research Area(s)

  • discretization invariant, multiscale problem, operator learning, partial differential equation