Gaussian Belief Propagation Network for Depth Completion
It addresses the challenge of handling sparse and irregular depth data in deep networks for depth completion, which is crucial for applications like robotics and autonomous driving, representing a novel hybrid approach rather than an incremental improvement.
The paper tackles depth completion from sparse depth measurements by introducing the Gaussian Belief Propagation Network (GBPN), a hybrid framework that integrates deep learning with probabilistic graphical models, achieving state-of-the-art performance on NYUv2 and KITTI benchmarks with superior robustness across varying sparsity conditions.
Depth completion aims to predict a dense depth map from a color image with sparse depth measurements. Although deep learning methods have achieved state-of-the-art (SOTA), effectively handling the sparse and irregular nature of input depth data in deep networks remains a significant challenge, often limiting performance, especially under high sparsity. To overcome this limitation, we introduce the Gaussian Belief Propagation Network (GBPN), a novel hybrid framework synergistically integrating deep learning with probabilistic graphical models for end-to-end depth completion. Specifically, a scene-specific Markov Random Field (MRF) is dynamically constructed by the Graphical Model Construction Network (GMCN), and then inferred via Gaussian Belief Propagation (GBP) to yield the dense depth distribution. Crucially, the GMCN learns to construct not only the data-dependent potentials of MRF but also its structure by predicting adaptive non-local edges, enabling the capture of complex, long-range spatial dependencies. Furthermore, we enhance GBP with a serial \& parallel message passing scheme, designed for effective information propagation, particularly from sparse measurements. Extensive experiments demonstrate that GBPN achieves SOTA performance on the NYUv2 and KITTI benchmarks. Evaluations across varying sparsity levels, sparsity patterns, and datasets highlight GBPN's superior performance, notable robustness, and generalizable capability.