Embarrassingly parallel MCMC using deep invertible transformations

Research output: Chapter in Book/Report/Conference proceedingConference contributionScientificpeer-review

Abstract

While MCMC methods have become a main work-horse for Bayesian inference, scaling them to large distributed datasets is still a challenge. Embarrassingly parallel MCMC strategies take a divide-and-conquer stance to achieve this by writing the target posterior as a product of subposteriors, running MCMC for each of them in parallel and subsequently combining the results. The challenge then lies in devising efficient aggregation strategies. Current strategies tradeoff between approximation quality, and costs of communication and computation. In this work, we introduce a novel method that addresses these issues simultaneously. Our key insight is to introduce a deep invertible transformation to approximate each of the subposteriors. These approximations can be made accurate even for complex distributions and serve as intermediate representations, keeping the total communication cost limited. Moreover, they enable us to sample from the product of the subposteriors using an efficient and stable importance sampling scheme. We demonstrate that the approach outperforms available state-of-the-art methods in a range of challenging scenarios, including high-dimensional and heterogeneous subposteriors.

Original languageEnglish
Title of host publication35th Conference on Uncertainty in Artificial Intelligence (UAI 2019)
PublisherAUAI Press
ISBN (Print)9781510891562
Publication statusPublished - 2019
MoE publication typeA4 Article in a conference publication
EventConference on Uncertainty in Artificial Intelligence - Tel Aviv, Israel
Duration: 22 Jul 201925 Jul 2019
Conference number: 35

Conference

ConferenceConference on Uncertainty in Artificial Intelligence
Abbreviated titleUAI
CountryIsrael
CityTel Aviv
Period22/07/201925/07/2019

Fingerprint Dive into the research topics of 'Embarrassingly parallel MCMC using deep invertible transformations'. Together they form a unique fingerprint.

  • Equipment

    Science-IT

    Mikko Hakala (Manager)

    School of Science

    Facility/equipment: Facility

  • Cite this

    Mesquita, D., Blomstedt, P., & Kaski, S. (2019). Embarrassingly parallel MCMC using deep invertible transformations. In 35th Conference on Uncertainty in Artificial Intelligence (UAI 2019) AUAI Press.