Analytic auto-differentiable $\Lambda$CDM cosmography

Abstract

I present general analytic expressions for distance calculations (comoving distance, time coordinate, and absorption distance) in the standard $\Lambda$CDM cosmology, allowing for the presence of radiation and for non-zero curvature. The solutions utilise the symmetric Carlson basis of elliptic integrals, which can be evaluated with fast numerical algorithms that allow trivial parallelisation on GPUs and automatic differentiation without the need for additional special functions. I introduce a PyTorch-based implementation in the phytorch.cosmology package and briefly examine its accuracy and speed in comparison with numerical integration and other known expressions (for special cases). Finally, I demonstrate an application to high-dimensional Bayesian analysis that utilises automatic differentiation through the distance calculations to efficiently derive posteriors for cosmological parameters from up to $10^6$ mock type Ia supernovae using variational inference.

arXiv commentmatches published version; 12 pages, 5 figures + appendix; phytorch available at https://github.com/kosiokarchev/phytorch

References in corpus (42)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• LSST: from Science Drivers to Reference Design and Anticipated Data Products
• The frontier of simulation-based inference
• The Pantheon+ Analysis: The Full Dataset and Light-Curve Release
• Mining gold from implicit models to improve likelihood-free inference
• Numerical computation of real or complex elliptic integrals
• Automatic physical inference with information maximising neural networks
• UNITY: Confronting Supernova Cosmology's Statistical and Systematic Uncertainties in a Unified Bayesian Framework
• A Hierarchical Bayesian SED Model for Type Ia Supernovae in the Optical to Near-Infrared
• The Type Ia Supernova Color-Magnitude Relation and Host Galaxy Dust: A Simple Hierarchical Bayesian Model
• BAHAMAS: new SNIa analysis reveals inconsistencies with standard cosmology
• FlowPM: Distributed TensorFlow Implementation of the FastPM Cosmological N-body Solver
• Field Level Neural Network Emulator for Cosmological N-body Simulations
• Complete solutions to the metric of spherically collapsing dust in an expanding spacetime with a cosmological constant
• Differentiable Strong Lensing: Uniting Gravity and Neural Nets through Differentiable Probabilistic Programming
• GIGA-Lens: Fast Bayesian Inference for Strong Gravitational Lens Modeling
• Using wavelets to capture deviations from smoothness in galaxy-scale strong lenses
• SICRET: Supernova Ia Cosmology with truncated marginal neural Ratio EsTimation
• Auto-Differentiable Spectrum Model for High-Dispersion Characterization of Exoplanets and Brown Dwarfs
• Steve: A hierarchical Bayesian model for Supernova Cosmology
• A Differentiable Model of the Assembly of Individual and Populations of Dark Matter Halos
• zBEAMS: A unified solution for supernova cosmology with redshift uncertainties
• Application of Bayesian graphs to SN Ia data analysis and compression
• DSPS: Differentiable Stellar Population Synthesis
• Differentiable Cosmological Simulation with Adjoint Method
• Phase Retrieval and Design with Automatic Differentiation
• Efficient Gravitational Wave Template Bank Generation with Differentiable Waveforms
• An Analytic Expression for the Growth Function in a Flat Universe with a Cosmological Constant
• A differentiable N-body code for transit timing and dynamical modeling. I. Algorithm and derivatives
• Analytical expressions and numerical evaluation of the luminosity distance in a flat cosmology
• Kernel Phase and Coronagraphy with Automatic Differentiation
• Numerical Strategies of Computing the Luminosity Distance
• pmwd: A Differentiable Cosmological Particle-Mesh $N$-body Library
• Differentiable Predictions for Large Scale Structure with SHAMNet
• Strong-lensing source reconstruction with variationally optimised Gaussian processes
• The Age-Redshift Relation for Standard Cosmology
• A curious relation between the flat cosmological model and the elliptic integral of the first kind
• Analytic Light Curve for Mutual Transits of Two Bodies Across a Limb-darkened Star
• CosmicRIM : Reconstructing Early Universe by Combining Differentiable Simulations with Recurrent Inference Machines
• An analytical solution in the complex plane for the luminosity distance in flat cosmology
• A new analytical solution for the distance modulus in flat cosmology
• On the Perturbative Picture and the Chang-Refsdal Lens Approximation for Planetary Microlensing

Cited by in corpus (2)

• Massive neutrinos and cosmic composition
• SIDE-real: Supernova Ia Dust Extinction with truncated marginal neural ratio estimation applied to real data