论文标题

APSNET:基于注意的点云采样

APSNet: Attention Based Point Cloud Sampling

论文作者

Ye, Yang, Yang, Xiulong, Ji, Shihao

论文摘要

处理大点云是一项具有挑战性的任务。因此,数据通常被降采样至较小的大小,以便可以将其存储,传输和处理更有效,而不会产生明显的性能降解。传统的任务无关采样方法(例如最远的点采样(FPS))在采样点云时不会考虑下游任务,因此通常会采样对任务的非信息点。本文探讨了针对3D点云的面向任务的采样,旨在采样专门针对下游的感兴趣任务的部分。与FPS相似,我们假设接下来要采样该点应该在很大程度上取决于已经采样的点。因此,我们将点云采样作为顺序生成过程,并开发基于注意力的点云采样网络(APSNET)来解决此问题。在每个时间步骤中,APSNet通过利用先前采样点的历史记录,并示例最有用的方法,从而参观云中的所有要点。提出了监督的学习和基于知识蒸馏的自我监督的APSNET学习。此外,研究了APSNET对多个样本量的联合训练,从而导致单个APSNET可以产生具有突出性能的样品的任意长度。广泛的实验表明,在各种下游任务中,APSNET与最先进的实验相比,包括3D点云分类,重建和注册。

Processing large point clouds is a challenging task. Therefore, the data is often downsampled to a smaller size such that it can be stored, transmitted and processed more efficiently without incurring significant performance degradation. Traditional task-agnostic sampling methods, such as farthest point sampling (FPS), do not consider downstream tasks when sampling point clouds, and thus non-informative points to the tasks are often sampled. This paper explores a task-oriented sampling for 3D point clouds, and aims to sample a subset of points that are tailored specifically to a downstream task of interest. Similar to FPS, we assume that point to be sampled next should depend heavily on the points that have already been sampled. We thus formulate point cloud sampling as a sequential generation process, and develop an attention-based point cloud sampling network (APSNet) to tackle this problem. At each time step, APSNet attends to all the points in a cloud by utilizing the history of previously sampled points, and samples the most informative one. Both supervised learning and knowledge distillation-based self-supervised learning of APSNet are proposed. Moreover, joint training of APSNet over multiple sample sizes is investigated, leading to a single APSNet that can generate arbitrary length of samples with prominent performances. Extensive experiments demonstrate the superior performance of APSNet against state-of-the-arts in various downstream tasks, including 3D point cloud classification, reconstruction, and registration.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源