Probabilistic Inference with Algebraic Constraints: Theoretical Limits and Practical Approximations
Zhe Zeng, Paolo Morettin, Fanqi Yan, Antonio Vergari, Guy Van den Broeck
Abstract
Weighted model integration (WMI) is a framework to perform advanced probabilistic inference in hybrid domains, i.e., on distributions over mixed continuous-discrete random variables and in the presence of complex logical and arithmetic constraints. In this work, we advance the WMI framework on both the theoretical and algorithmic side. First, we trace the boundaries of tractability for WMI inference in terms of two key properties of a WMI problem's dependency structure: sparsity and diameter. We prove that exact inference is only efficient if that structure is tree-shaped with logarithmic diameter. While this result deepens our theoretical understanding of WMI it hinders the practical applicability of exact WMI solvers to large problems. To overcome this, we propose the first approximate WMI solver that does not resort to sampling, but performs exact inference on an approximate model. Our solution iteratively performs message passing in a relaxed problem structure to recover lost dependencies. As our experiments show, it scales to problems that are out of the reach of exact WMI solvers while delivering accurate approximations. * Authors contributed equally. This research was performed while F.Y. and P.M. were visiting UCLA.
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.
Cited by top-tier papers4
- A Unified Approach to Count-Based Weakly Supervised LearningVinay Shukla, Zhe Zeng, Kareem Ahmed, Guy Van den BroeckNeurIPS 2023 · 14 citations
- Collapsed Inference for Bayesian Deep LearningZhe Zeng, Guy Van den BroeckNeurIPS 2023 · 10 citations
- SIMPLE: A Gradient Estimator for k-Subset SamplingKareem Ahmed, Zhe Zeng, Mathias Niepert, Guy Van den BroeckICLR 2023 · 2 citations
- The Theory and Practice of MAP Inference over Non-Convex ConstraintsLeander Kurscheidt, Gabriele Masina, Roberto Sebastiani, Antonio VergariICML 2026 · 1 citation
Builds on1
Related papers
- Learning Weighted Model Integration DistributionsPaolo Morettin, Samuel Kolb, Stefano Teso, Andrea PasseriniAAAI 2020 · 9 citations
- Hybrid Memoised Wake-Sleep: Approximate Inference at the Discrete-Continuous InterfaceTuan Anh Le, Katherine M. Collins, Luke Hewitt, Kevin Ellis et al.ICLR 2022 · 5 citations
- λPSI: exact inference for higher-order probabilistic programsTimon Gehr, Samuel Steffen, Martin T. VechevPLDI 2020 · 29 citations
- Continuous Mixtures of Tractable Probabilistic ModelsAlvaro H. C. Correia, Gennaro Gala, Erik Quaeghebeur, Cassio P. de Campos et al.AAAI 2023 · 26 citations
- Scaling exact inference for discrete probabilistic programsSteven Holtzen, Guy Van den Broeck, Todd D. MillsteinOOPSLA 2020 · 85 citations
