Scalable Belief Propagation via Relaxed Scheduling
Vitaly Aksenov, Dan Alistarh, Janne H. Korhonen
Abstract
The ability to leverage large-scale hardware parallelism has been one of the key enablers of the accelerated recent progress in machine learning. Consequently, there has been considerable effort invested into developing efficient parallel variants of classic machine learning algorithms. However, despite the wealth of knowledge on parallelization, some classic machine learning algorithms often prove hard to parallelize efficiently while maintaining convergence. In this paper, we focus on efficient parallel algorithms for the key machine learning task of inference on graphical models, in particular on the fundamental belief propagation algorithm. We address the challenge of efficiently parallelizing this classic paradigm by showing how to leverage scalable relaxed schedulers in this context. We present an extensive empirical study, showing that our approach outperforms previous parallel belief propagation implementations both in terms of scalability and in terms of wall-clock convergence time, on a range of practical applications. This update rule is applied to messages until the values of messages have converged to a stable solution, at which point the algorithm is said to have terminated.
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 c5a33ba3-6fb6-4af0-b78d-632544df9ff7Cited by top-tier papers2
- Multi-queues can be state-of-the-art priority schedulersAnastasiia Postnikova, Nikita Koval, Giorgi Nadiradze, Dan AlistarhPPoPP 2022 · 18 citations
- A scalable architecture for reprioritizing ordered parallelismGilead Posluns, Yan Zhu, Guowei Zhang, Mark C. JeffreyISCA 2022 · 8 citations
Related papers
- Learning General Latent-Variable Graphical Models with Predictive Belief PropagationBorui Wang, Geoffrey J. GordonAAAI 2020 · 1 citation
- Belief Propagation Converges to Gaussian Distributions in Sparsely-Connected Factor GraphsTom Yates, Yuzhou Cheng, Ignacio Alzugaray, Danyal Akarca et al.ICML 2026
- A parallel framework for constraint-based bayesian network learning via markov blanket discoveryAnkit Srivastava, Sriram P. Chockalingam, Srinivas AluruSC 2020 · 11 citations
- Load and MLP-Aware Thread Orchestration for Recommendation Systems Inference on CPUsRishabh Jain, Teyuh Chou, Onur Kayiran, John Kalamatianos et al.ASPLOS 2025 · 4 citations
- Learning in Deep Factor Graphs with Gaussian Belief PropagationSeth Nabarro, Mark van der Wilk, Andrew J. DavisonICML 2024 · 1 citation
