A solver-in-the-loop framework for end-to-end differentiable coastal hydrodynamics

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cardoso-Bihlo, Elsa, Bihlo, Alex
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918434970796032
author Cardoso-Bihlo, Elsa
Bihlo, Alex
author_facet Cardoso-Bihlo, Elsa
Bihlo, Alex
contents Numerical simulation of wave propagation and run-up is a cornerstone of coastal engineering and tsunami hazard assessment. However, applying these forward models to inverse problems, such as bathymetry estimation, source inversion, and structural optimization, remains notoriously difficult due to the rigidity and high computational cost of deriving discrete adjoints. In this paper, we introduce AegirJAX, a fully differentiable hydrodynamic solver based on the depth-integrated, non-hydrostatic shallow-water equations. By implementing the solver entirely within a reverse-mode automatic differentiation framework, AegirJAX treats the time-marching physics loop as a continuous computational graph. We demonstrate the framework's versatility across a suite of scientific machine learning tasks: (1) discovering regime-specific neural corrections for model misspecifications in highly dispersive wave propagation; (2) performing continuous topology optimization for breakwater design; (3) training recurrent neural networks in-the-loop for active wave cancellation; and (4) inverting hidden bathymetry and submarine landslide kinematics directly from downstream sensor data. The proposed differentiable paradigm fundamentally blurs the line between forward simulation and inverse optimization, offering a unified, end-to-end framework for coastal hydrodynamics.
format Preprint
id arxiv_https___arxiv_org_abs_2604_07129
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle A solver-in-the-loop framework for end-to-end differentiable coastal hydrodynamics
Cardoso-Bihlo, Elsa
Bihlo, Alex
Fluid Dynamics
Machine Learning
Numerical Analysis
Atmospheric and Oceanic Physics
Numerical simulation of wave propagation and run-up is a cornerstone of coastal engineering and tsunami hazard assessment. However, applying these forward models to inverse problems, such as bathymetry estimation, source inversion, and structural optimization, remains notoriously difficult due to the rigidity and high computational cost of deriving discrete adjoints. In this paper, we introduce AegirJAX, a fully differentiable hydrodynamic solver based on the depth-integrated, non-hydrostatic shallow-water equations. By implementing the solver entirely within a reverse-mode automatic differentiation framework, AegirJAX treats the time-marching physics loop as a continuous computational graph. We demonstrate the framework's versatility across a suite of scientific machine learning tasks: (1) discovering regime-specific neural corrections for model misspecifications in highly dispersive wave propagation; (2) performing continuous topology optimization for breakwater design; (3) training recurrent neural networks in-the-loop for active wave cancellation; and (4) inverting hidden bathymetry and submarine landslide kinematics directly from downstream sensor data. The proposed differentiable paradigm fundamentally blurs the line between forward simulation and inverse optimization, offering a unified, end-to-end framework for coastal hydrodynamics.
title A solver-in-the-loop framework for end-to-end differentiable coastal hydrodynamics
topic Fluid Dynamics
Machine Learning
Numerical Analysis
Atmospheric and Oceanic Physics
url https://arxiv.org/abs/2604.07129