Active Learning for Neural PDE Solvers

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Musekamp, Daniel, Kalimuthu, Marimuthu, Holzmüller, David, Takamoto, Makoto, Niepert, Mathias
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912288046317568
author Musekamp, Daniel
Kalimuthu, Marimuthu
Holzmüller, David
Takamoto, Makoto
Niepert, Mathias
author_facet Musekamp, Daniel
Kalimuthu, Marimuthu
Holzmüller, David
Takamoto, Makoto
Niepert, Mathias
contents Solving partial differential equations (PDEs) is a fundamental problem in science and engineering. While neural PDE solvers can be more efficient than established numerical solvers, they often require large amounts of training data that is costly to obtain. Active learning (AL) could help surrogate models reach the same accuracy with smaller training sets by querying classical solvers with more informative initial conditions and PDE parameters. While AL is more common in other domains, it has yet to be studied extensively for neural PDE solvers. To bridge this gap, we introduce AL4PDE, a modular and extensible active learning benchmark. It provides multiple parametric PDEs and state-of-the-art surrogate models for the solver-in-the-loop setting, enabling the evaluation of existing and the development of new AL methods for neural PDE solving. We use the benchmark to evaluate batch active learning algorithms such as uncertainty- and feature-based methods. We show that AL reduces the average error by up to 71% compared to random sampling and significantly reduces worst-case errors. Moreover, AL generates similar datasets across repeated runs, with consistent distributions over the PDE parameters and initial conditions. The acquired datasets are reusable, providing benefits for surrogate models not involved in the data generation.
format Preprint
id arxiv_https___arxiv_org_abs_2408_01536
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Active Learning for Neural PDE Solvers
Musekamp, Daniel
Kalimuthu, Marimuthu
Holzmüller, David
Takamoto, Makoto
Niepert, Mathias
Machine Learning
Artificial Intelligence
Computational Engineering, Finance, and Science
Neural and Evolutionary Computing
Solving partial differential equations (PDEs) is a fundamental problem in science and engineering. While neural PDE solvers can be more efficient than established numerical solvers, they often require large amounts of training data that is costly to obtain. Active learning (AL) could help surrogate models reach the same accuracy with smaller training sets by querying classical solvers with more informative initial conditions and PDE parameters. While AL is more common in other domains, it has yet to be studied extensively for neural PDE solvers. To bridge this gap, we introduce AL4PDE, a modular and extensible active learning benchmark. It provides multiple parametric PDEs and state-of-the-art surrogate models for the solver-in-the-loop setting, enabling the evaluation of existing and the development of new AL methods for neural PDE solving. We use the benchmark to evaluate batch active learning algorithms such as uncertainty- and feature-based methods. We show that AL reduces the average error by up to 71% compared to random sampling and significantly reduces worst-case errors. Moreover, AL generates similar datasets across repeated runs, with consistent distributions over the PDE parameters and initial conditions. The acquired datasets are reusable, providing benefits for surrogate models not involved in the data generation.
title Active Learning for Neural PDE Solvers
topic Machine Learning
Artificial Intelligence
Computational Engineering, Finance, and Science
Neural and Evolutionary Computing
url https://arxiv.org/abs/2408.01536