Belief propagation

From HandWiki
Short description: Algorithm for statistical inference on graphical models

Belief propagation, also known as sum–product message passing, is a message-passing algorithm for performing inference on graphical models, such as Bayesian networks and Markov random fields. It calculates the marginal distribution for each unobserved node (or variable), conditional on any observed nodes (or variables). Belief propagation is commonly used in artificial intelligence and information theory, and has demonstrated empirical success in numerous applications, including low-density parity-check codes, turbo codes, free energy approximation, and satisfiability.[1]

The algorithm was first proposed by Judea Pearl in 1982,[2] who formulated it as an exact inference algorithm on trees, later extended to polytrees.[3] While the algorithm is not exact on general graphs, it has been shown to be a useful approximate algorithm.[4]

Motivation

Given a finite set of discrete random variables [math]\displaystyle{ X_1, \ldots, X_n }[/math] with joint probability mass function [math]\displaystyle{ p }[/math], a common task is to compute the marginal distributions of the [math]\displaystyle{ X_i }[/math]. The marginal of a single [math]\displaystyle{ X_i }[/math] is defined to be

[math]\displaystyle{ p_{X_i}(x_i) = \sum_{\mathbf{x}': x'_i = x_i} p(\mathbf{x}') }[/math]

where [math]\displaystyle{ \mathbf x' = (x'_1, \ldots, x'_n) }[/math] is a vector of possible values for the [math]\displaystyle{ X_i }[/math], and the notation [math]\displaystyle{ \mathbf x' : x'_i = x_i }[/math] means that the sum is taken over those [math]\displaystyle{ \mathbf x' }[/math] whose [math]\displaystyle{ i }[/math]th coordinate is equal to [math]\displaystyle{ x_i }[/math].

Computing marginal distributions using this formula quickly becomes computationally prohibitive as the number of variables grows. For example, given 100 binary variables [math]\displaystyle{ X_1, \ldots, X_{100} }[/math], computing a single marginal [math]\displaystyle{ X_i }[/math] using [math]\displaystyle{ p }[/math] and the above formula would involve summing over [math]\displaystyle{ 2^{99} \approx 6.34 \times 10^{29} }[/math] possible values for [math]\displaystyle{ \mathbf x' }[/math]. If it is known that the probability mass function [math]\displaystyle{ p }[/math] factors in a convenient way, belief propagation allows the marginals to be computed much more efficiently.

Description of the sum-product algorithm

Variants of the belief propagation algorithm exist for several types of graphical models (Bayesian networks and Markov random fields[5] in particular). We describe here the variant that operates on a factor graph. A factor graph is a bipartite graph containing nodes corresponding to variables [math]\displaystyle{ V }[/math] and factors [math]\displaystyle{ F }[/math], with edges between variables and the factors in which they appear. We can write the joint mass function:

[math]\displaystyle{ p(\mathbf{x}) = \prod_{a \in F} f_a (\mathbf{x}_a) }[/math]

where [math]\displaystyle{ \mathbf x_a }[/math] is the vector of neighboring variable nodes to the factor node [math]\displaystyle{ a }[/math]. Any Bayesian network or Markov random field can be represented as a factor graph by using a factor for each node with its parents or a factor for each node with its neighborhood respectively.[6]

The algorithm works by passing real valued functions called messages along the edges between the hidden nodes. More precisely, if [math]\displaystyle{ v }[/math] is a variable node and [math]\displaystyle{ a }[/math] is a factor node connected to [math]\displaystyle{ v }[/math] in the factor graph, then the messages [math]\displaystyle{ \mu_{v \to a} }[/math] from [math]\displaystyle{ v }[/math] to [math]\displaystyle{ a }[/math] and the messages [math]\displaystyle{ \mu_{a \to v} }[/math] from [math]\displaystyle{ a }[/math] to [math]\displaystyle{ v }[/math] are real-valued functions [math]\displaystyle{ \mu_{v \to a}, \mu_{a \to v} : \operatorname{Dom}(v) \to \mathbb R }[/math], whose domain is the set of values that can be taken by the random variable associated with [math]\displaystyle{ v }[/math], denoted [math]\displaystyle{ \operatorname{Dom}(v) }[/math]. These messages contain the "influence" that one variable exerts on another. The messages are computed differently depending on whether the node receiving the message is a variable node or a factor node. Keeping the same notation:

  • A message [math]\displaystyle{ \mu_{v \to a}: \operatorname{Dom}(v) \to \mathbb R }[/math] from a variable node [math]\displaystyle{ v }[/math] to a factor node [math]\displaystyle{ a }[/math] is defined by [math]\displaystyle{ \mu_{v \to a} (x_v) = \prod_{a^* \in N(v)\setminus\{a\} } \mu_{a^* \to v} (x_v) }[/math]for [math]\displaystyle{ x_v \in \operatorname{Dom}(v) }[/math], where [math]\displaystyle{ N(v) }[/math] is the set of neighboring factor nodes of [math]\displaystyle{ v }[/math]. If [math]\displaystyle{ N(v)\setminus\{a\} }[/math] is empty then [math]\displaystyle{ \mu_{v \to a}(x_v) }[/math] is set to the uniform distribution over [math]\displaystyle{ \operatorname{Dom}(v) }[/math].
  • A message [math]\displaystyle{ \mu_{a \to v}: \operatorname{Dom}(v) \to \mathbb R }[/math] from a factor node [math]\displaystyle{ a }[/math] to a variable node [math]\displaystyle{ v }[/math] is defined to be the product of the factor with messages from all other nodes, marginalized over all variables except the one associated with [math]\displaystyle{ v }[/math],[math]\displaystyle{ \mu_{a \to v} (x_v) = \sum_{\mathbf{x}'_a: x'_v = x_v } \left( f_a (\mathbf{x}'_a) \prod_{v^* \in N(a) \setminus \{v\}} \mu_{v^* \to a} (x'_{v^*}) \right) }[/math]for [math]\displaystyle{ x_v \in \operatorname{Dom}(v) }[/math], where [math]\displaystyle{ N(a) }[/math] is the set of neighboring (variable) nodes to [math]\displaystyle{ a }[/math]. If [math]\displaystyle{ N(a) \setminus \{v\} }[/math] is empty, then [math]\displaystyle{ \mu_{a \to v} (x_v) = f_a(x_v) }[/math], since in this case [math]\displaystyle{ x_v = x_a }[/math].

As shown by the previous formula: the complete marginalization is reduced to a sum of products of simpler terms than the ones appearing in the full joint distribution. This is the reason that belief propagation is sometimes called sum-product message passing, or the sum-product algorithm.

In a typical run, each message will be updated iteratively from the previous value of the neighboring messages. Different scheduling can be used for updating the messages. In the case where the graphical model is a tree, an optimal scheduling converges after computing each message exactly once (see next sub-section). When the factor graph has cycles, such an optimal scheduling does not exist, and a typical choice is to update all messages simultaneously at each iteration.

Upon convergence (if convergence happened), the estimated marginal distribution of each node is proportional to the product of all messages from adjoining factors (missing the normalization constant):

[math]\displaystyle{ p_{X_v} (x_v) \propto \prod_{a \in N(v)} \mu_{a \to v} (x_v). }[/math]

Likewise, the estimated joint marginal distribution of the set of variables belonging to one factor is proportional to the product of the factor and the messages from the variables:

[math]\displaystyle{ p_{X_a} (\mathbf{x}_a) \propto f_a(\mathbf{x}_a) \prod_{v \in N(a)} \mu_{v \to a} (x_v). }[/math]

In the case where the factor graph is acyclic (i.e. is a tree or a forest), these estimated marginal actually converge to the true marginals in a finite number of iterations. This can be shown by mathematical induction.

Exact algorithm for trees

In the case when the factor graph is a tree, the belief propagation algorithm will compute the exact marginals. Furthermore, with proper scheduling of the message updates, it will terminate after two full passes through the tree. This optimal scheduling can be described as follows:

Before starting, the graph is oriented by designating one node as the root; any non-root node which is connected to only one other node is called a leaf.

In the first step, messages are passed inwards: starting at the leaves, each node passes a message along the (unique) edge towards the root node. The tree structure guarantees that it is possible to obtain messages from all other adjoining nodes before passing the message on. This continues until the root has obtained messages from all of its adjoining nodes.

The second step involves passing the messages back out: starting at the root, messages are passed in the reverse direction. The algorithm is completed when all leaves have received their messages.

Approximate algorithm for general graphs

Although it was originally designed for acyclic graphical models, the Belief Propagation algorithm can be used in general graphs. The algorithm is then sometimes called loopy belief propagation, because graphs typically contain cycles, or loops. The initialization and scheduling of message updates must be adjusted slightly (compared with the previously described schedule for acyclic graphs) because graphs might not contain any leaves. Instead, one initializes all variable messages to 1 and uses the same message definitions above, updating all messages at every iteration (although messages coming from known leaves or tree-structured subgraphs may no longer need updating after sufficient iterations). It is easy to show that in a tree, the message definitions of this modified procedure will converge to the set of message definitions given above within a number of iterations equal to the diameter of the tree.

The precise conditions under which loopy belief propagation will converge are still not well understood; it is known that on graphs containing a single loop it converges in most cases, but the probabilities obtained might be incorrect.[7] Several sufficient (but not necessary) conditions for convergence of loopy belief propagation to a unique fixed point exist.[8] There exist graphs which will fail to converge, or which will oscillate between multiple states over repeated iterations. Techniques like EXIT charts can provide an approximate visualization of the progress of belief propagation and an approximate test for convergence.

There are other approximate methods for marginalization including variational methods and Monte Carlo methods.

One method of exact marginalization in general graphs is called the junction tree algorithm, which is simply belief propagation on a modified graph guaranteed to be a tree. The basic premise is to eliminate cycles by clustering them into single nodes.

Related algorithm and complexity issues

A similar algorithm is commonly referred to as the Viterbi algorithm, but also known as a special case of the max-product or min-sum algorithm, which solves the related problem of maximization, or most probable explanation. Instead of attempting to solve the marginal, the goal here is to find the values [math]\displaystyle{ \mathbf{x} }[/math] that maximizes the global function (i.e. most probable values in a probabilistic setting), and it can be defined using the arg max:

[math]\displaystyle{ \operatorname*{\arg\max}_{\mathbf{x}} g(\mathbf{x}). }[/math]

An algorithm that solves this problem is nearly identical to belief propagation, with the sums replaced by maxima in the definitions.[9]

It is worth noting that inference problems like marginalization and maximization are NP-hard to solve exactly and approximately (at least for relative error) in a graphical model. More precisely, the marginalization problem defined above is #P-complete and maximization is NP-complete.

The memory usage of belief propagation can be reduced through the use of the Island algorithm (at a small cost in time complexity).

Relation to free energy

The sum-product algorithm is related to the calculation of free energy in thermodynamics. Let Z be the partition function. A probability distribution

[math]\displaystyle{ P(\mathbf{X}) = \frac{1}{Z} \prod_{f_j} f_j(x_j) }[/math]

(as per the factor graph representation) can be viewed as a measure of the internal energy present in a system, computed as

[math]\displaystyle{ E(\mathbf{X}) = -\log \prod_{f_j} f_j(x_j). }[/math]

The free energy of the system is then

[math]\displaystyle{ F = U - H = \sum_{\mathbf{X}} P(\mathbf{X}) E(\mathbf{X}) + \sum_{\mathbf{X}} P(\mathbf{X}) \log P(\mathbf{X}). }[/math]

It can then be shown that the points of convergence of the sum-product algorithm represent the points where the free energy in such a system is minimized. Similarly, it can be shown that a fixed point of the iterative belief propagation algorithm in graphs with cycles is a stationary point of a free energy approximation.[10]

Generalized belief propagation (GBP)

Belief propagation algorithms are normally presented as message update equations on a factor graph, involving messages between variable nodes and their neighboring factor nodes and vice versa. Considering messages between regions in a graph is one way of generalizing the belief propagation algorithm.[10] There are several ways of defining the set of regions in a graph that can exchange messages. One method uses ideas introduced by Kikuchi in the physics literature,[11][12][13] and is known as Kikuchi's cluster variation method.[14]

Improvements in the performance of belief propagation algorithms are also achievable by breaking the replicas symmetry in the distributions of the fields (messages). This generalization leads to a new kind of algorithm called survey propagation (SP), which have proved to be very efficient in NP-complete problems like satisfiability[1] and graph coloring.

The cluster variational method and the survey propagation algorithms are two different improvements to belief propagation. The name generalized survey propagation (GSP) is waiting to be assigned to the algorithm that merges both generalizations.

Gaussian belief propagation (GaBP)

Gaussian belief propagation is a variant of the belief propagation algorithm when the underlying distributions are Gaussian. The first work analyzing this special model was the seminal work of Weiss and Freeman.[15]

The GaBP algorithm solves the following marginalization problem:

[math]\displaystyle{ P(x_i) = \frac{1}{Z} \int_{j \ne i} \exp(-\tfrac 1 2 x^TAx + b^Tx)\,dx_j }[/math]

where Z is a normalization constant, A is a symmetric positive definite matrix (inverse covariance matrix a.k.a. precision matrix) and b is the shift vector.

Equivalently, it can be shown that using the Gaussian model, the solution of the marginalization problem is equivalent to the MAP assignment problem:

[math]\displaystyle{ \underset{x}{\operatorname{argmax}}\ P(x) = \frac{1}{Z} \exp(-\tfrac 1 2 x^TAx + b^Tx). }[/math]

This problem is also equivalent to the following minimization problem of the quadratic form:

[math]\displaystyle{ \underset{x}{\operatorname{min}}\ 1/2x^TAx - b^Tx. }[/math]

Which is also equivalent to the linear system of equations

[math]\displaystyle{ Ax = b. }[/math]

Convergence of the GaBP algorithm is easier to analyze (relatively to the general BP case) and there are two known sufficient convergence conditions. The first one was formulated by Weiss et al. in the year 2000, when the information matrix A is diagonally dominant. The second convergence condition was formulated by Johnson et al.[16] in 2006, when the spectral radius of the matrix

[math]\displaystyle{ \rho (I - |D^{-1/2}AD^{-1/2}|) \lt 1 \, }[/math]

where D = diag(A). Later, Su and Wu established the necessary and sufficient convergence conditions for synchronous GaBP and damped GaBP, as well as another sufficient convergence condition for asynchronous GaBP. For each case, the convergence condition involves verifying 1) a set (determined by A) being non-empty, 2) the spectral radius of a certain matrix being smaller than one, and 3) the singularity issue (when converting BP message into belief) does not occur.[17]

The GaBP algorithm was linked to the linear algebra domain,[18] and it was shown that the GaBP algorithm can be viewed as an iterative algorithm for solving the linear system of equations Ax = b where A is the information matrix and b is the shift vector. Empirically, the GaBP algorithm is shown to converge faster than classical iterative methods like the Jacobi method, the Gauss–Seidel method, successive over-relaxation, and others.[19] Additionally, the GaBP algorithm is shown to be immune to numerical problems of the preconditioned conjugate gradient method[20]

Syndrome-based BP decoding

The previous description of BP algorithm is called the codeword-based decoding, which calculates the approximate marginal probability [math]\displaystyle{ P(x|X) }[/math], given received codeword [math]\displaystyle{ X }[/math]. There is an equivalent form,[21] which calculate [math]\displaystyle{ P(e|s) }[/math], where [math]\displaystyle{ s }[/math] is the syndrome of the received codeword [math]\displaystyle{ X }[/math] and [math]\displaystyle{ e }[/math] is the decoded error. The decoded input vector is [math]\displaystyle{ x=X+e }[/math]. This variation only changes the interpretation of the mass function [math]\displaystyle{ f_a(X_a) }[/math]. Explicitly, the messages are

[math]\displaystyle{ \forall x_v\in Dom(v),\; \mu_{v \to a} (x_v) = P(X_v)\prod_{a^* \in N(v)\setminus\{a\} } \mu_{a^* \to v} (x_v). }[/math]
where [math]\displaystyle{ P(X_v) }[/math] is the prior error probability on variable [math]\displaystyle{ v }[/math][math]\displaystyle{ \forall x_v\in Dom(v),\; \mu_{a \to v} (x_v) = \sum_{\mathbf{x}'_a: x'_v = x_v } \delta(\text{syndrome}({\mathbf x}'_v)={\mathbf s}) \prod_{v^* \in N(a) \setminus \{v\}} \mu_{v^* \to a} (x'_{v^*}). }[/math]

This syndrome-based decoder doesn't require information on the received bits, thus can be adapted to quantum codes, where the only information is the measurement syndrome.

In the binary case, [math]\displaystyle{ x_i \in \{0,1\} }[/math], those messages can be simplified to cause an exponential reduction of [math]\displaystyle{ 2^{|\{v\}|+|N(v)|} }[/math] in the complexity[22][23]

Define log-likelihood ratio [math]\displaystyle{ l_v=\log \frac{u_{v \to a}(x_v=0)}{u_{v \to a} (x_v=1)} }[/math], [math]\displaystyle{ L_a=\log \frac{u_{a \to v}(x_v=0)}{u_{a \to v} (x_v=1)} }[/math], then

[math]\displaystyle{ v \to a: l_v=l_v^{(0)}+\sum_{a^* \in N(v)\setminus\{a\}} (L_{a^*}) }[/math]
[math]\displaystyle{ a \to v: L_a = (-1)^{s_a} 2 \tanh^{-1} \prod_{v^* \in N(a)\setminus\{v\}} \tanh (l_{v^*}/2) }[/math]
where [math]\displaystyle{ l_v^{(0)}=\log (P(x_v=0)/P(x_v=1)) = \text{const} }[/math]

The posterior log-likelihood ratio can be estimated as [math]\displaystyle{ l_v=l_v^{(0)}+\sum_{a \in N(v)} (L_{a}) }[/math]

References

  1. 1.0 1.1 Braunstein, A.; Mézard, M.; Zecchina, R. (2005). "Survey propagation: An algorithm for satisfiability". Random Structures & Algorithms 27 (2): 201–226. doi:10.1002/rsa.20057. 
  2. Pearl, Judea (1982). "Reverend Bayes on inference engines: A distributed hierarchical approach". AAAI-82: Pittsburgh, PA. Menlo Park, California: AAAI Press. pp. 133–136. https://www.aaai.org/Papers/AAAI/1982/AAAI82-032.pdf. Retrieved 2009-03-28. 
  3. Kim, Jin H.; Pearl, Judea (1983). "A computational model for combined causal and diagnostic reasoning in inference systems". IJCAI-83: Karlsruhe, Germany. 1. pp. 190–193. http://www.ijcai.org/Proceedings/83-1/Papers/041.pdf. Retrieved 2016-03-20. 
  4. Probabilistic Reasoning in Intelligent Systems: Networks of Plausible Inference (2nd ed.). San Francisco, CA: Morgan Kaufmann. 1988. ISBN 978-1-55860-479-7. 
  5. Yedidia, J.S.; Freeman, W.T.; Y. (January 2003). "Understanding Belief Propagation and Its Generalizations". in Lakemeyer, Gerhard; Nebel, Bernhard. Exploring Artificial Intelligence in the New Millennium. Morgan Kaufmann. pp. 239–236. ISBN 978-1-55860-811-5. http://www.merl.com/publications/TR2001-022/. Retrieved 2009-03-30. 
  6. Wainwright, M. J.; Jordan, M. I. (2007). "2.1 Probability Distributions on Graphs". Graphical Models, Exponential Families, and Variational Inference. 1. 5–9. doi:10.1561/2200000001. 
  7. Weiss, Yair (2000). "Correctness of Local Probability Propagation in Graphical Models with Loops". Neural Computation 12 (1): 1–41. doi:10.1162/089976600300015880. PMID 10636932. 
  8. Mooij, J; Kappen, H (2007). "Sufficient Conditions for Convergence of the Sum–Product Algorithm". IEEE Transactions on Information Theory 53 (12): 4422–4437. doi:10.1109/TIT.2007.909166. 
  9. Löliger, Hans-Andrea (2004). "An Introduction to Factor Graphs". IEEE Signal Processing Magazine 21 (1): 28–41. doi:10.1109/msp.2004.1267047. Bibcode2004ISPM...21...28L. 
  10. 10.0 10.1 Yedidia, J.S.; Freeman, W.T.; Weiss, Y.; Y. (July 2005). "Constructing free-energy approximations and generalized belief propagation algorithms". IEEE Transactions on Information Theory 51 (7): 2282–2312. doi:10.1109/TIT.2005.850085. http://www.merl.com/publications/TR2004-040/. Retrieved 2009-03-28. 
  11. Kikuchi, Ryoichi (1951-03-15). "A Theory of Cooperative Phenomena". Physical Review 81 (6): 988–1003. doi:10.1103/PhysRev.81.988. Bibcode1951PhRv...81..988K. 
  12. Kurata, Michio; Kikuchi, Ryoichi; Watari, Tatsuro (1953). "A Theory of Cooperative Phenomena. III. Detailed Discussions of the Cluster Variation Method". The Journal of Chemical Physics 21 (3): 434–448. doi:10.1063/1.1698926. Bibcode1953JChPh..21..434K. 
  13. Kikuchi, Ryoichi; Brush, Stephen G. (1967). "Improvement of the Cluster‐Variation Method". The Journal of Chemical Physics 47 (1): 195–203. doi:10.1063/1.1711845. Bibcode1967JChPh..47..195K. 
  14. Pelizzola, Alessandro (2005). "Cluster variation method in statistical physics and probabilistic graphical models" (in en). Journal of Physics A: Mathematical and General 38 (33): R309–R339. doi:10.1088/0305-4470/38/33/R01. ISSN 0305-4470. Bibcode2005JPhA...38R.309P. [yes|permanent dead link|dead link}}]
  15. Weiss, Yair; Freeman, William T. (October 2001). "Correctness of Belief Propagation in Gaussian Graphical Models of Arbitrary Topology". Neural Computation 13 (10): 2173–2200. doi:10.1162/089976601750541769. PMID 11570995. 
  16. Malioutov, Dmitry M.; Johnson, Jason K.; Willsky, Alan S. (October 2006). "Walk-sums and belief propagation in Gaussian graphical models". Journal of Machine Learning Research 7: 2031–2064. http://jmlr.csail.mit.edu/papers/v7/malioutov06a.html. Retrieved 2009-03-28. 
  17. Su, Qinliang; Wu, Yik-Chung (March 2015). "On convergence conditions of Gaussian belief propagation". IEEE Trans. Signal Process. 63 (5): 1144–1155. doi:10.1109/TSP.2015.2389755. Bibcode2015ITSP...63.1144S. 
  18. Gaussian belief propagation solver for systems of linear equations. By O. Shental, D. Bickson, P. H. Siegel, J. K. Wolf, and D. Dolev, IEEE Int. Symp. on Inform. Theory (ISIT), Toronto, Canada, July 2008. http://www.cs.huji.ac.il/labs/danss/p2p/gabp/
  19. Linear Detection via Belief Propagation. Danny Bickson, Danny Dolev, Ori Shental, Paul H. Siegel and Jack K. Wolf. In the 45th Annual Allerton Conference on Communication, Control, and Computing, Allerton House, Illinois, 7 Sept.. http://www.cs.huji.ac.il/labs/danss/p2p/gabp/
  20. Distributed large scale network utility maximization. D. Bickson, Y. Tock, A. Zymnis, S. Boyd and D. Dolev. In the International symposium on information theory (ISIT), July 2009. http://www.cs.huji.ac.il/labs/danss/p2p/gabp/
  21. Dave, Maulik A. (2006-12-01). "Review of "Information Theory, Inference, and Learning Algorithms by David J. C. MacKay", Cambridge University Press, 2003". ACM SIGACT News 37 (4): 34–36. doi:10.1145/1189056.1189063. ISSN 0163-5700. 
  22. Filler, Tomas (17 November 2009). "Simplification of the Belief propagation algorithm". http://dde.binghamton.edu/filler/mct/lectures/22/mct-lect22-v0.pdf. 
  23. Liu, Ye-Hua; Poulin, David (2019-05-22). "Neural Belief-Propagation Decoders for Quantum Error-Correcting Codes". Physical Review Letters 122 (20): 200501. doi:10.1103/physrevlett.122.200501. ISSN 0031-9007. PMID 31172756. 

Further reading