Skip to yearly menu bar Skip to main content


Poster
in
Workshop: NeurIPS 2024 Workshop: Machine Learning and the Physical Sciences

Emulation and Assessment of Gradient-Based Samplers in Cosmology

Arrykrishna Mootoovaloo · David Alonso · Jaime Ruiz-Zapatero · Carlos Garcia-Garcia


Abstract: We assess gradient-based samplers like the No-U-Turn Sampler (NUTS) compared to traditional Metropolis-Hastings algorithms in tomographic 3×2 point analyses using DES Year 1 data and a simulated LSST-like survey. These studies involve 20 and 32 nuisance parameters, respectively. We implement a differentiable forward model using JAXCOSMO and derive parameter constraints using NUTS and Metropolis-Hastings algorithms. NUTS shows a relative efficiency gain of O(10) in terms of effective samples per likelihood evaluation but only a factor of 2 in terms of computational time due to the higher gradient computation cost. We validate these results with analytical multivariate distributions, concluding that NUTS can be beneficial for sampling high-dimensional parameter spaces in Cosmology, though the efficiency gain is modest for moderate dimensions (O(50)).

Chat is not available.