Yi Mao
Current Organizations / Institutes
The organizations/institutes from the scholar's latest article
Previous Organizations / Institutes
The organizations / institutes from the scholar's other articles in the past ten years
Yi Mao has a total of 198 co-authors. Here are the top co-authors by number of shared publications:
- Paulo Montero-Camacho
Department of Mathematics and Theory Peng Cheng Laboratory 518066 Shenzhen Guangdong China
10 shared publications - Xiaosheng Zhao
The Johns Hopkins University
9 shared publications - Meng Zhou
National Astronomical Observatories Chinese Academy of Sciences 100101 Beijing People's Republic of China
5 shared publications Benjamin D. Wandelt
4 shared publications- Benjamin D. Wandelt
Johns Hopkins University
4 shared publications - Jianrong Tan
Department of Physics and Astronomy, University of Pennsylvania, Philadelphia, PA, USA
4 shared publications
Latest Publications
- Open Access
- 5 August 2026
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 1007 (1) , 49
Foreground Subtraction with a Tensor-based Oriented Singular Value Decomposition Method for H I Experiments
We introduce a native tensor-based framework for foreground mitigation in 21 cm intensity mapping (IM), utilizing the Oriented Singular Value Decomposition (O-SVD) algorithm. While 21 cm IM is a powerful probe of the large-scale structure of the Universe, its efficacy is severely limited by astrophysical foregrounds that are orders of magnitude brighter than the cosmological signal. Traditional mitigation strategies often necessitate flattening multidimensional data cubes into 2D matrices, a process that potentially compromises the intrinsic spatial–spectral correlations by treating distinct spatial pixels as independent samples. By treating multifrequency sky maps and angular power spectra as third-order tensors, the O-SVD method performs decomposition directly on the multilinear manifold, preserving the underlying physical topology and leveraging the distinct coherence properties of astrophysical foregrounds across different dimensions. We demonstrate the performance and versatility of the O-SVD framework through its application to high-fidelity simulations from the Square Kilometre Array Science Data Challenge 3a and real-world observational data from the Tianlai Cylinder Pathfinder Array. Our results indicate that O-SVD provides a robust and universal approach for foreground subtraction, achieving high-fidelity signal recovery while offering superior performance compared to conventional matrix-based SVD methods.
- 5 August 2026
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 1007 (1) , 49
Open AccessForeground Subtraction with a Tensor-based Oriented Singular Value Decomposition Method for H I Experiments
We introduce a native tensor-based framework for foreground mitigation in 21 cm intensity mapping (IM), utilizing the Oriented Singular Value Decomposition (O-SVD) algorithm. While 21 cm IM is a powerful probe of the large-scale structure of the Universe, its efficacy is severely limited by astrophysical foregrounds that are orders of magnitude brighter than the cosmological signal. Traditional mitigation strategies often necessitate flattening multidimensional data cubes into 2D matrices, a process that potentially compromises the intrinsic spatial–spectral correlations by treating distinct spatial pixels as independent samples. By treating multifrequency sky maps and angular power spectra as third-order tensors, the O-SVD method performs decomposition directly on the multilinear manifold, preserving the underlying physical topology and leveraging the distinct coherence properties of astrophysical foregrounds across different dimensions. We demonstrate the performance and versatility of the O-SVD framework through its application to high-fidelity simulations from the Square Kilometre Array Science Data Challenge 3a and real-world observational data from the Tianlai Cylinder Pathfinder Array. Our results indicate that O-SVD provides a robust and universal approach for foreground subtraction, achieving high-fidelity signal recovery while offering superior performance compared to conventional matrix-based SVD methods.
- Open Access
- 19 February 2026
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 998 (2) , 329
How to Evaluate the Sufficiency and Complementarity of Summary Statistics for Cosmic Fields: An Information-theoretic Perspective
The advent of increasingly advanced surveys and cosmic tracers has motivated the development of new inference techniques and novel approaches to extracting information from cosmic fields. A central challenge in this endeavor is to quantify the information content carried by these summary statistics in cosmic fields. In particular, how should we assess which statistics are more informative than others and assess the exact degree of complementarity of the information from each statistic? Here, we introduce mutual information (MI), which provides, from an information-theoretic perspective, a natural framework for assessing the sufficiency and complementarity of summary statistics in cosmological data. We demonstrate how MI can be applied to typical inference tasks to make information-theoretic evaluations, using two representative examples: the cosmic microwave background map, from which the power spectrum (PS) extracts almost all information as is expected for a Gaussian random field, and the 21 cm brightness temperature map, from which the scattering transform extracts the most non-Gaussian information but which is complementary to the PS and bispectrum. Our results suggest that MI offers a robust theoretical foundation for evaluating and improving summaries, thereby enabling a deeper understanding of cosmic fields from an information-theoretic perspective.
- 19 February 2026
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 998 (2) , 329
Open AccessHow to Evaluate the Sufficiency and Complementarity of Summary Statistics for Cosmic Fields: An Information-theoretic Perspective
The advent of increasingly advanced surveys and cosmic tracers has motivated the development of new inference techniques and novel approaches to extracting information from cosmic fields. A central challenge in this endeavor is to quantify the information content carried by these summary statistics in cosmic fields. In particular, how should we assess which statistics are more informative than others and assess the exact degree of complementarity of the information from each statistic? Here, we introduce mutual information (MI), which provides, from an information-theoretic perspective, a natural framework for assessing the sufficiency and complementarity of summary statistics in cosmological data. We demonstrate how MI can be applied to typical inference tasks to make information-theoretic evaluations, using two representative examples: the cosmic microwave background map, from which the power spectrum (PS) extracts almost all information as is expected for a Gaussian random field, and the 21 cm brightness temperature map, from which the scattering transform extracts the most non-Gaussian information but which is complementary to the PS and bispectrum. Our results suggest that MI offers a robust theoretical foundation for evaluating and improving summaries, thereby enabling a deeper understanding of cosmic fields from an information-theoretic perspective.
- 19 November 2025
- Published by American Physical Society (APS) in Physical Review D
- Vol. 112 (10)
Constraining fuzzy dark matter with the 21-cm power spectrum from cosmic dawn and reionization
- 19 November 2025
- Published by American Physical Society (APS) in Physical Review D
- Vol. 112 (10)
Constraining fuzzy dark matter with the 21-cm power spectrum from cosmic dawn and reionization
- Open Access
- 14 November 2025
- Published by Oxford University Press (OUP) in Monthly Notices of the Royal Astronomical Society
- Vol. 545 (4)
Tantalizing evidence of reionization relics in the eBOSS DR16 Ly α forest correlations: a preference for early reionization
Cosmic reionization of H i leaves enduring relics in the post-reionization intergalactic medium, potentially influencing the Lyman $\alpha$ (Ly $\alpha$) forest down to redshifts as low as $z \approx 2$, which is the so-called ‘memory of reionization’ effect. Here, we re-analyse the baryonic acoustic oscillation (BAO) measurements from Ly $\alpha$ absorption and quasar correlations using data from the extended Baryonic Oscillation Spectroscopic Survey (eBOSS) Data Release 16 (DR16), incorporating for the first time the memory of reionization in the Ly $\alpha$ forest. Three distinct scenarios of reionization timeline are considered in our analyses. We find that the recovered BAO parameters ($\alpha _\parallel$, $\alpha _\perp$) remain consistent with the original eBOSS DR16 analysis. However, models incorporating reionization relics provide a better fit to the data, with a tantalizing preference for early reionization, consistent with recent findings from the James Webb Space Telescope. Furthermore, the inclusion of reionization relics significantly impacts the non-BAO parameters. For instance, we report deviations of up to $3\sigma$ in the Ly $\alpha$ redshift-space distortion parameter and $\sim 7\sigma$ in the linear Ly $\alpha$ bias for the late reionization scenario. Our findings suggest that the eBOSS Ly $\alpha$ data are more accurately described by models that incorporate a broad-band enhancement to the Ly $\alpha$ forest power spectrum, highlighting the importance of accounting for reionization relics in cosmological analyses.- 14 November 2025
- Published by Oxford University Press (OUP) in Monthly Notices of the Royal Astronomical Society
- Vol. 545 (4)
Open AccessTantalizing evidence of reionization relics in the eBOSS DR16 Ly α forest correlations: a preference for early reionization
Cosmic reionization of H i leaves enduring relics in the post-reionization intergalactic medium, potentially influencing the Lyman $\alpha$ (Ly $\alpha$) forest down to redshifts as low as $z \approx 2$, which is the so-called ‘memory of reionization’ effect. Here, we re-analyse the baryonic acoustic oscillation (BAO) measurements from Ly $\alpha$ absorption and quasar correlations using data from the extended Baryonic Oscillation Spectroscopic Survey (eBOSS) Data Release 16 (DR16), incorporating for the first time the memory of reionization in the Ly $\alpha$ forest. Three distinct scenarios of reionization timeline are considered in our analyses. We find that the recovered BAO parameters ($\alpha _\parallel$, $\alpha _\perp$) remain consistent with the original eBOSS DR16 analysis. However, models incorporating reionization relics provide a better fit to the data, with a tantalizing preference for early reionization, consistent with recent findings from the James Webb Space Telescope. Furthermore, the inclusion of reionization relics significantly impacts the non-BAO parameters. For instance, we report deviations of up to $3\sigma$ in the Ly $\alpha$ redshift-space distortion parameter and $\sim 7\sigma$ in the linear Ly $\alpha$ bias for the late reionization scenario. Our findings suggest that the eBOSS Ly $\alpha$ data are more accurately described by models that incorporate a broad-band enhancement to the Ly $\alpha$ forest power spectrum, highlighting the importance of accounting for reionization relics in cosmological analyses. - Open Access
How to evaluate the sufficiency and complementarity of summary statistics for cosmic fields: an information-theoretic perspective
The advent of increasingly advanced surveys and cosmic tracers has motivated the development of new inference techniques and novel approaches to extracting information from cosmic fields. A central challenge in this endeavor is to quantify the information content carried by these summary statistics in cosmic fields. In particular, how should we assess which statistics are more informative than others and assess the exact degree of complementarity of the information from each statistic? Here, we introduce mutual information (MI) that provides, from an information-theoretic perspective, a natural framework for assessing the sufficiency and complementarity of summary statistics in cosmological data. We demonstrate how MI can be applied to typical inference tasks to make information-theoretic evaluations, using two representative examples: the cosmic microwave background map, from which the power spectrum extracts almost all information as is expected for a Gaussian random field, and the 21~cm brightness temperature map, from which the scattering transform extracts the most non-Gaussian information but is complementary to power spectrum and bispectrum. Our results suggest that MI offers a robust theoretical foundation for evaluating and improving summaries, thereby enabling a deeper understanding of cosmic fields from an information-theoretic perspective.
Open AccessHow to evaluate the sufficiency and complementarity of summary statistics for cosmic fields: an information-theoretic perspective
The advent of increasingly advanced surveys and cosmic tracers has motivated the development of new inference techniques and novel approaches to extracting information from cosmic fields. A central challenge in this endeavor is to quantify the information content carried by these summary statistics in cosmic fields. In particular, how should we assess which statistics are more informative than others and assess the exact degree of complementarity of the information from each statistic? Here, we introduce mutual information (MI) that provides, from an information-theoretic perspective, a natural framework for assessing the sufficiency and complementarity of summary statistics in cosmological data. We demonstrate how MI can be applied to typical inference tasks to make information-theoretic evaluations, using two representative examples: the cosmic microwave background map, from which the power spectrum extracts almost all information as is expected for a Gaussian random field, and the 21~cm brightness temperature map, from which the scattering transform extracts the most non-Gaussian information but is complementary to power spectrum and bispectrum. Our results suggest that MI offers a robust theoretical foundation for evaluating and improving summaries, thereby enabling a deeper understanding of cosmic fields from an information-theoretic perspective.
- Open Access
- 13 October 2025
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 992 (2) , 173
Multifidelity Emulator for Large-scale 21 cm Lightcone Images: A Few-shot Transfer Learning Approach with Generative Adversarial Network
Emulators using machine learning techniques have emerged to efficiently generate mock data matching the large survey volume for upcoming experiments, as an alternative approach to large-scale numerical simulations. However, high-fidelity emulators have become computationally expensive as the simulation volume grows to hundreds of megaparsecs. Here, we present a multifidelity emulation of large-scale 21 cm lightcone images from the epoch of reionization, which is realized by applying the few-shot transfer learning to training generative adversarial networks (GAN) from small-scale to large-scale simulations. Specifically, a GAN emulator is first trained with a huge number of small-scale simulations, and then transfer-learned with only a limited number of large-scale simulations, to emulate large-scale 21 cm lightcone images. We test the precision of our transfer-learned GAN emulator in terms of representative statistics including global 21 cm brightness temperature history, 2D power spectrum, and scattering transform coefficients. We demonstrate that the lightcone images generated by the transfer-learned GAN emulator can reach the percentage level precision in most cases on small scales, and the error on large scales only increases mildly to the level of a few tens of percent. Nevertheless, our multifidelity emulation technique saves a significant portion of computational resources that are mostly consumed for generating training samples for GAN. On estimate, the computational resource by training GAN completely with large-scale simulations would be 1 to 2 orders of magnitude larger than using our multifidelity technique. This implies that our technique allows for emulating high-fidelity, traditionally computationally prohibitive, images in an economic manner.
- 13 October 2025
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 992 (2) , 173
Open AccessMultifidelity Emulator for Large-scale 21 cm Lightcone Images: A Few-shot Transfer Learning Approach with Generative Adversarial Network
Emulators using machine learning techniques have emerged to efficiently generate mock data matching the large survey volume for upcoming experiments, as an alternative approach to large-scale numerical simulations. However, high-fidelity emulators have become computationally expensive as the simulation volume grows to hundreds of megaparsecs. Here, we present a multifidelity emulation of large-scale 21 cm lightcone images from the epoch of reionization, which is realized by applying the few-shot transfer learning to training generative adversarial networks (GAN) from small-scale to large-scale simulations. Specifically, a GAN emulator is first trained with a huge number of small-scale simulations, and then transfer-learned with only a limited number of large-scale simulations, to emulate large-scale 21 cm lightcone images. We test the precision of our transfer-learned GAN emulator in terms of representative statistics including global 21 cm brightness temperature history, 2D power spectrum, and scattering transform coefficients. We demonstrate that the lightcone images generated by the transfer-learned GAN emulator can reach the percentage level precision in most cases on small scales, and the error on large scales only increases mildly to the level of a few tens of percent. Nevertheless, our multifidelity emulation technique saves a significant portion of computational resources that are mostly consumed for generating training samples for GAN. On estimate, the computational resource by training GAN completely with large-scale simulations would be 1 to 2 orders of magnitude larger than using our multifidelity technique. This implies that our technique allows for emulating high-fidelity, traditionally computationally prohibitive, images in an economic manner.
- Open Access
- 29 September 2025
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 991 (2) , 195
Prospects for kSZ2–21 cm2 Cross Correlations during Reionization
The 21 cm line and the patchy kinetic Sunyaev–Zel’dovich (kSZ) effect are promising and complementary probes of the Epoch of Reionization. A challenge for cross correlating these two signals is that foreground avoidance or removal algorithms applied to the 21 cm data inevitably sacrifice Fourier modes with long wavelengths along the line of sight (i.e., low-k ∥ modes), yet only these same modes contribute to the kSZ signal. Here we show that a suitable kSZ2 × 21 cm2 cross-correlation statistic nevertheless remains nonvanishing, even after filtering out the corrupted low-k ∥Fourier modes from the 21 cm data. We simulate the kSZ2 × 21 cm2 cross-correlation signal across reionization-era redshifts and find distinctive redshift evolution. This signal peaks early in the reionization history, when the volume-averaged fraction is around 0.1 ≲ x HII ≲ 0.2, after which it changes sign and reaches a minimum near reionization’s midpoint (x HII ∼ 0.5), while the signal gradually vanishes as reionization completes. These trends appear generic across three simulated models which differ in their reionization histories. We forecast the detectability of the kSZ2 × 21 cm2 cross-power spectrum for the HERA and SKA1-Low 21 cm experiments in combination with current and next-generation cosmic microwave background (CMB) surveys including the Simons Observatory, CMB-S4, and CMB-HD. We find that a high-significance detection (signal-to-noise ratio ≳ 5σ) is possible with SKA1-Low and CMB-S4.
- 29 September 2025
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 991 (2) , 195
Open AccessProspects for kSZ2–21 cm2 Cross Correlations during Reionization
The 21 cm line and the patchy kinetic Sunyaev–Zel’dovich (kSZ) effect are promising and complementary probes of the Epoch of Reionization. A challenge for cross correlating these two signals is that foreground avoidance or removal algorithms applied to the 21 cm data inevitably sacrifice Fourier modes with long wavelengths along the line of sight (i.e., low-k ∥ modes), yet only these same modes contribute to the kSZ signal. Here we show that a suitable kSZ2 × 21 cm2 cross-correlation statistic nevertheless remains nonvanishing, even after filtering out the corrupted low-k ∥Fourier modes from the 21 cm data. We simulate the kSZ2 × 21 cm2 cross-correlation signal across reionization-era redshifts and find distinctive redshift evolution. This signal peaks early in the reionization history, when the volume-averaged fraction is around 0.1 ≲ x HII ≲ 0.2, after which it changes sign and reaches a minimum near reionization’s midpoint (x HII ∼ 0.5), while the signal gradually vanishes as reionization completes. These trends appear generic across three simulated models which differ in their reionization histories. We forecast the detectability of the kSZ2 × 21 cm2 cross-power spectrum for the HERA and SKA1-Low 21 cm experiments in combination with current and next-generation cosmic microwave background (CMB) surveys including the Simons Observatory, CMB-S4, and CMB-HD. We find that a high-significance detection (signal-to-noise ratio ≳ 5σ) is possible with SKA1-Low and CMB-S4.
- 3 September 2025
- Published by IOP Publishing in Research in Astronomy and Astrophysics
- Vol. 25 (10) , 105010
A Halo-based Empirical Model for the Atomic Hydrogen Content
Atomic hydrogen (H i) gas, mostly residing in dark matter halos after cosmic reionization, is the fuel for star formation. Its relation to the properties of the host halo is the key to understand the cosmic H i distribution and to mock H i large scale structure accurately and conveniently for the large area H i intensity mapping surveys in the future. In this work, we propose a flexible empirical model of H i–halo relation. In this model, while the H i mass depends primarily on the mass of host halo, there is also secondary dependence on other halo properties. We apply our model to the observation data of the Arecibo Fast Legacy ALFA (ALFALFA) survey, and find it can successfully fit the cosmic H i abundance (ΩH I ), average H i–halo mass relation 〈M H I ∣M h〉, and the H i clustering. The best fit of the ALFALFA data rejects the model with no secondary halo dependence of H i and the model with secondary dependence on the halo spin parameter (λ), and shows a strong dependence on halo formation time (a 1/2) and halo concentration (c vir). In an attempt to explain these findings from the perspective of hydrodynamical simulations, the IllustrisTNG simulation confirms the dependence of H i mass on secondary halo parameters. However, the IllustrisTNG results show a strong dependence on λ and a weak dependence on c vir and a 1/2, and also predict a much larger value of H i clustering on large scales than observations. This interesting difference between simulation and observation calls for improvements in understanding the H i–halo relation from both theoretical and observational sides.
- 3 September 2025
- Published by IOP Publishing in Research in Astronomy and Astrophysics
- Vol. 25 (10) , 105010
A Halo-based Empirical Model for the Atomic Hydrogen Content
Atomic hydrogen (H i) gas, mostly residing in dark matter halos after cosmic reionization, is the fuel for star formation. Its relation to the properties of the host halo is the key to understand the cosmic H i distribution and to mock H i large scale structure accurately and conveniently for the large area H i intensity mapping surveys in the future. In this work, we propose a flexible empirical model of H i–halo relation. In this model, while the H i mass depends primarily on the mass of host halo, there is also secondary dependence on other halo properties. We apply our model to the observation data of the Arecibo Fast Legacy ALFA (ALFALFA) survey, and find it can successfully fit the cosmic H i abundance (ΩH I ), average H i–halo mass relation 〈M H I ∣M h〉, and the H i clustering. The best fit of the ALFALFA data rejects the model with no secondary halo dependence of H i and the model with secondary dependence on the halo spin parameter (λ), and shows a strong dependence on halo formation time (a 1/2) and halo concentration (c vir). In an attempt to explain these findings from the perspective of hydrodynamical simulations, the IllustrisTNG simulation confirms the dependence of H i mass on secondary halo parameters. However, the IllustrisTNG results show a strong dependence on λ and a weak dependence on c vir and a 1/2, and also predict a much larger value of H i clustering on large scales than observations. This interesting difference between simulation and observation calls for improvements in understanding the H i–halo relation from both theoretical and observational sides.
- Open Access
- 24 June 2025
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 987 (1) , 18
Modeling Foreground Spatial Variations in 21 cm Gaussian Process Component Separation
The 21 cm line from neutral hydrogen captures rich cosmological information. However, the 21 cm signals are highly contaminated by foreground emission, making accurate component separation vital for cosmological analyses. Gaussian processes (GPs) have been utilized for component separation in 21 cm data analyses, exploiting the distinct spectral behavior of the cosmological and foreground signals, which are modeled through the GP covariance kernel. Previous approaches have employed a global GP kernel along all lines of sight (LoS). In this work, we study Bayesian approaches that allow for spatial variations in foreground kernel parameters, testing them against simulated H i intensity mapping observations, which measures the total H i intensity in large voxels. We consider a no-pooling (NP) model, treating each LoS independently by fitting for separate covariance kernels, and a hierarchical Gaussian process (HGP) model, allowing for variation in kernel parameters between different LoS, regularized through a global hyperprior. We find that including spatial variations in the GP kernel parameters results in a significant improvement in cosmological signal recovery, achieving up to a 30% reduction in the standard deviation of the residual distribution. Allowing for spatial variations also improves the recovery of the H i power spectra and wavelet scattering transform coefficients. While the NP model achieves superior recovery as measured by the residual distribution, it is computationally demanding, faces convergence challenges, and is prone to overfitting. The HGP model strikes a balance between accuracy and robustness. Further improvements to the HGP model will require more physically-motivated modeling of foreground spatial variations. Our code can be found in the following repository: https://github.com/dkn16/H21F.
- 24 June 2025
- Published by American Astronomical Society in The Astrophysical Journal
- Vol. 987 (1) , 18
Open AccessModeling Foreground Spatial Variations in 21 cm Gaussian Process Component Separation
The 21 cm line from neutral hydrogen captures rich cosmological information. However, the 21 cm signals are highly contaminated by foreground emission, making accurate component separation vital for cosmological analyses. Gaussian processes (GPs) have been utilized for component separation in 21 cm data analyses, exploiting the distinct spectral behavior of the cosmological and foreground signals, which are modeled through the GP covariance kernel. Previous approaches have employed a global GP kernel along all lines of sight (LoS). In this work, we study Bayesian approaches that allow for spatial variations in foreground kernel parameters, testing them against simulated H i intensity mapping observations, which measures the total H i intensity in large voxels. We consider a no-pooling (NP) model, treating each LoS independently by fitting for separate covariance kernels, and a hierarchical Gaussian process (HGP) model, allowing for variation in kernel parameters between different LoS, regularized through a global hyperprior. We find that including spatial variations in the GP kernel parameters results in a significant improvement in cosmological signal recovery, achieving up to a 30% reduction in the standard deviation of the residual distribution. Allowing for spatial variations also improves the recovery of the H i power spectra and wavelet scattering transform coefficients. While the NP model achieves superior recovery as measured by the residual distribution, it is computationally demanding, faces convergence challenges, and is prone to overfitting. The HGP model strikes a balance between accuracy and robustness. Further improvements to the HGP model will require more physically-motivated modeling of foreground spatial variations. Our code can be found in the following repository: https://github.com/dkn16/H21F.
- Open Access
- 1 May 2025
- Published by American Astronomical Society in The Astrophysical Journal Supplement Series
- Vol. 278 (1) , 25
synax: A Differentiable and GPU-accelerated Synchrotron Simulation Package
We introduce synax (https://github.com/dkn16/Synax), a novel library for automatically differentiable simulation of Galactic synchrotron emission. Built on the JAX framework, synax leverages JAX’s capabilities, including batch acceleration, just-in-time compilation, and hardware-specific optimizations (CPU, GPU, TPU). Crucially, synax uses JAX’s automatic differentiation (AD) mechanism, enabling precise computation of analytical derivatives with respect to any model parameters. This facilitates powerful inference algorithms, such as Hamiltonian Monte Carlo (HMC) and gradient-based optimization, which enables inference over models that would otherwise be computationally prohibitive. In its initial release, synax supports synchrotron intensity and polarization calculations down to GHz frequencies, alongside several models of the Galactic magnetic field (GMF), cosmic-ray spectra, and thermal electron density fields. When running synax on the CPU we obtain identical performance to hammurabi, a state-of-the-art synchrotron simulation package, while on the GPU synax brings a 20-fold enhancement in efficiency. We further demonstrate the potential of AD in enabling full posterior inference using gradient-based inference algorithms. Using synax with HMC to perform inference over a four-parameter test model, we attain a twofold improvement compared to standard random walk Metropolis–Hastings (RWMH). When applied to a more complex 16-parameter model, HMC is still able to obtain accurate posterior expectations, while RWMH fails to converge. We also showcase the application of synax to optimizing the GMF based on the Haslam 408 MHz map, achieving residuals with a standard deviation below 1 K.
- 1 May 2025
- Published by American Astronomical Society in The Astrophysical Journal Supplement Series
- Vol. 278 (1) , 25
Open Accesssynax: A Differentiable and GPU-accelerated Synchrotron Simulation Package
We introduce synax (https://github.com/dkn16/Synax), a novel library for automatically differentiable simulation of Galactic synchrotron emission. Built on the JAX framework, synax leverages JAX’s capabilities, including batch acceleration, just-in-time compilation, and hardware-specific optimizations (CPU, GPU, TPU). Crucially, synax uses JAX’s automatic differentiation (AD) mechanism, enabling precise computation of analytical derivatives with respect to any model parameters. This facilitates powerful inference algorithms, such as Hamiltonian Monte Carlo (HMC) and gradient-based optimization, which enables inference over models that would otherwise be computationally prohibitive. In its initial release, synax supports synchrotron intensity and polarization calculations down to GHz frequencies, alongside several models of the Galactic magnetic field (GMF), cosmic-ray spectra, and thermal electron density fields. When running synax on the CPU we obtain identical performance to hammurabi, a state-of-the-art synchrotron simulation package, while on the GPU synax brings a 20-fold enhancement in efficiency. We further demonstrate the potential of AD in enabling full posterior inference using gradient-based inference algorithms. Using synax with HMC to perform inference over a four-parameter test model, we attain a twofold improvement compared to standard random walk Metropolis–Hastings (RWMH). When applied to a more complex 16-parameter model, HMC is still able to obtain accurate posterior expectations, while RWMH fails to converge. We also showcase the application of synax to optimizing the GMF based on the Haslam 408 MHz map, achieving residuals with a standard deviation below 1 K.