Accelerating Radiative Transfer in Planetary Atmospheres by Orders of Magnitude with Neural Network Emulators
- 1California Institute of Technology
- 2University of California Santa Cruz
- 3NASA Ames Research Center
Abstract
Radiative transfer calculations are essential for modeling planetary atmospheres. However, standard methods are computationally demanding and impose accuracy-speed trade-offs. High computational costs force numerical simplifications in large models (e.g., general circulation models) that degrade the accuracy of the simulation. Radiative transfer calculations are an ideal candidate for machine learning emulation; fundamentally, radiative transfer is a well-defined physical mapping from a static atmospheric profile to the resulting fluxes, and high-fidelity training data can be created from first-principles calculations. We developed a radiative transfer emulator using an encoder-only transformer neural network architecture, trained on 1D profiles representative of solar-composition hot Jupiter atmospheres. Our emulator reproduced bolometric two-stream layer fluxes with mean test set errors of ∼1% compared to the traditional method and achieved speedups of more than 100× in batched GPU inference. In matched comparisons against a long short-term memory model (LSTM) baseline, a transformer was consistently the more accurate and more robust architecture when training data were limited, while the LSTM achieved better accuracy at full scale. Emulating radiative transfer with machine learning opens up the possibility for faster and more accurate routines within planetary atmospheric models.
Related
- Accelerating Chemical Kinetics for Exoplanet Atmospheres using Neural Networks
- Energy deposition in planetary and exoplanetary atmospheres induced by cosmic rays
- The 3D Cosmic Shoreline for Nurturing Planetary Atmospheres
- Planet formation in chemically diverse and evolving discs II. Chemical fingerprints in planetary atmospheres
- Benchmarking Machine Learning Emulators of Stellar Evolution for Precision Asteroseismology
BibTeX
@article{malsky2026accelerating,
title = {{Accelerating Radiative Transfer in Planetary Atmospheres by Orders of Magnitude with Neural Network Emulators}},
author = {Isaac Malsky and Tiffany Kataria and Natasha E. Batalha and Matthew Graham},
journal = {The Astrophysical Journal},
year = {2026},
month = nov,
doi = {10.3847/1538-4357/aeb019},
url = {https://iopscience.iop.org/article/10.3847/1538-4357/aeb019},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Accelerating Radiative Transfer in Planetary Atmospheres by Orders of Magnitude with Neural Network Emulators
AU - Isaac Malsky
AU - Tiffany Kataria
AU - Natasha E. Batalha
AU - Matthew Graham
PY - 2026
DA - 2026/11/05
JO - The Astrophysical Journal
DO - 10.3847/1538-4357/aeb019
UR - https://iopscience.iop.org/article/10.3847/1538-4357/aeb019
AB - Radiative transfer calculations are essential for modeling planetary atmospheres. However, standard methods are computationally demanding and impose accuracy-speed trade-offs. High computational costs force numerical simplifications in large models (e.g., general circulation models) that degrade the accuracy of the simulation. Radiative transfer calculations are an ideal candidate for machine learning emulation; fundamentally, radiative transfer is a well-defined physical mapping from a static atmospheric profile to the resulting fluxes, and high-fidelity training data can be created from first-principles calculations. We developed a radiative transfer emulator using an encoder-only transformer neural network architecture, trained on 1D profiles representative of solar-composition hot Jupiter atmospheres. Our emulator reproduced bolometric two-stream layer fluxes with mean test set errors of ∼1% compared to the traditional method and achieved speedups of more than 100× in batched GPU inference. In matched comparisons against a long short-term memory model (LSTM) baseline, a transformer was consistently the more accurate and more robust architecture when training data were limited, while the LSTM achieved better accuracy at full scale. Emulating radiative transfer with machine learning opens up the possibility for faster and more accurate routines within planetary atmospheric models.
SN - 1538-4357
SN - 0004-637X
ER -