Towards end-to-end likelihood-free inference with convolutional neural networks

Br J Math Stat Psychol. 2020 Feb;73(1):23-43. doi: 10.1111/bmsp.12159. Epub 2019 Feb 22.

Abstract

Complex simulator-based models with non-standard sampling distributions require sophisticated design choices for reliable approximate parameter inference. We introduce a fast, end-to-end approach for approximate Bayesian computation (ABC) based on fully convolutional neural networks. The method enables users of ABC to derive simultaneously the posterior mean and variance of multidimensional posterior distributions directly from raw simulated data. Once trained on simulated data, the convolutional neural network is able to map real data samples of variable size to the first two posterior moments of the relevant parameter's distributions. Thus, in contrast to other machine learning approaches to ABC, our approach allows us to generate reusable models that can be applied by different researchers employing the same model. We verify the utility of our method on two common statistical models (i.e., a multivariate normal distribution and a multiple regression scenario), for which the posterior parameter distributions can be derived analytically. We then apply our method to recover the parameters of the leaky competing accumulator (LCA) model and we reference our results to the current state-of-the-art technique, which is the probability density estimation (PDA). Results show that our method exhibits a lower approximation error compared with other machine learning approaches to ABC. It also performs similarly to PDA in recovering the parameters of the LCA model.

Keywords: approximate Bayesian computation; convolutional network; leaky competing accumulator; likelihood-free inference; machine-learning.

MeSH terms

  • Algorithms*
  • Bayes Theorem*
  • Computer Simulation
  • Humans
  • Likelihood Functions
  • Machine Learning
  • Neural Networks, Computer*
  • Regression Analysis