福州大学学报(自然科学版)2024,Vol.52Issue(1) :1-6.DOI:10.7631/issn.1000-2243.22528

基于曲率图卷积的非均匀点云掩码自编码器

Curvature graph convolution for nonuniform point cloud masked autoencoders

黄敏明 傅仰耿
福州大学学报(自然科学版)2024,Vol.52Issue(1) :1-6.DOI:10.7631/issn.1000-2243.22528

基于曲率图卷积的非均匀点云掩码自编码器

Curvature graph convolution for nonuniform point cloud masked autoencoders

黄敏明 1傅仰耿1
扫码查看

作者信息

  • 1. 福州大学计算机与大数据学院,福建 福州 350108
  • 折叠

摘要

提出一种基于曲率图卷积的非均匀分组与掩码策略,用以优化掩码自编码器.首先,提出曲率图卷积以避免固定邻域导致的归纳偏差;其次,在曲率图卷积后引入图池化层,根据点云局部特征进行池化操作并分组;最后,在池化层输出特征的基础上学习每个分组的掩码概率来避免冗余.实验结果表明,本方法能有效提高点云掩码自编码器在下游任务的泛化效果,在ModelNet40 上的分类精度达到93.7%,在Completion3Dv2 上的补全精度达到 5.08,均优于目前主流方法.

Abstract

A nonuniform grouping and mask strategy are proposed based on curvature graph convolu-tion to further optimize the mask autoencoder.First,curvature graph convolution is proposed to avoid the induction bias caused by fixed neighborhoods.Second,a graph pooling layer is introduced after curvature convolution,which is pooled and grouped according to the local features of point clouds.Finally,the mask probability of each group is learned based on the output features of the pooling layer to avoid redundancy.Experiments show that our method can effectively improve the performance of mask autoencoder in downstream tasks.Our pretrained models achieve 93.7%classification accuracy on ModelNet40 and 5.08 completion accuracy on Completion3Dv2,outperforming current mainstream methods.

关键词

自编码器/点云/图卷积神经网络/预训练/自监督学习

Key words

autoencoders/point cloud/graph neural network/pre-training/self-supervised learning

引用本文复制引用

基金项目

国家自然科学基金资助项目(12271098)

国家自然科学基金资助项目(61773123)

福建省自然科学基金资助项目(2019J01647)

出版年

2024
福州大学学报(自然科学版)
福州大学

福州大学学报(自然科学版)

CSTPCD北大核心
影响因子:0.35
ISSN:1000-2243
参考文献量18
段落导航相关论文