Hardware-Efficient Belief Propagation

Yen-Chieh Lai

National Taiwan University

 

IEEE Conference on Computer Vision and Pattern Recognition (CVPR) 2009

(patent pending)


Abstract

Belief propagation (BP) is an effective algorithm for solving energy minimization problems in computer vision. However, it requires enormous memory, bandwidth, and computation because messages are iteratively passed between nodes in the Markov random field (MRF). In this paper, we propose two methods to address this problem.
The first method is a message passing scheme called tile-based belief propagation. The key idea of this method is that a message can be well approximated from other faraway ones. We split the MRF into many tiles and perform BP within each one. To preserve the global optimality, we store the outgoing boundary messages of a tile and use them when performing BP in the neighboring tiles. The tile-based BP only requires 1-5% memory and 0.2-1% bandwidth of the ordinary BP.
The second method is an O(L) message construction algorithm for the robust functions commonly used for describing the smoothness terms in the energy function. We find that many variables in constructing a message are repetitive; thus these variables can be calculated once and reused many times. The proposed algorithms are suitable for parallel implementations. We design a low-power VLSI circuit for disparity estimation that can construct 440M messages per second and generate high quality disparity maps in near real-time. We also implement the proposed algorithms on a GPU, which can calculate messages 4 times faster than the sequential O(L) method.

The floorplan of the developed VLSI chip. It can generate 440M 64-D messages per second using less than 500mW power.


Downloads

Paper [PDF]

Supplemental material [PDF]

Reference implementation (compatible to the Middlebury MRF library) [ZIP]
Poster [PDF]


Acknowledgements

This project is supported in part by NSC in grant 96-2628-E-002-005-MY2 and Himax Technologies Inc. in grant 96SB20.


@InProceedings{Liang09,
  author    = {Chia-Kai Liang and Chao-Chung Cheng and Yen-Chieh Lai and Liang-Gee Chen and Homer H. Chen},
  title     = {Hardware-Efficient Belief Propagation},
  booktitle = {IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  pages     = {80--87},
  year      = {2009}
}

Last update: July 3, 2009, Chia-Kai Liang