基于节点采样的子结构代表层次池化图卷积网络模型

    Sub-structure Representative Hierarchical Pooling Graph Convolutional Network Model Based on Node Sampling

    • 摘要: 为解决目前基于节点采样的图池化方法中所存在的评估节点重要性的策略过于简单以及子结构特征信息大量丢失等问题, 提出了基于节点采样的子结构代表层次池化模型(sub-structure representative hierarchical pooling model based on node sampling, SsrPool)。该模型主要包括子结构代表节点选择模块和子结构代表节点特征生成模块2个部分。首先, 子结构代表节点选择模块同时考虑了节点特征信息以及结构信息, 利用不同方法评估节点重要性并通过不同重要性分数协作产生鲁棒的节点排名以指导节点选择。其次, 子结构代表节点特征生成模块通过特征融合保留局部子结构特征信息。通过将SsrPool与现有神经网络相结合, 在不同规模公共数据集上的图分类实验结果证明了SsrPool的有效性。

       

      Abstract: To solve the problems of the current graph pooling method based on the node sampling, such as the simplistic strategy of node importance evaluation and the massive loss of sub-structure feature information of graph, a sub-structure representative hierarchical pooling model based on node sampling dubbed as SsrPool was proposed. This method mainly includes a sub-structure representative node selection module and a sub-structure representative node feature generation module. First, considering both the structure and feature information of the graph, different methods were used to evaluate the node importance, and the collaboration of varying importance scores generated a robust node ranking to guide node selection in the sub-structure representative node selection module. Second, sub-structure features information was retained through feature fusion in the sub-structure representative features generation module. By combining the SsrPool model with existing graph neural networks, experimental results of graph classification on different public datasets demonstrate the effectiveness of SsrPool.

       

    /

    返回文章
    返回