一种线性时间和空间的局部点云几何编码器:基于向量化核混合(VecKM)
计算机视觉与模式识别
2024-07-02 v4 计算几何
摘要
我们提出了VecKM,一种描述性强且计算高效的局部点云几何编码器。VecKM利用一种独特的方法,通过向量化核混合来表示局部点云。这种表示的描述性由两个定理支持,这些定理验证了其重建和保持局部形状相似性的能力。与对局部点云进行下采样的现有编码器不同,VecKM使用所有邻近点构建局部几何编码,产生更具描述性的编码。此外,VecKM计算高效且可扩展到大型点云输入:VecKM将内存成本从降低到;并将主要运行时成本从计算个MLP降低到个MLP,其中是点云的大小,是邻域大小,是编码维度,是一个边际因子。这种效率源于VecKM独特的可分解特性,该特性消除了显式将点分组为邻居的需要。在法线估计任务中,VecKM不仅展示了100倍的推理速度提升,还实现了最高的准确性和最强的鲁棒性。在分类和分割任务中,将VecKM作为预处理模块集成,其性能始终优于PointNet、PointNet++和点Transformer基线,并且运行速度始终快达10倍。
引用
@article{arxiv.2404.01568,
title = {A Linear Time and Space Local Point Cloud Geometry Encoder via Vectorized Kernel Mixture (VecKM)},
author = {Dehao Yuan and Cornelia Fermüller and Tahseen Rabbani and Furong Huang and Yiannis Aloimonos},
journal= {arXiv preprint arXiv:2404.01568},
year = {2024}
}
备注
ICML2024 Conference Paper