Bathymetry Inversion using a Deep-Learning-Based Surrogate for Shallow Water Equations Solvers
Not on the map. Why?
Abstract (arXiv)
River bathymetry is critical for many aspects of water resources management. We propose and demonstrate a bathymetry inversion method using a deep-learning-based surrogate for shallow water equations solvers. The surrogate uses the convolutional autoencoder with a shared-encoder, separate-decoder architecture. It encodes the input bathymetry and decodes to separate outputs for flow-field variables. A gradient-based optimizer is used to perform bathymetry inversion with the trained surrogate. Two physically-based constraints on both bed elevation value and slope have to be added as inversion loss regularizations to obtain usable inversion results. Using the "L-curve" criterion, a heuristic approach was proposed to determine the regularization parameters. Both the surrogate model and the inversion algorithm show good performance. We found the bathymetry inversion process has two distinctive stages, which resembles the sculptural process of initial broad-brush calving and final detailing. The inversion loss due to flow prediction error reaches its minimum in the first stage and remains almost constant afterward. The bed elevation value and slope regularizations play the dominant role in the second stage in selecting the most probable solution. We also found the surrogate architecture (whether with both velocity and water surface elevation or velocity only as outputs) does not show significant impact on inversion result.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 10 of its 65 references that are in the map. First posted on arXiv 5 Mar 2022.
Builds on
10 of 65 references.
References · 10
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015Automatic differentiation in machine learning: a surveycited by 855
- 2017A trans-disciplinary review of deep learning research for water resources scientistscited by 47
- 2020Benchmarking deep inverse models over time, and the neural-adjoint methodcited by 28
- 2020From calibration to parameter learning: Harnessing the scaling effects of big data in geoscientific modelingcited by 26
- 2017Inversion using a new low-dimensional representation of complex binary geological media based on a deep neural networkcited by 24
- 2020Markov Chain Monte Carlo with Neural Network Surrogates: Application to Contaminant Source Identificationcited by 9
- 2020Application of deep learning to large scale riverine flow velocity estimationcited by 5
- 2021Surrogate Model for Shallow Water Equations Solvers with Deep Learningcited by 2
Built on by
Citing papers per year: 2023: 2; 2024: 0; 2025: 2; 2026: 2.
6 papers.
- 2023Differentiable modeling to unify machine learning and physical models and advance Geosciencescited by 4
- 2025Scientific Machine Learning of Flow Resistance Using Universal Shallow Water Equations with Differentiable Programmingcited by 3
- 2025A Neural Operator Emulator for Coastal and Riverine Shallow Water Dynamicscited by 2
- 2023PETAL: Physics Emulation Through Averaged Linearizations for Solving Inverse Problemscited by 1
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospectscited by 1
- 2026Finite Volume-Informed Neural Network Framework for 2D Shallow Water Equations: Rugged Loss Landscapes and the Importance of Data Guidancecited by 0
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospects16 May 2026
- 2026Finite Volume-Informed Neural Network Framework for 2D Shallow Water Equations: Rugged Loss Landscapes and the Importance of Data Guidance9 May 2026
- 2025A Neural Operator Emulator for Coastal and Riverine Shallow Water Dynamics20 Feb 2025
- 2025Scientific Machine Learning of Flow Resistance Using Universal Shallow Water Equations with Differentiable Programming18 Feb 2025
- 2023PETAL: Physics Emulation Through Averaged Linearizations for Solving Inverse Problems18 May 2023
- 2023Differentiable modeling to unify machine learning and physical models and advance Geosciences10 Jan 2023