通过遮挡完成的无监督点云预训练

我们描述了一种用于点云的简单预训练方法。该方法分为三个步骤：1. 遮蔽在相机视图中被遮挡的所有点；2. 学习一个编码器-解码器模型以重建这些被遮挡的点；3. 使用编码器的权重作为下游点云任务的初始化。我们发现，即使仅从ModelNet40构建单一的预训练数据集，这种方法也能在不同的数据集和编码器上提高多种下游任务的准确性。具体而言，我们的方法在物体分类以及基于部件和语义分割任务中均优于先前的预训练方法。我们对预训练特征进行了研究，发现它们导致了广泛的下游最小值，具有较高的变换不变性，并且其激活与部件标签高度相关。代码和数据可在以下地址获取：https://github.com/hansen7/OcCo