关键词:
等变图神经网络
三维图数据
E(3)等变群
排列等变
点云分类
三维模型分类
摘要:
作为三维图数据的重要特征,等变性近年来一直受到研究者们的密切关注.研究者们通过设计针对不同等变性的量子门集来构建等变量子图神经网络,显著提升了图学习模型的性能.然而,三维图数据包含多种等变性,如旋转等变性、平移等变性、反射等变性和排列等变性等,使研究者们难以构建满足所有等变性的量子等变图神经网络.为了解决上述问题,本文设计了一种三维图节点相对坐标计算方法,并提出了一种新型的量子群等变图神经网络.在该网络中,我们构建了一个群等变编码模块以编码相对坐标,并用一个排列等变卷积模块来学习编码的几何信息,最后采用一个纠缠模块来聚合全局特征. ModelNet40数据集和ShapeNet数据集上的分类实验表明,所提模型表现优于QGCNN, sQCNN-3D和PI-QSVM,同时在训练参数更少的情况下,准确度接近Q3DGL.