Divide, Conquer, and Combine: a New Inference Strategy for Probabilistic Programs with Stochastic Support
Yuan Zhou, Hongseok Yang, Yee Whye Teh, Tom Rainforth
Abstract
Universal probabilistic programming systems (PPSs) provide a powerful framework for specifying rich probabilistic models. They further attempt to automate the process of drawing inferences from these models, but doing this successfully is severely hampered by the wide range of non--standard models they can express. As a result, although one can specify complex models in a universal PPS, the provided inference engines often fall far short of what is required. In particular, we show that they produce surprisingly unsatisfactory performance for models where the support varies between executions, often doing no better than importance sampling from the prior. To address this, we introduce a new inference framework: Divide, Conquer, and Combine, which remains efficient for such models, and show how it can be implemented as an automated and generic PPS inference engine. We empirically demonstrate substantial performance improvements over existing approaches on three examples.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 9731a41e-ef2c-471f-b4f2-a0b7ad58481eCited by top-tier papers6
- Tensor Program Optimization with Probabilistic ProgramsJunru Shao, Xiyou Zhou, Siyuan Feng, Bohan Hou et al.NeurIPS 2022 · 85 citations
- Inference of Probabilistic Programs with Moment-Matching Gaussian MixturesFrancesca Randone, Luca Bortolussi, Emilio Incerto, Mirco TribastonePOPL 2024 · 12 citations
- Nonparametric Hamiltonian Monte CarloCarol Mak, Fabian Zaiser, Luke OngICML 2021 · 7 citations
- Symbolic parallel adaptive importance sampling for probabilistic program analysisYicheng Luo, Antonio Filieri, Yuan ZhouFSE 2021 · 3 citations
- Rethinking Variational Inference for Probabilistic Programs with Stochastic SupportTim Reichelt, Luke Ong, Thomas RainforthNeurIPS 2022 · 3 citations
Related papers
- Probabilistic Programming with Programmable Divide-Conquer-Combine Inference on Modern HardwareMarkus Böck, Jürgen CitoOOPSLA 2026 · 1 citation
- Automatic Reparameterisation of Probabilistic ProgramsMaria I. Gorinova, Dave Moore, Matthew D. HoffmanICML 2020 · 33 citations
- Nonparametric Involutive Markov Chain Monte CarloCarol Mak, Fabian Zaiser, Luke OngICML 2022 · 2 citations
- Fast Inference for Probabilistic Graphical ModelsJiantong Jiang, Zeyi Wen, Atif Bin Mansoor, Ajmal MianUSENIX ATC 2024 · 4 citations
- Multi-Language Probabilistic ProgrammingSam Stites, John M. Li, Steven HoltzenOOPSLA 2025 · 2 citations
