Three-dimensional Conditional Diffusion Models for Cosmological 21 cm Lightcone Emulation
astro-ph.IM, astro-ph.CO, cs.LG
Submitted: 2026-05-27
Updated: 2026-08-31
Comments: Accepted for publication in The Astrophysical Journal. Revised version incorporating referee comments
Code: https://github.com/Xsmos/ml21cm
License: http://creativecommons.org/licenses/by/4.0/
The gist: We investigate conditional diffusion modeling for three-dimensional 21 cm lightcone emulation, focusing on cubes with a sky-plane size of 64 times64 and a line-of-sight depth up to 1024 cells.
Terminology
Abstract
We investigate conditional diffusion modeling for three-dimensional 21 cm lightcone emulation, focusing on cubes with a sky-plane size of 64 times64 and a line-of-sight depth up to 1024 cells. Relative to earlier 2D studies, the 3D setting is substantially harder because memory limits enforce very small micro-batches while the underlying voxel distribution is highly skewed and long tailed. We perform controlled comparisons across preprocessing choices, dynamic-range compression settings, architecture depth, and training duration using 25, 600 training lightcones and validation ensembles at fixed parameter points. For validation, each reference parameter point contains 800 21cmFAST realizations with independent initial conditions, and we use 800 samples per model and per reference set for the reported ensemble comparisons. We evaluate generated lightcones with complementary diagnostics in both image and summary-statistic spaces: brightness-temperature slices, the global signal, the power spectrum, and reduced scattering coefficients. Across the tested configurations, preprocessing is the dominant factor governing stable training and the resulting physical fidelity. Among the configurations explored here, Yeo-Johnson preprocessing combined with moderate amplitude compression gives the most consistently favorable trade-off, with the strongest quantitative support coming from rankings based on the standard-deviation-normalized mean absolute error (MAE std) of the global signal and qualitatively compatible behavior in the complementary diagnostics. At the same time, visually plausible 3D samples still retain measurable biases in two-point and higher-order statistics. We therefore view the present work as a simulation-level baseline for three-dimensional 21 cm emulation and for future studies that incorporate more realistic observational effects.
Sources
- Emulating cosmological multifields with generative adversarial networks
- Multi-fidelity Emulator for Cosmological Large Scale 21 cm Lightcone Images: a Few-shot Transfer Learning Approach with GAN
- Generative Adversarial Networks
Related papers
- A signal dedispersion algorithm for imaging-based transient searches
- AVICA: A fully automated CASA pipeline for large volume VLBI data calibration
- Spectral Map Making with SPHEREx
- Long-Integration Magnetar Burst Observatory (LIMBO): Instrument Summary and Early FRB Rate Constraints
- Towards independent event horizon imaging of the supermassive black holes in M87 and the Milky Way
- A PINK update: Improvements to the CELEBI fast radio burst data reduction and analysis pipeline