arXiv Analytics

Sign in

arXiv:2204.04518 [cs.LG]AbstractReferencesReviewsResources

Attention U-Net as a surrogate model for groundwater prediction

Maria Luisa Taccari, Jonathan Nuttall, Xiaohui Chen, He Wang, Bennie Minnema, Peter K. Jimack

Published 2022-04-09Version 1

Numerical simulations of groundwater flow are used to analyze and predict the response of an aquifer system to its change in state by approximating the solution of the fundamental groundwater physical equations. The most used and classical methodologies, such as Finite Difference (FD) and Finite Element (FE) Methods, use iterative solvers which are associated with high computational cost. This study proposes a physics-based convolutional encoder-decoder neural network as a surrogate model to quickly calculate the response of the groundwater system. Holding strong promise in cross-domain mappings, encoder-decoder networks are applicable for learning complex input-output mappings of physical systems. This manuscript presents an Attention U-Net model that attempts to capture the fundamental input-output relations of the groundwater system and generates solutions of hydraulic head in the whole domain given a set of physical parameters and boundary conditions. The model accurately predicts the steady state response of a highly heterogeneous groundwater system given the locations and piezometric head of up to 3 wells as input. The network learns to pay attention only in the relevant parts of the domain and the generated hydraulic head field corresponds to the target samples in great detail. Even relative to coarse finite difference approximations the proposed model is shown to be significantly faster than a comparative state-of-the-art numerical solver, thus providing a base for further development of the presented networks as surrogate models for groundwater prediction.

Journal: Advances in Water Resources, Volume 163, May 2022, 104169
Categories: cs.LG
Related articles: Most relevant | Search more
arXiv:2105.14625 [cs.LG] (Published 2021-05-30)
Surrogate Model Based Hyperparameter Tuning for Deep Learning with SPOT
arXiv:2201.12896 [cs.LG] (Published 2022-01-30)
Augmenting Novelty Search with a Surrogate Model to Engineer Meta-Diversity in Ensembles of Classifiers
arXiv:2204.12088 [cs.LG] (Published 2022-04-26)
A physics-informed deep neural network for surrogate modeling in classical elasto-plasticity