Factor graph

Examples

Consider a function that factorizes as follows:

${\displaystyle 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 ${\displaystyle 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 ${\displaystyle X_k}$ is defined as

${\displaystyle g_k(X_k)=\sum _{X_{\overline{k}}}g(X_1,X_2,\dots ,X_n)}$

where the notation ${\displaystyle X_{\overline{k}}}$ means that the summation goes over all the variables, except ${\displaystyle 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, where ${\displaystyle 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 Bayesian networks and Markov 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.