Neural varifolds: an aggregate representation for quantifying the geometry of point clouds
Juheon Lee, Xiaohao Cai, Carola-Bibian Schönlieb, Simon Masnou
TL;DR
This work introduces neural varifolds, a geometry-aware representation for point clouds that encodes surfaces as measures over positions and tangent planes, enabling principled comparison via RKHS-based kernels. By leveraging neural tangent kernels, the authors build two variants (NTK1 and NTK2) that yield varifold-based similarity metrics across sets of point clouds, and demonstrate their utility for shape matching, few-shot shape classification, and 3D shape reconstruction. The approach achieves strong performance in shape matching and few-shot classification, and competes with state-of-the-art reconstruction methods, while presenting favorable computation in several settings. The neural varifold framework advances geometry-aware learning on unstructured 3D data by connecting deep learning with geometric measure theory and NTK theory.
Abstract
Point clouds are popular 3D representations for real-life objects (such as in LiDAR and Kinect) due to their detailed and compact representation of surface-based geometry. Recent approaches characterise the geometry of point clouds by bringing deep learning based techniques together with geometric fidelity metrics such as optimal transportation costs (e.g., Chamfer and Wasserstein metrics). In this paper, we propose a new surface geometry characterisation within this realm, namely a neural varifold representation of point clouds. Here the surface is represented as a measure/distribution over both point positions and tangent spaces of point clouds. The varifold representation quantifies not only the surface geometry of point clouds through the manifold-based discrimination, but also subtle geometric consistencies on the surface due to the combined product space. This study proposes neural varifold algorithms to compute the varifold norm between two point clouds using neural networks on point clouds and their neural tangent kernel representations. The proposed neural varifold is evaluated on three different sought-after tasks -- shape matching, few-shot shape classification and shape reconstruction. Detailed evaluation and comparison to the state-of-the-art methods demonstrate that the proposed versatile neural varifold is superior in shape matching and few-shot shape classification, and is competitive for shape reconstruction.
