SuBloNet: Sparse Super Block Networks for Large Scale Volumetric Fusion

Rückert D, Stamminger M (2021)


Publication Language: English

Publication Type: Conference contribution, Original article

Publication year: 2021

ISBN: 978-3-03868-161-8

DOI: 10.2312/vmv.20211375

Abstract

Training and inference of convolutional neural networks (CNNs) on truncated signed distance fields (TSDFs) is a challenging task. Large parts of the scene are usually empty, which makes dense implementations inefficient in terms of memory consumption and compute throughput. However, due to the truncation distance, non-zero values are grouped around the surface creating small dense blocks inside the large empty space. We show that this structure can be exploited by storing the TSDF in a block sparse tensor and then decomposing it into rectilinear super blocks. A super block is a dense 3d cuboid of variable size and can be processed by conventional CNNs. We analyze the rectilinear decomposition and present a formulation for computing the bandwidth-optimal solution given a specific network architecture. However, this solution is NP-complete, therefore we also a present a heuristic approach for fast training and inference tasks. We verify the effectiveness of SuBloNet and report a speedup of 4x towards dense implementations and 1.7x towards state-of-the-art sparse implementations. Using the super block architecture, we show that recurrent volumetric fusion is now possible on large scale scenes. Such a systems is able to reconstruct high-quality surfaces from few noisy depth images.

Authors with CRIS profile

How to cite

APA:

Rückert, D., & Stamminger, M. (2021). SuBloNet: Sparse Super Block Networks for Large Scale Volumetric Fusion. In Andres, Bjoern and Campen, Marcel and Sedlmair, Michael (Eds.), Proceedings of the Vision, Modeling and Visualization 2021.

MLA:

Rückert, Darius, and Marc Stamminger. "SuBloNet: Sparse Super Block Networks for Large Scale Volumetric Fusion." Proceedings of the Vision, Modeling and Visualization 2021 Ed. Andres, Bjoern and Campen, Marcel and Sedlmair, Michael, 2021.

BibTeX: Download