gov.sandia.cognition.graph.inference

Class SumProductBeliefPropagation<LabelType>

java.lang.Object

gov.sandia.cognition.graph.inference.SumProductInferencingAlgorithm<LabelType>

gov.sandia.cognition.graph.inference.SumProductBeliefPropagation<LabelType>

Type Parameters:

LabelType - The type that the nodes' labels can take on. Note that these values will be considered enumerations internally.

All Implemented Interfaces:

EnergyFunctionSolver<LabelType>

@PublicationReference(author="Jonahtan S. Yedidia, William T. Freeman, and Yair Weiss",
                      title="Understanding Belief Propagation and its Generalizations",
                      type=TechnicalReport,
                      year=2001,
                      notes="Institution: Mitsubishi Electric Research Laboratories")
public class SumProductBeliefPropagation<LabelType>
extends SumProductInferencingAlgorithm<LabelType>

This class implements the sum-product belief propagation algorithm for arbitrary energy functions. It has been tested against graph-based energy functions with widely varying node degrees and overall scales. It runs the algorithm in parallel on as many cores as the caller specifies.

Author:

tong, jdwendt

Field Summary

Fields inherited from class gov.sandia.cognition.graph.inference.SumProductInferencingAlgorithm

DEFAULT_EPS, DEFAULT_MAX_ITERATIONS, DEFAULT_NUM_THREADS, fn, nodes

Constructor Summary

Constructors

Constructor and Description
SumProductBeliefPropagation() Creates a new BP solver with the default settings.
SumProductBeliefPropagation(int maxNumIterations) Creates a new BP solver with the default settings excepting max number of iterations.
SumProductBeliefPropagation(int maxNumIterations, double eps, int numThreads) Creates a new BP solver with the specified settings.

Method Summary

Modifier and Type	Method and Description
protected void	computeTemporaryMessage(int edge) Private helper that computes the temporary message for the specified edge for the current iteration going in the specified direction (a different message for each direction)

Methods inherited from class gov.sandia.cognition.graph.inference.SumProductInferencingAlgorithm

getBelief, init, solve

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SumProductBeliefPropagation

public SumProductBeliefPropagation(int maxNumIterations,
                                   double eps,
                                   int numThreads)

Creates a new BP solver with the specified settings.

Parameters:

maxNumIterations - The maximum number of iterations that will be run

eps - The stopping epsilon that will be used

numThreads - The number of threads that will be used

SumProductBeliefPropagation

public SumProductBeliefPropagation(int maxNumIterations)

Creates a new BP solver with the default settings excepting max number of iterations.

Parameters:

maxNumIterations - The maximum number of iterations that will be run

SumProductBeliefPropagation

public SumProductBeliefPropagation()

Creates a new BP solver with the default settings.

Method Detail

computeTemporaryMessage

protected void computeTemporaryMessage(int edge)

Description copied from class: SumProductInferencingAlgorithm

Private helper that computes the temporary message for the specified edge for the current iteration going in the specified direction (a different message for each direction)

Specified by:

computeTemporaryMessage in class SumProductInferencingAlgorithm<LabelType>

Parameters:

edge - The edge to compute the temporary message for (you can't replace the current message yet as the current message may be needed for other edges).