Arbitrary Marginal Neural Ratio Estimation for Simulation-based Inference

Abstract

In many areas of science, complex phenomena are modeled by stochastic parametric simulators, often featuring high-dimensional parameter spaces and intractable likelihoods. In this context, performing Bayesian inference can be challenging. In this work, we present a novel method that enables amortized inference over arbitrary subsets of the parameters, without resorting to numerical integration, which makes interpretation of the posterior more convenient. Our method is efficient and can be implemented with arbitrary neural network architectures. We demonstrate the applicability of the method on parameter inference of binary black hole systems from gravitational waves observations.

Fourth Workshop on Machine Learning and the Physical Sciences (NeurIPS 2021)

References in corpus (4)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• Robust parameter estimation for compact binaries with ground-based gravitational-wave observations using the LALInference software library
• Automatic Posterior Transformation for Likelihood-Free Inference
• Learning about an exponential amount of conditional distributions