Direct Amortized Likelihood Ratio Estimation

Authors

  • Adam D. Cobb SRI International
  • Brian Matejek SRI International
  • Daniel Elenius SRI International
  • Anirban Roy SRI International
  • Susmit Jha SRI International

DOI:

https://doi.org/10.1609/aaai.v38i18.30018

Keywords:

RU: Probabilistic Inference, ML: Bayesian Learning

Abstract

We introduce a new amortized likelihood ratio estimator for likelihood-free simulation-based inference (SBI). Our estimator is simple to train and estimates the likelihood ratio using a single forward pass of the neural estimator. Our approach directly computes the likelihood ratio between two competing parameter sets which is different from the previous approach of comparing two neural network output values. We refer to our model as the direct neural ratio estimator (DNRE). As part of introducing the DNRE, we derive a corresponding Monte Carlo estimate of the posterior. We benchmark our new ratio estimator and compare to previous ratio estimators in the literature. We show that our new ratio estimator often outperforms these previous approaches. As a further contribution, we introduce a new derivative estimator for likelihood ratio estimators that enables us to compare likelihood-free Hamiltonian Monte Carlo (HMC) with random-walk Metropolis-Hastings (MH). We show that HMC is equally competitive, which has not been previously shown. Finally, we include a novel real-world application of SBI by using our neural ratio estimator to design a quadcopter. Code is available at https://github.com/SRI-CSL/dnre.

Published

2024-03-24

How to Cite

Cobb, A. D., Matejek, B., Elenius, D., Roy, A., & Jha, S. (2024). Direct Amortized Likelihood Ratio Estimation. Proceedings of the AAAI Conference on Artificial Intelligence, 38(18), 20362-20369. https://doi.org/10.1609/aaai.v38i18.30018

Issue

Section

AAAI Technical Track on Reasoning under Uncertainty