高级检索
刘复昌, 李晨璇, 王延斌, 缪永伟. 结合MAML和Dirichlet过程的小样本点云分类[J]. 计算机辅助设计与图形学学报.
引用本文: 刘复昌, 李晨璇, 王延斌, 缪永伟. 结合MAML和Dirichlet过程的小样本点云分类[J]. 计算机辅助设计与图形学学报.
Few-shot Point Clouds Classification based on MAML and Dirichlet Process[J]. Journal of Computer-Aided Design & Computer Graphics.
Citation: Few-shot Point Clouds Classification based on MAML and Dirichlet Process[J]. Journal of Computer-Aided Design & Computer Graphics.

结合MAML和Dirichlet过程的小样本点云分类

Few-shot Point Clouds Classification based on MAML and Dirichlet Process

  • 摘要: 点云被广泛使用在各种三维应用场景中, 但是实际应用中通常存在扫描、标注费时费力等局限性, 因此基于小样本数据集的点云分类网络更加符合应用需求. 为了有效地提高深度学习分类算法在小样本点云数据集上的分类效果, 提出一种针对小样本数据集的点云分类方法. 针对训练数据集不平衡问题, 首先采用基于相似度依赖的Dirichlet中餐馆过程对数据集进行预处理, 其在无需人工指定聚类个数的前提下对样本进行重新聚类以提升分类网络在小样本数据集上的性能; 然后在重新聚类后的样本上使用MAML(Model-Agnostic Meta-Learning)算法训练PointNet++, 达到用少量点云样本就能快速适应新任务的能力. 该方法不但降低了模型对数据量的依赖, 提高模型泛化能力, 而且成功地把MAML算法从二维图像分类拓展到三维点云分类中; 在Modelnet40数据集上的实验结果表明, 与PointNet++相比, 该方法的训练时间减少了一半, 分类准确率平均提高6.67%; 实验验证了该方法针对小样本数据集上的有效性.

     

    Abstract: Point clouds are widely used in various 3D application scenes. However, there are some issues in the real-world scenarios, such as labor-consuming and time-costing on data annotation. Few-shot point clouds classification neural networks are popular and can satisfy practical application requirements. To improve the performance of deep learning models in small sample point cloud classification, we propose a point clouds classification method based on MAML and Dirichlet process. To handle the imbalance of data sets, this paper proposes a similarity dependent Dirichlet process called Chinese restaurant model to cluster the dataset. This cluster process can complete clustering without manually specifying the number of clusters and result in higher classification accuracy with the optimal clustering. Subsequently, we apply MAML framework to train PointNet++ network based on samples processed by our cluster model. With the help of our data clustering, the MAML algorithm can adapt to new tasks quickly through a small number of data samples. The proposed method alleviates the issue of data hungry on deep learning for point clouds and improve the generalization of neural networks, as well as successfully extending the MAML algorithm from two-dimensional image classification to three-dimensional point clouds classification. Compared with PointNet++ model, the training time is reduced by half and the overall accuracy is improved by 6.67%. Experimental results demonstrate the effectiveness of our proposed small sample point cloud classification algorithm on public dataset.

     

/

返回文章
返回