Machine-Learned Preconditioners for Linear Solvers in Geophysical Fluid Flows

Jan Ackmann, Peter D. Düben, Tim N. Palmer, Piotr K. Smolarkiewicz

arXiv preprint · 06 October 2020 · ECMWF, NSF NCAR / UCAR, University of Oxford

In one sentence. Machine-learned preconditioners for the semi-implicit linear solver in a shallow-water model are competitive with conventional preconditioners and generalize beyond the training range.

Abstract

It is tested whether machine learning methods can be used for preconditioning to increase the performance of the linear solver -- the backbone of the semi-implicit, grid-point model approach for weather and climate models. Embedding the machine-learning method within the framework of a linear solver circumvents potential robustness issues that machine learning approaches are often criticized for, as the linear solver ensures that a sufficient, pre-set level of accuracy is reached. The approach does not require prior availability of a conventional preconditioner and is highly flexible regarding complexity and machine learning design choices. Several machine learning methods are used to learn the optimal preconditioner for a shallow-water model with semi-implicit timestepping that is conceptually similar to more complex atmosphere models. The machine-learning preconditioner is competitive with a conventional preconditioner and provides good results even if it is used outside of the dynamical range of the training dataset.

Labels

Labels and summary are generated by Claude and reviewed weekly. Wrong? Edit this entry.