Factor graph

Not to be confused with Graph factorization.

In probability theory and its applications, a factor graph is a particular type of graphical model, with applications in Bayesian inference, that enables efficient computation of marginal distributions through the sum-product algorithm. One of the important success stories of factor graphs and the sum-product algorithm is the decoding of capacity-approaching error-correcting codes, such as LDPC and turbo codes.

A factor graph is an example of a hypergraph, in that an arrow (i.e., a factor node) can connect more than one (normal) node.

When there are no free variables, the factor graph of a function f is equivalent to the constraint graph of f, which is an instance to a constraint satisfaction problem.

Definition

A factor graph is a bipartite graph representing the factorization of a function. Given a factorization of a function $g(X_{1},X_{2},\dots ,X_{n})$,

$g(X_{1},X_{2},\dots ,X_{n})=\prod _{j=1}^{m}f_{j}(S_{j}),$

where $S_{j}\subseteq \{X_{1},X_{2},\dots ,X_{n}\}$, the corresponding factor graph $G=(X,F,E)$ consists of variable vertices $X=\{X_{1},X_{2},\dots ,X_{n}\}$, factor vertices $F=\{f_{1},f_{2},\dots ,f_{m}\}$, and edges $E$. The edges depend on the factorization as follows: there is an undirected edge between factor vertex $f_{j}$ and variable vertex $X_{k}$ when $X_{k}\in S_{j}$. The function is tacitly assumed to be real-valued: $g(X_{1},X_{2},\dots ,X_{n})\in {\mathbb {R}}$.

Factor graphs can be combined with message passing algorithms to efficiently compute certain characteristics of the function $g(X_{1},X_{2},\dots ,X_{n})$, such as the marginal distributions.

Examples

An example factor graph

Consider a function that factorizes as follows:

$g(X_{1},X_{2},X_{3})=f_{1}(X_{1})f_{2}(X_{1},X_{2})f_{3}(X_{1},X_{2})f_{4}(X_{2},X_{3})$,

with a corresponding factor graph shown on the right. Observe that the factor graph has a cycle. If we merge $f_{2}(X_{1},X_{2})f_{3}(X_{1},X_{2})$ into a single factor, the resulting factor graph will be a tree. This is an important distinction, as message passing algorithms are usually exact for trees, but only approximate for graphs with cycles.

Message passing on factor graphs

A popular message passing algorithm on factor graphs is the sum-product algorithm, which efficiently computes all the marginals of the individual variables of the function. In particular, the marginal of variable $X_{k}$ is defined as

$g_{k}(X_{k})=\sum _{X_{\bar {k}}}g(X_{1},X_{2},\dots ,X_{n})$

where the notation $X_{\bar {k}}$ means that the summation goes over all the variables, except $X_{k}$. The messages of the sum-product algorithm are conceptually computed in the vertices and passed along the edges. A message from or to a variable vertex is always a function of that particular variable. For instance, when a variable is binary, the messages over the edges incident to the corresponding vertex can be represented as vectors of length 2: the first entry is the message evaluated in 0, the second entry is the message evaluated in 1. When a variable belongs to the field of real numbers, messages can be arbitrary functions, and special care needs to be taken in their representation.

In practice, the sum-product algorithm is used for statistical inference, whereby $g(X_{1},X_{2},\dots ,X_{n})$ is a joint distribution or a joint likelihood function, and the factorization depends on the conditional independencies among the variables.

The Hammersley–Clifford theorem shows that other probabilistic models such as Markov networks and Bayesian networks can be represented as factor graphs; the latter representation is frequently used when performing inference over such networks using belief propagation. On the other hand, Bayesian networks are more naturally suited for generative models, as they can directly represent the causalities of the model.

See also

• Belief propagation
• Bayesian inference
• Conditional probability
• Markov network
• Bayesian network
• Hammersley–Clifford theorem

External links

• A tutorial-style dissertation by Volker Koch
• An Introduction to Factor Graphs by Hans-Andrea Loeliger, IEEE Signal Processing Magazine, January 2004, pp. 28–41.
• dimple an open-source tool for building and solving factor graphs in MATLAB.

References

• Clifford (1990), "Markov random fields in statistics", in Grimmett, G.R.; Welsh, D.J.A. (eds.), Disorder in Physical Systems, J.M. Hammersley Festschrift, Oxford University Press, pp. 19–32
• Frey, Brendan J. (2003), "Extending Factor Graphs so as to Unify Directed and Undirected Graphical Models", in jain, Nitin (ed.), UAI'03, Proceedings of the 19th Conference in Uncertainty in Artificial Intelligence, August 7–10, Acapulco, Mexico, Morgan Kaufmann, pp. 257–264
• Kschischang, Frank R.; Frey, Brendan J.; Loeliger, Hans-Andrea (2001), "Factor Graphs and the Sum-Product Algorithm", IEEE Transactions on Information Theory, 47 (2): 498–519, doi:10.1109/18.910572, retrieved 2008-02-06.
• Wymeersch, Henk (2007), Iterative Receiver Design, Cambridge University Press, ISBN 0-521-87315-0