Reduced-Order Surrogates for Forced Flexible Mesh Coastal-Ocean Models
Pith reviewed 2026-05-16 07:28 UTC · model grok-4.3
The pith
Koopman autoencoders with forcings deliver accurate year-long reduced-order surrogates for coastal-ocean models.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
A Koopman autoencoder that maps the state to latent space and applies a learned linear temporal operator there, regularized for eigenvalue stability and trained on forcings and boundary conditions, produces bounded long-term trajectories whose accuracy reaches relative RMS errors of 0.0068-0.14 across three coastal regimes, matching or exceeding POD performance while remaining within practical tolerance of in-situ observations.
What carries the argument
Koopman autoencoder whose encoder-decoder pair produces a latent representation governed by a learned linear temporal operator that is regularized for eigenvalue stability and driven by meteorological forcings and boundary conditions.
If this is right
- The 300-1400x inference speed-up enables ensemble forecasting and multi-decadal climate simulations that remain computationally infeasible with the full physics model.
- Water-surface-elevation errors stay within a few centimeters of the physics-based model, keeping the surrogate usable for many operational coastal applications.
- In two of the three tested regimes the Koopman formulation yields higher accuracy than POD-based surrogates under identical temporal-unrolling conditions.
- Current-velocity fields show the largest relative errors while water-surface elevations remain the most accurate variable across all cases.
Where Pith is reading between the lines
- The same latent-linear-operator construction could be applied to other externally forced fluid systems such as atmospheric or riverine models where full physics runs are too slow.
- If eigenvalue regularization continues to bound trajectories under modest forcing changes, the surrogates could support real-time nowcasting without frequent retraining.
- Systematic tests on extreme-event forcings would clarify whether the current accuracy levels hold when the underlying dynamics deviate strongly from the training distribution.
Load-bearing premise
The learned linear temporal operator in latent space remains stable and accurate when driven by real meteorological forcings outside the three tested regimes.
What would settle it
Long-term simulation under forcings from a fourth untested dynamical regime in which the surrogate's water-level or velocity trajectories diverge from the full physics model or in-situ data by amounts substantially larger than the reported 0.0068-0.14 relative RMS range.
Figures
read the original abstract
While proper orthogonal decomposition (POD)-based surrogates are widely explored for hydrodynamic applications, the use of Koopman autoencoders for real-world coastal-ocean modelling remains relatively limited. This paper introduces a flexible Koopman autoencoder formulation that incorporates meteorological forcings and boundary conditions, and systematically compares its performance against POD-based surrogates. The Koopman autoencoder employs a learned linear temporal operator in latent space, enabling eigenvalue regularization to promote temporal stability. This strategy is evaluated alongside temporal unrolling techniques for achieving stable and accurate long-term predictions. The models are assessed on three test cases spanning distinct dynamical regimes, with prediction horizons up to one year at 30-minute temporal resolution. Across all cases, the reduced order surrogates with temporal unrolling achieve high accuracy with relative root-mean-squared-errors of 0.0068-0.14 and $R^2$-values of 0.61-0.995, where prediction errors are largest for current velocities, and smallest for water surface elevations. In two of the three cases, the Koopman Autoencoder have higher accuracy than the POD-based surrogates. Comparing to in-situ observations, the surrogate yields -0.64% to 12% increase in water surface elevation prediction error when compared to prediction errors of the physics-based model. These error levels, corresponding to a few centimeters, are acceptable for many practical applications, while inference speed-ups of 300-1400x enables workflows such as ensemble forecasting and long climate simulations for coastal-ocean modelling.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces a flexible Koopman autoencoder formulation for reduced-order surrogates of forced flexible-mesh coastal-ocean models. The approach learns a linear temporal operator in latent space with eigenvalue regularization for stability, incorporates meteorological forcings and boundary conditions, and employs temporal unrolling for long-term rollouts. It systematically compares performance to POD-based surrogates on three distinct dynamical regimes with prediction horizons up to one year at 30-minute resolution, reporting rRMSE values of 0.0068–0.14 and R² values of 0.61–0.995 (best for water surface elevation, worst for currents). In two of three cases the Koopman model outperforms POD; against in-situ observations the surrogate increases water-level error by –0.64% to 12% relative to the full physics model while delivering 300–1400× speed-ups.
Significance. If the reported accuracy and stability hold under broader forcing conditions, the work would be significant for operational coastal modeling. It supplies concrete, reproducible error metrics against both physics-based output and real observations, demonstrates practical speed-ups that enable ensemble forecasting and long climate runs, and shows that a regularized Koopman operator can match or exceed POD accuracy in forced regimes. These elements directly address the computational bottleneck in coastal-ocean applications.
major comments (2)
- The headline claim that the regularized Koopman operator produces bounded long-term trajectories under real meteorological forcings rests on the three tested regimes spanning the relevant variability. No explicit out-of-distribution sensitivity analysis (e.g., scaling wind or pressure amplitudes beyond the training envelope) is described; if the latent eigenvalues drift under distribution shift, the 1-year rollouts would diverge even if short-term training metrics remain good. This assumption is load-bearing for the operational utility asserted in the abstract.
- The manuscript states that temporal unrolling is combined with eigenvalue regularization to achieve stable predictions, yet the precise interaction between the two mechanisms (e.g., how the regularization term is weighted during unrolled training) is not quantified in the results. Without this, it is unclear whether the reported rRMSE and R² values are attributable to the Koopman operator itself or to the unrolling procedure.
minor comments (2)
- Clarify the exact architecture of the encoder/decoder (number of layers, latent dimension) and the form of the learned linear operator matrix; these details are needed for reproducibility even if the abstract reports aggregate metrics.
- The abstract reports R² = 0.61–0.995; the lower end of this range should be discussed explicitly—does it correspond to a particular variable or regime, and does it still support the claim of “high accuracy” for practical use?
Simulated Author's Rebuttal
We thank the referee for the detailed and constructive report. The two major comments identify important aspects of the stability claims and training procedure that warrant clarification and additional analysis. We address each point below and will incorporate revisions to strengthen the manuscript.
read point-by-point responses
-
Referee: The headline claim that the regularized Koopman operator produces bounded long-term trajectories under real meteorological forcings rests on the three tested regimes spanning the relevant variability. No explicit out-of-distribution sensitivity analysis (e.g., scaling wind or pressure amplitudes beyond the training envelope) is described; if the latent eigenvalues drift under distribution shift, the 1-year rollouts would diverge even if short-term training metrics remain good. This assumption is load-bearing for the operational utility asserted in the abstract.
Authors: We agree that the long-term boundedness claim would be more robust with explicit out-of-distribution testing. In the revised manuscript we will add a dedicated sensitivity subsection that scales wind speed and pressure amplitudes by factors of 1.5× and 2.0× relative to the training envelope, reports the resulting shifts in latent eigenvalues, and quantifies rollout divergence over the one-year horizon. These results will be presented alongside the existing three-regime evaluation. revision: yes
-
Referee: The manuscript states that temporal unrolling is combined with eigenvalue regularization to achieve stable predictions, yet the precise interaction between the two mechanisms (e.g., how the regularization term is weighted during unrolled training) is not quantified in the results. Without this, it is unclear whether the reported rRMSE and R² values are attributable to the Koopman operator itself or to the unrolling procedure.
Authors: We will revise the methods and results sections to quantify the interaction. The revised text will explicitly state the weighting coefficient λ_reg used for the eigenvalue regularization term within the unrolled loss, provide the full composite loss expression, and include an ablation table comparing (i) unrolling alone, (ii) regularization alone, and (iii) the combined formulation on the same three test cases. This will isolate the contribution of each component to the reported accuracy metrics. revision: yes
Circularity Check
No load-bearing circularity; accuracy claims rest on independent physics-based and observational benchmarks
full rationale
The paper trains Koopman autoencoders (with learned linear latent operator and eigenvalue regularization) and POD surrogates on data generated by the full physics-based coastal-ocean model, then reports relative RMSE and R² on held-out temporal segments against the same model's outputs plus in-situ observations. These metrics are external to the fitted parameters; the central result is an empirical performance comparison across three regimes, not a derivation that reduces to its own inputs by construction. No self-definitional equations, fitted-input-as-prediction, or load-bearing self-citations appear in the described workflow. Eigenvalue regularization is a standard training constraint, not a tautology that forces the reported accuracy numbers.
Axiom & Free-Parameter Ledger
Reference graph
Works this paper leans on
-
[1]
Fore- casting Sequential Data using Consistent Koopman Autoencoders. doi:10.48550/arXiv.2003.02236. Bogaard, T., de Lima Rego, J., Vatvani, D., Virasami, R., Verlaan, M.,
-
[2]
Brunton, S.L., Brunton, B.W., Proctor, J.L., Kutz, J.N.,
doi:10.1029/2024GL112835. Brunton, S.L., Brunton, B.W., Proctor, J.L., Kutz, J.N.,
-
[3]
doi:10.1371/journal.pone.0150171. Cho, K., van Merrienboer, B., Bahdanau, D., Bengio, Y.,
-
[4]
On the Properties of Neural Machine Translation: Encoder-Decoder Approaches
On the properties of neural machine translation: Encoder-decoder approaches. arXiv:1409.1259. Città di Venezia,
work page internal anchor Pith review Pith/arXiv arXiv
-
[5]
piattaforma ismar-cnr. URL:https://comune. venezia.it/it/content/3-piattaforma-ismar-cnr. visited: 2025-12-
work page 2025
-
[6]
CMEMS, 2024a. E.U. Copernicus Marine Service Information (CMEMS). In- situ ocean TAC. URL:https://marineinsitu.eu/dashboard/. visited: 2024-11-11. CMEMS, 2024b. E.U. Copernicus Marine Service Information (CMEMS). Marine Data Store (MDS): Baltic Sea- In Situ Near Real Time Observa- tions. doi:10.48670/moi-00032. visited: 2024-10-02. Darts,
-
[7]
Torch Forecasting Models. URL:https://unit8co.github. io/darts/userguide/torch_forecasting_models. visited: 2026-04-16. 31 Deo, M., Sridhar Naidu, C.,
work page 2026
-
[8]
Real time wave forecasting using neural networks. Ocean Engineering 26, 191–203. doi:10.1016/S0029- 8018(97)10025-7. DHI,
-
[9]
Ijmuiden ver wind farm zone. part a: Metocean modelling. URL: https://offshorewind.rvo.nl/file/download/bfa49f34-f894-4562- 882f-eb1a8b7497e9/ijv_20231222-dhi-metocean-modelling.pdf. DHI, 2024a. Hydrodynamic model of Øresund: MIKE 21 model setup, out- puts and observation data. doi:10.5281/zenodo.14160710. (1.0) [Dataset]. DHI, 2024b. MIKE 21 Flow Model F...
-
[10]
Finding structure with randomness: Probabilistic algorithms for constructing approximate matrix decompositions.arXiv:0909.4061. Herzen, J., Lässig, F., Piazzetta, S.G., Neuer, T., Tafti, L., Raille, G., Pot- telbergh, T.V., Pasieka, M., Skrodzki, A., Huguenin, N., Dumonal, M., Koscisz, J., Bader, D., Gusset, F., Benheddi, M., Williamson, C., Kosin- ski, M...
work page internal anchor Pith review Pith/arXiv arXiv
-
[11]
CoRR abs/2110.03224.arXiv:2110.03224
Darts: User-friendlymodernmachine learning for time series. CoRR abs/2110.03224.arXiv:2110.03224. Hochreiter, S., Schmidhuber, J.,
-
[12]
1997, Neural Computation, 9, 1735, 10.1162/neco.1997.9.8.1735
Long Short-Term Memory. Neural Computation 9, 1735–1780. doi:10.1162/neco.1997.9.8.1735. Jiang, P., Meinert, N., Jordão, H., Weisser, C., Holgate, S., Lavin, A., Lüt- jens, B., Newman, D., Wainwright, H., Walker, C., Barnard, P.,
-
[13]
Digital twin earth – coasts: Developing a fast and physics-informed surro- gate model for coastal floods via neural operators.arXiv:2110.07100. 32 Kaiser, E., Kutz, J.N., Brunton, S.L.,
-
[14]
Machine Learning: Science and Technology 2, 035023
Data-driven discovery of Koop- man eigenfunctions for control. Machine Learning: Science and Technology 2, 035023. doi:10.1088/2632-2153/abf0f5. Kingma, D.P., Ba, J.,
-
[15]
Neural emulator superiority: When machine learning for PDEs surpasses its training data.arXiv:2510.23111. Koopman, B.O.,
-
[16]
Proceedings of the National Academy of Sciences 17, 315–318
Hamiltonian systems and transformation in Hilbert space. Proceedings of the National Academy of Sciences 17, 315–318. doi:10.1073/pnas.17.5.315. Lesser, G.R., Roelvink, D.A., van Kester, J.A.T.M., Stelling, G.S.,
-
[17]
Coastal Engineering 51, 883–915
Development and validation of a three-dimensional morphological model. Coastal Engineering 51, 883–915. doi:10.1016/j.coastaleng.2004.07.014. Lightning AI,
-
[18]
The ai cloud pytorch developers love. URL: https://lightning.ai/. visited: 2025-12-17. List, B., Chen, L.W., Bali, K., Thuerey, N.,
work page 2025
-
[19]
Differentiability in unrolled training of neural physics simulators on transient dynamics. arXiv:2402.12971. Lortie, L., Dahdah, S., Forbes, J.R.,
-
[20]
Forward-Backward Extended DMD with an Asymptotic Stability Constraint.arXiv:2403.10623. Loshchilov, I., Hutter, F.,
-
[21]
Decoupled Weight Decay Regularization
Fixing weight decay regularization in Adam. CoRR abs/1711.05101.arXiv:1711.05101. Luettich, R.A., Westerink, J.J., Scheffner, N.W.,
work page internal anchor Pith review Pith/arXiv arXiv
- [22]
-
[23]
Spectral Properties of Dynamical Systems, Model Reduction and Decompositions. Nonlinear Dynamics 41, 309–325. doi:10.1007/s11071-005-2824-x. 33 Morton, J., Witherden, F.D., Kochenderfer, M.J.,
-
[24]
Deep variational Koopman models: Inferring Koopman observations for uncertainty-aware dynamics modeling and control.arXiv:1902.09742. Nayak, I., Chakrabarti, A., Kumar, M., Teixeira, F.L., Goswami, D.,
work page internal anchor Pith review Pith/arXiv arXiv 1902
-
[25]
Temporally-consistent Koopman autoencoders for forecasting dynamical systems. Scientific Reports 15, 22127. doi:10.1038/s41598-025-05222-7. NEA,
-
[26]
Netherlands Enterprise Agency: IJmuiden Ver - Wind & Water. URL:https://offshorewind.rvo.nl/page/view/4c4cb8e3-a1c7- 404f-917f-a2f6e0d8c803/wind-en-water-ijmuiden-ver. visited: 2024- 11-08. Nogueira, A.C., Almeida, J.L.d.S., Auger, G., Watson, C.D.,
work page 2024
-
[27]
Springer International Publishing
Re- duced Order Modeling of Dynamical Systems Using Artificial Neural Net- works Applied to Water Circulation. Springer International Publishing. p. 116–136. doi:10.1007/978-3-030-59851-8_8. Otto, S.E., Rowley, C.W.,
-
[28]
Linearly-recurrent autoencoder networks for learning dynamicsarXiv:1712.01378. Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., Desmaison, A., Köpf, A., Yang, E., DeVito, Z., Raison, M., Tejani, A., Chilamkurthy, S., Steiner, B., Fang, L., Bai, J., Chintala, S.,
work page internal anchor Pith review Pith/arXiv arXiv
-
[29]
Pytorch: Animperativestyle, high-performance deep learning library.arXiv:1912.01703. Pedregosa, F., Varoquaux, G., Gramfort, A., Michel, V., Thirion, B., Grisel, O., Blondel, M., Prettenhofer, P., Weiss, R., Dubourg, V., Vanderplas, J., Passos, A., Cournapeau, D., Brucher, M., Perrot, M., Duchesnay, E.,
work page internal anchor Pith review Pith/arXiv arXiv 1912
-
[30]
Dynamic mode decomposition with control. SIAM Journal on Applied Dynamical Systems 15, 142–161. doi:10.1137/15M1013857. Proctor, J.L., Brunton, S.L., Kutz, J.N.,
-
[31]
Generalizing Koopman The- ory to Allow for Inputs and Control. SIAM Journal on Applied Dynamical Systems 17, 909–930. doi:10.1137/16M1062296. 34 Pytorch, 2025a. Adamw. URL:https://docs.pytorch.org/docs/stable/ generated/torch.optim.AdamW.html. urldate: 2025-12-19. Pytorch, 2025b. ReduceLROnPlateau. URL:https://docs.pytorch.org/ docs/stable/generated/torch...
-
[32]
Analyzing Koopman approaches to physics-informed machine learning for long-term sea-surface temperature forecasting.arXiv:2010.00399. Rivera-Casillas, P., Dutta, S., Cai, S., Loveland, M., Nath, K., Shukla, K., Trahan, C., Lee, J., Farthing, M., Dawson, C.,
-
[33]
A Neural Operator-Based Emulator for Regional Shallow Water Dynamics.arXiv pre-print
A Neu- ral Operator-Based Emulator for Regional Shallow Water Dynamics. doi:10.48550/arXiv.2502.14782. Rowley, C.W., Mezić, I., Bagheri, S., Schlatter, P., Henningson, D.S.,
-
[34]
doi: 10.1017/S0022112009992059
Spectral analysis of nonlinear flows. Journal of Fluid Mechanics 641, 115–127. doi:10.1017/S0022112009992059. Schmid, P.J.,
-
[35]
Dynamic mode decomposition of numerical and experimental data. Journal of Fluid Mechanics 656, 5–28. doi:10.1017/S0022112010001217. Shi, H., Meng, M.Q.H.,
-
[36]
Deep Koopman Operator with Control for Nonlinear Systems. doi:10.48550/arXiv.2202.08004. Takeishi, N., Kawahara, Y., Yairi, T.,
-
[37]
Coastal Engineering 21, 301–331
Statistical and hydrodynamic models for the operational forecasting of floods in the venice lagoon. Coastal Engineering 21, 301–331. doi:10.1016/0378-3839(93)90012-W. Williams, M.O., Kevrekidis, I.G., Rowley, C.W.,
-
[38]
Journal of Nonlinear Science , 1307–1346doi:10.1007/s00332-015- 9258-5
A data–driven ap- proximation of the koopman operator: Extending dynamic mode decom- position. Journal of Nonlinear Science , 1307–1346doi:10.1007/s00332-015- 9258-5. 35 Appendix A. Test cases This section presents information relevant to the three test cases. Appendix A.1. Forcing variables and dimensions Table A.5 gives an overview of the forcing variab...
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.