用图神经网络学习点云上的拉普拉斯算子,显著提升精度和泛化能力。
Neural Laplacian Operator for 3D Point Clouds
- 基于KNN图构建图神经网络,直接学习点云的拉普拉斯算子。
- 在ShapeNet子集上训练,误差比之前方法降低一个数量级。
- 擅长处理稀疏点云中的细结构与锐边,适合未见形状的泛化应用。
离散拉普拉斯算子在3D几何处理中至关重要,但其在点云上的定义仍具挑战性。以往方法依赖局部三角剖分近似底层流形以定义算子,但鲁棒性和精度不足。本文仅使用输入点云的K近邻(KNN)图,通过图神经网络(GNN)学习拉普拉斯算子。由于真实算子定义在具有不同连通性的网格流形上,无法直接用于训练。为此,我们提出一种新训练策略:通过模拟真实算子在一组探测函数上的行为,使学习到的算子表现与真实算子相似。我们在ShapeNet子集上训练模型,并在多种点云上评估。相比已有方法,本方法误差降低一个数量级,尤其在稀疏点云、细结构或锐边场景中表现优异。模型对未见形状具有强泛化能力。利用学习到的算子,我们直接在点云上执行一系列拉普拉斯相关几何处理算法,获得高精度结果,为点云几何处理开辟新可能。代码与训练模型见https://github.com/IntelligentGeometry/NeLo。
原文摘要 · Abstract (English)
The discrete Laplacian operator holds a crucial role in 3D geometry processing, yet it is still challenging to define it on point clouds. Previous works mainly focused on constructing a local triangulation around each point to approximate the underlying manifold for defining the Laplacian operator, which may not be robust or accurate. In contrast, we simply use the K-nearest neighbors (KNN) graph constructed from the input point cloud and learn the Laplacian operator on the KNN graph with graph neural networks (GNNs). However, the ground-truth Laplacian operator is defined on a manifold mesh with a different connectivity from the KNN graph and thus cannot be directly used for training. To train the GNN, we propose a novel training scheme by imitating the behavior of the ground-truth Laplacian operator on a set of probe functions so that the learned Laplacian operator behaves similarly to the ground-truth Laplacian operator. We train our network on a subset of ShapeNet and evaluate it across a variety of point clouds. Compared with previous methods, our method reduces the error by an order of magnitude and excels in handling sparse point clouds with thin structures or sharp features. Our method also demonstrates a strong generalization ability to unseen shapes. With our learned Laplacian operator, we further apply a series of Laplacian-based geometry processing algorithms directly to point clouds and achieve accurate results, enabling many exciting possibilities for geometry processing on point clouds. The code and trained models are available at https://github.com/IntelligentGeometry/NeLo.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。