Accelerating Redshift-Conditioned Galaxy Image Synthesis with One-step Generative Modeling
On the map: Diffusion models → Inverse problems and image restoration
Abstract (arXiv)
Understanding galaxy morphology evolution across cosmic time requires models that can generate realistic galaxy populations conditioned on redshift. In this work, we study efficient redshift-conditioned generative modeling for astrophysical image synthesis using diffusion models and pixel-MeanFlow. We first review the connections between score-based diffusion models, Flow Matching, one-step generative models, and modern diffusion samplers. We then evaluate DDPM, DDIM, DEIS-AB2, DPM++2M, and one-step pixel-MeanFlow on the GalaxiesML-64 dataset using morphology-based metrics, including ellipticity, semi-major axis, S'ersic index, and isophotal area. Our results show a clear accuracy-efficiency trade-off: standard DDPM sampling achieves the best distributional fidelity but requires high computational cost, while second-order samplers substantially improve efficiency over DDIM. Pixel-MeanFlow enables single-step generation and achieves competitive performance on several morphology statistics, though it remains weaker than many-step DDPM for fine-grained structure. Our results demonstrate that one-step generative models can recover key galaxy morphology statistics at orders-of-magnitude lower computational cost, opening a path toward efficient conditional simulators for large cosmological surveys and simulation-based scientific inference.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 43 of its 62 references that are in the map. First posted on arXiv 17 May 2026.
Builds on
43 of 62 references.
Closest ancestors · 4
Same direction, newest first
- 2025FlowLensing: Simulating Gravitational Lensing with Flow Matching9 Oct
- 2024Understanding Galaxy Morphology Evolution Through Cosmic Time via Redshift Conditioned Diffusion Models27 Nov
- 2022Can denoising diffusion probabilistic models generate realistic astrophysical fields?22 Nov
- 2021Realistic galaxy image simulation via score-based generative models2 Nov
Other references · 39
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2022Generative Adversarial Networkscited by 5,627
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2023Consistency Modelscited by 1,822
27 more
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
- 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,043
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 938
- 2025Mean Flows for One-step Generative Modelingcited by 561
- 2022Fast Sampling of Diffusion Models with Exponential Integratorcited by 532
- 2024One Step Diffusion via Shortcut Modelscited by 391
- 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Modelscited by 283
- 2022Pangu-Weather: A 3D High-Resolution Model for Fast and Accurate Global Weather Forecastcited by 166
- 2018i-RevNet: Deep Invertible Networkscited by 137
- 2025Inductive Moment Matchingcited by 113
- 2025Improved Mean Flows: On the Challenges of Fastforward Generative Modelscited by 110
- 2023DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal Forecastingcited by 109
- 2024Rolling Diffusion Modelscited by 104
- 2023Flow Matching for Scalable Simulation-Based Inferencecited by 101
- 2023PreDiff: Precipitation Nowcasting with Latent Diffusion Modelscited by 98
- 2026One-step Latent-free Image Generation with Pixel Mean Flowscited by 53
- 2025AlphaFlow: Understanding and Improving MeanFlow Modelscited by 41
- 2025Latent Bayesian Optimization via Autoregressive Normalizing Flowscited by 19
- 2017CosmoGAN: creating high-fidelity weak lensing convergence maps using Generative Adversarial Networkscited by 17
- 2025Elucidated Rolling Diffusion Models for Probabilistic Forecasting of Complex Dynamicscited by 13
- 2024How DREAMS are made: Emulating Satellite Galaxy and Subhalo Populations with Diffusion Models and Point Cloudscited by 5
- 2025On the Design of One-step Diffusion via Shortcutting Flow Pathscited by 4
- 2026Uni-Flow: a unified autoregressive-diffusion model for complex multiscale flowscited by 4
- 2026MENO: MeanFlow-Enhanced Neural Operators for Dynamical Systemscited by 3
- 2020Emulation of cosmological mass maps with conditional generative adversarial networkscited by 2
- 2024Invertible ResNets for Inverse Imaging Problems: Competitive Performance with Provable Regularization Propertiescited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Using Galaxy Evolution as Source of Physics-Based Ground Truth for Generative Models
- 2025Investigation on deep learning-based galaxy image translation models
- 2023Unified framework for diffusion generative models in SO(3): applications in computer vision and astrophysics
- 2023A point cloud approach to generative modeling for galaxy surveys at the field level
- 2025Conditional Diffusion-Flow models for generating 3D cosmic density fields: applications to f(R) cosmologies
- 2025Category-based Galaxy Image Generation via Diffusion Models
- 2025Can AI Dream of Unseen Galaxies? Conditional Diffusion Model for Galaxy Morphology Augmentation
- 2026Cosmo3DFlow: Wavelet Flow Matching for Spatial-to-Spectral Compression in Reconstructing the Early Universe