A Fully-Pipelined Fpga Design for Tree-Reweighted Message Passing Algorithm

Wenlai Zhao,Haohuan Fu,Guangwen Yang
DOI: https://doi.org/10.1109/fccm.2014.59
2014-01-01
Abstract:Maximum a posteriori (MAP) probability inference algorithms for Markov Random Field (MRF) are widely used in many applications, especially for solving label assignment problems in computer vision. Sequential tree-reweighted message passing (TRW-S) is one of the inference algorithms which shows good quality in finding optimal solutions. However, the performance of TRW-S on CPU cannot meet the requirements of many real-time applications due to the sequential scheme and the high memory, bandwidth and computational costs. In our work, we propose a fully pipelined FPGA design for the tree-reweighted message passing algorithm. We first propose a modified algorithm based on TRW-S, which can be fully pipelined in hardware. Then we build a hybrid CPU/FPGA system to test the performance of the design using stereo matching application. Experimental results show that our design provides about 100 times speedup over a single-core software implementation of TRW-S. And compared with an existing multi-FPGA design, we can achieve 2 times speedup on the overall performance and 8 times speedup for one FPGA card. Moreover, our design can work at a video rate in many cases (167 frame/sec for a standard stereo matching test case), which makes it a promising work for many real-time applications.
What problem does this paper attempt to address?