One-shot acceleration of transient PDE solvers via online-learned preconditioners

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Khodak, Mikhail, Jung, Min Ki, Wynne, Brian, Chow, Edmond, Kolemen, Egemen
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911301095129088
author Khodak, Mikhail
Jung, Min Ki
Wynne, Brian
Chow, Edmond
Kolemen, Egemen
author_facet Khodak, Mikhail
Jung, Min Ki
Wynne, Brian
Chow, Edmond
Kolemen, Egemen
contents Data-driven acceleration of scientific computing workflows has been a high-profile aim of machine learning (ML) for science, with numerical simulation of transient partial differential equations (PDEs) being one of the main applications. The focus thus far has been on methods that require classical simulations to train, which when combined with the data-hungriness and optimization challenges of neural networks has caused difficulties in demonstrating a convincing advantage against strong classical baselines. We consider an alternative paradigm in which the learner uses a classical solver's own data to accelerate it, enabling a one-shot speedup of the simulation. Concretely, since transient PDEs often require solving a sequence of related linear systems, the feedback from repeated calls to a linear solver such as preconditioned conjugate gradient (PCG) can be used by a bandit algorithm to online-learn an adaptive sequence of solver configurations (e.g. preconditioners). The method we develop, PCGBandit, is implemented directly on top of the popular open source software OpenFOAM, which we use to show its effectiveness on a set of fluid and magnetohydrodynamics (MHD) problems.
format Preprint
id arxiv_https___arxiv_org_abs_2509_08765
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle One-shot acceleration of transient PDE solvers via online-learned preconditioners
Khodak, Mikhail
Jung, Min Ki
Wynne, Brian
Chow, Edmond
Kolemen, Egemen
Computational Physics
Machine Learning
Numerical Analysis
Data-driven acceleration of scientific computing workflows has been a high-profile aim of machine learning (ML) for science, with numerical simulation of transient partial differential equations (PDEs) being one of the main applications. The focus thus far has been on methods that require classical simulations to train, which when combined with the data-hungriness and optimization challenges of neural networks has caused difficulties in demonstrating a convincing advantage against strong classical baselines. We consider an alternative paradigm in which the learner uses a classical solver's own data to accelerate it, enabling a one-shot speedup of the simulation. Concretely, since transient PDEs often require solving a sequence of related linear systems, the feedback from repeated calls to a linear solver such as preconditioned conjugate gradient (PCG) can be used by a bandit algorithm to online-learn an adaptive sequence of solver configurations (e.g. preconditioners). The method we develop, PCGBandit, is implemented directly on top of the popular open source software OpenFOAM, which we use to show its effectiveness on a set of fluid and magnetohydrodynamics (MHD) problems.
title One-shot acceleration of transient PDE solvers via online-learned preconditioners
topic Computational Physics
Machine Learning
Numerical Analysis
url https://arxiv.org/abs/2509.08765