arXiv Machine Learning

ThousandWorlds: A benchmark for climate emulation of potentially habitable exoplanets

arXiv:2606. 18338v1 Announce Type: new Abstract: The search for life beyond Earth will depend on detecting faint signatures in the atmospheres of potentially habitable exoplanets.

Hugging Face Trending Papers
Jun 17

Optimal scenario design for climate emulation

As deep learning for physical systems continues to grow in popularity, efforts to improve generalizability have primarily focused on designing architectures that embed physical constraints. However, for machine-learning surrogate climate models (emulators), we show that the low structural diversity in existing scenarios commonly used to generate training data places a ceiling on predictive skill.

arXiv Machine Learning
Jun 18

Optimal scenario design for climate emulation

arXiv:2606. 19302v1 Announce Type: cross Abstract: As deep learning for physical systems continues to grow in popularity, efforts to improve generalizability have primarily focused on designing architectures that embed physical constraints.

By Christopher B. Womack, Shahine Bouabid, Andrei Sokolov, Popat Salunke, Glenn Flierl, Sebastian D. Eastham, Noelle E. Selin
arXiv Machine Learning
Sep 14

4D Parallelism Unlocks Exascale Bayesian Neural Networks for High-Fidelity Atmospheric Modeling

BEAST is the first Bayesian Swin Transformer for atmospheric forecasting at 0.25° global resolution, capable of quantifying both aleatoric and epistemic uncertainty. The authors introduce a 4D‑parallelization scheme with domain‑tensor‑parallelism and a novel uncertainty parallel method, allowing a 2.4‑billion‑parameter model to reach 3.96 EFLOP/s on 20,480 NVIDIA GH200 GPUs. A 700‑million‑parameter BEAST trained on 40 years of data achieves predictive skill comparable to state‑of‑the‑art probabilistic AI models, predicts extreme events with high accuracy, and generates large ensembles 3–4× faster than the current best AI model.

By Deifilia Kieckhefen, Juan Pedro Guti\'errez Hermosillo Muriedas, Lars Helge Heyen, Mathis Bode, Iida Hakulinen, Andreas Herten, Chelsea Maria John, Thorsten Kurth, Anni Moisala, Asena Karolin \"Ozdemir, Kaleb Phipps, Oskar Taubert, Arvid Weyrauch, Markus G\"otz, Charlotte Debus
arXiv Machine Learning
Sep 2

Accelerating Chemical Kinetics for Exoplanet Atmospheres using Neural Networks

The paper introduces a neural‑network based local‑box chemical kinetics solver for exoplanet atmospheres, employing a residual flow‑map architecture. It achieves microsecond‑scale inference with percent‑level accuracy across a wide range of temperatures, pressures, time steps, and compositional variations, outperforming other machine‑learning models and handling the extreme stiffness of atmospheric chemistry. The surrogate model offers a flexible, efficient alternative to classical solvers for state‑to‑state flow‑map problems in numerical simulations.

By Isaac Malsky, Xi Zhang, Tiffany Kataria, Matthew Graham, Ziyu Huang, Boris Bonev, Shang-Min Tsai, Elspeth K. H. Lee
arXiv Machine Learning
Jul 14

How Can Machine Learning Emulators Best Support Climate Science?

arXiv:2603. 22320v3 Announce Type: replace Abstract: For decades, physics-based climate models have been used to provide insights for climate decision-making.

By Luca Schmidt, Nina Effenberger, Vitus Benson, Philine L. Bommer, Robert Brunstein, Mikel N. Legasa, Maxim Samarin, Maybritt Schillinger
arXiv Machine Learning
Aug 28

SimCast-S2S: An Efficient Generative Model for Subseasonal Precipitation Forecasting via Transfer Learning from Climate Simulations

SimCast‑S2S is a generative latent‑diffusion model designed for probabilistic subseasonal‑to‑seasonal precipitation forecasting. It tackles three key challenges: it uses a diffusion pipeline to capture uncertainty, operates in a compact latent space to enable efficient large‑ensemble generation, and leverages transfer learning with low‑rank adaptation to train on limited reanalysis data after pretraining on climate simulations. The model outperforms deep‑learning baselines and competes with, or surpasses, operational systems such as the ECMWF‑S2S baseline without requiring extensive post‑processing.

By Hiep V. Dang, Antonios Mamalakis