290 likes | 376 Views
A Trainable Graph Combination Scheme for Belief Propagation. Kai Ju Liu New York University. Images. Pairwise Markov Random Field. 4. 1. 2. 3. 5. Basic structure: vertices, edges. and observed value y i. Compatibility between states and observed values,.
E N D
A Trainable Graph Combination Scheme for Belief Propagation Kai Ju Liu New York University
Pairwise Markov Random Field 4 1 2 3 5 • Basic structure: vertices, edges
and observed value yi • Compatibility between states and observed values, • Compatibility between neighboring vertices i and j, Pairwise Markov Random Field • Basic structure: vertices, edges • Vertex i has set of possible states Xi
Marginal probability: Pairwise MRF: Probabilities • Joint probability: • Advantage: allows average over ambiguous states • Disadvantage: complexity exponential in number of vertices
Belief Propagation 4 1 2 3 5
Messages propagate information: Belief Propagation • Beliefs replace probabilities:
When can we calculate beliefs exactly? When do beliefs equal probabilities? When is belief propagation efficient? Answer: Singly-Connected Graphs (SCG’s) • Graphs without loops • Messages terminate at leaf vertices • Beliefs equal probabilities • Complexity in previous example reduced from 13S5 to 24S2 BP: Questions
Messages do not terminate 1 2 4 3 BP on Loopy Graphs • Energy approximation schemes [Freeman et al.] • Standard belief propagation • Generalized belief propagation • Standard belief propagation • Approximates Gibbs free energy of system by Bethe free energy • Iterates, requiring convergence criteria
BP on Loopy Graphs • Tree-based reparameterization [Wainwright] • Reparameterizes distributions on singly-connected graphs • Convergence improved compared to standard belief propagation • Permits calculation of bounds on approximation errors
BP-TwoGraphs • Eliminates iteration • Utilizes advantages of SCG’s
Calculate beliefs on each set of SCG’s: • Select set of beliefs with minimum entropy BP-TwoGraphs • Consider loopy graph with n vertices • Select two sets of SCG’s that approximate the graph
Rectangular grid of pixel vertices Hi: horizontal graphs Gi: vertical graphs BP-TwoGraphs on Images original graph vertical graph horizontal graph
Image Segmentation add noise segment
Image Segmentation Revisited add noise ground truth max-flow ground truth
Rectangular grid of pixel vertices Hi: horizontal lines Gi: vertical lines BP-TwoLines original graph vertical line horizontal line
Boundary-Based Image Segmentation: Window Vertices • Square 2-by-2 window of pixels • Each pixel has two states • foreground • background
Conclusion • BP-TwoGraphs • Accurate and efficient • Extensive use of beliefs • Trainable parameters • Future work • Multiple states • Stereo • Image fusion