PDEBENCH: An Extensive Benchmark for Scientific Machine Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Takamoto, Makoto, Praditia, Timothy, Leiteritz, Raphael, MacKinlay, Dan, Alesiani, Francesco, Pflüger, Dirk, Niepert, Mathias
Format: Preprint
Publié: 2022
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912001143341056
author Takamoto, Makoto
Praditia, Timothy
Leiteritz, Raphael
MacKinlay, Dan
Alesiani, Francesco
Pflüger, Dirk
Niepert, Mathias
author_facet Takamoto, Makoto
Praditia, Timothy
Leiteritz, Raphael
MacKinlay, Dan
Alesiani, Francesco
Pflüger, Dirk
Niepert, Mathias
contents Machine learning-based modeling of physical systems has experienced increased interest in recent years. Despite some impressive progress, there is still a lack of benchmarks for Scientific ML that are easy to use but still challenging and representative of a wide range of problems. We introduce PDEBench, a benchmark suite of time-dependent simulation tasks based on Partial Differential Equations (PDEs). PDEBench comprises both code and data to benchmark the performance of novel machine learning models against both classical numerical simulations and machine learning baselines. Our proposed set of benchmark problems contribute the following unique features: (1) A much wider range of PDEs compared to existing benchmarks, ranging from relatively common examples to more realistic and difficult problems; (2) much larger ready-to-use datasets compared to prior work, comprising multiple simulation runs across a larger number of initial and boundary conditions and PDE parameters; (3) more extensible source codes with user-friendly APIs for data generation and baseline results with popular machine learning models (FNO, U-Net, PINN, Gradient-Based Inverse Method). PDEBench allows researchers to extend the benchmark freely for their own purposes using a standardized API and to compare the performance of new models to existing baseline methods. We also propose new evaluation metrics with the aim to provide a more holistic understanding of learning methods in the context of Scientific ML. With those metrics we identify tasks which are challenging for recent ML methods and propose these tasks as future challenges for the community. The code is available at https://github.com/pdebench/PDEBench.
format Preprint
id arxiv_https___arxiv_org_abs_2210_07182
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle PDEBENCH: An Extensive Benchmark for Scientific Machine Learning
Takamoto, Makoto
Praditia, Timothy
Leiteritz, Raphael
MacKinlay, Dan
Alesiani, Francesco
Pflüger, Dirk
Niepert, Mathias
Machine Learning
Computer Vision and Pattern Recognition
Fluid Dynamics
Geophysics
Machine learning-based modeling of physical systems has experienced increased interest in recent years. Despite some impressive progress, there is still a lack of benchmarks for Scientific ML that are easy to use but still challenging and representative of a wide range of problems. We introduce PDEBench, a benchmark suite of time-dependent simulation tasks based on Partial Differential Equations (PDEs). PDEBench comprises both code and data to benchmark the performance of novel machine learning models against both classical numerical simulations and machine learning baselines. Our proposed set of benchmark problems contribute the following unique features: (1) A much wider range of PDEs compared to existing benchmarks, ranging from relatively common examples to more realistic and difficult problems; (2) much larger ready-to-use datasets compared to prior work, comprising multiple simulation runs across a larger number of initial and boundary conditions and PDE parameters; (3) more extensible source codes with user-friendly APIs for data generation and baseline results with popular machine learning models (FNO, U-Net, PINN, Gradient-Based Inverse Method). PDEBench allows researchers to extend the benchmark freely for their own purposes using a standardized API and to compare the performance of new models to existing baseline methods. We also propose new evaluation metrics with the aim to provide a more holistic understanding of learning methods in the context of Scientific ML. With those metrics we identify tasks which are challenging for recent ML methods and propose these tasks as future challenges for the community. The code is available at https://github.com/pdebench/PDEBench.
title PDEBENCH: An Extensive Benchmark for Scientific Machine Learning
topic Machine Learning
Computer Vision and Pattern Recognition
Fluid Dynamics
Geophysics
url https://arxiv.org/abs/2210.07182