English

CODES: Benchmarking Coupled ODE Surrogates

Machine Learning 2024-11-21 v2 Instrumentation and Methods for Astrophysics Computational Physics

Abstract

We introduce CODES, a benchmark for comprehensive evaluation of surrogate architectures for coupled ODE systems. Besides standard metrics like mean squared error (MSE) and inference time, CODES provides insights into surrogate behaviour across multiple dimensions like interpolation, extrapolation, sparse data, uncertainty quantification and gradient correlation. The benchmark emphasizes usability through features such as integrated parallel training, a web-based configuration generator, and pre-implemented baseline models and datasets. Extensive documentation ensures sustainability and provides the foundation for collaborative improvement. By offering a fair and multi-faceted comparison, CODES helps researchers select the most suitable surrogate for their specific dataset and application while deepening our understanding of surrogate learning behaviour.

Cite

@article{arxiv.2410.20886,
  title  = {CODES: Benchmarking Coupled ODE Surrogates},
  author = {Robin Janssen and Immanuel Sulzer and Tobias Buck},
  journal= {arXiv preprint arXiv:2410.20886},
  year   = {2024}
}

Comments

13 pages, 10 figures, accepted for the Machine Learning and the Physical Sciences workshop at NeurIPS 2024, source code available on GitHub at https://github.com/robin-janssen/CODES-Benchmark

R2 v1 2026-06-28T19:37:49.854Z