We present an attention-based spatial graph convolution (AGC) for graph neural networks (GNNs). Existing AGCs focus on only using node-wise features and utilizing one type of attention function when calculating attention weights. Instead, we propose two methods to improve the representational power of AGCs by utilizing 1) structural information in a high-dimensional space and 2) multiple attention functions when calculating their weights. The first method computes a local structure representation of a graph in a high-dimensional space. The second method utilizes multiple attention functions simultaneously in one AGC. Both approaches can be combined. We also propose a GNN for the classification of point clouds and that for the prediction of point labels in a point cloud based on the proposed AGC. According to experiments, the proposed GNNs perform better than existing methods. Our codes open at https://github.com/liyang-tuat/SFAGC.
翻译:我们提出了一种基于注意力的空间图卷积(AGC)用于图神经网络。现有的AGC仅关注利用节点特征,并在计算注意力权重时使用单一注意力函数。相反,我们提出了两种方法以提高AGC的表示能力:1)利用高维空间中的结构信息;2)在计算注意力权重时利用多种注意力函数。第一种方法计算图在高维空间中的局部结构表示。第二种方法在一个AGC中同时使用多种注意力函数。这两种方法可以结合使用。我们还基于所提出的AGC,提出了用于点云分类的图神经网络以及用于点云中点标签预测的图神经网络。实验表明,所提出的图神经网络性能优于现有方法。我们的代码已在 https://github.com/liyang-tuat/SFAGC 开源。