Skip to yearly menu bar Skip to main content


Poster
in
Workshop: Machine Learning for Astrophysics

Neural Posterior Estimation with Differentiable Simulator

Justine Zeghal · Francois Lanusse · Alexandre Boucaud · Benjamin Remy · Eric Aubourg


Abstract:

Simulation-Based Inference (SBI) is a promising Bayesian inference framework that alleviates the need for analytic likelihoods to estimate parameter distributions. Recent advances using neural density estimators in SBI algorithms have demonstrated the ability to achieve high-fidelity posteriors, at the expense of a large number of simulations ; which makes their application potentially very time-consuming when using complex physical simulations. In this work we focus on boosting the posterior density estimation using the gradients of the simulator. We present a new method to perform Neural Posterior Estimation (NPE) with a differentiable simulator and demonstrate the accelerated convergence of the NPE, up to a speedup factor of 2 below 100 simulations on classical SBI benchmark problems.

Chat is not available.