论文标题
图形分类的简单但有效的方法
A Simple yet Effective Method for Graph Classification
论文作者
论文摘要
在深层神经网络中,通常可以通过增加先前开发的基本模型的复杂性来获得更好的结果。但是,目前尚不清楚是否有一种方法可以通过降低此类模型的复杂性来提高性能。直观地,考虑到一个问题,更简单的数据结构带有更简单的算法。在这里,我们研究了改善图形分类性能的可行性,同时简化学习过程。受图形结构熵的启发,我们将数据样本从图形转换为编码树,这是图形数据的简单但必不可少的结构。此外,我们提出了一个新的消息传递方案,称为层次报告,其中特征通过遵循编码树的层次结构从叶子节点传递到根节点。然后,我们提出一个树内核和一个卷积网络,以实现我们的图形分类方案。通过设计的消息传递方案,Tree内核和卷积网络的运行时复杂性低于$ O(n)$,而Weisfeiler-Lehman subtree内核和其他至少$ O(HM)$的图形神经网络。我们通过几种图形分类基准来验证我们的方法,并证明它们比竞争方法更好地实现了更好的性能和更低的计算消耗。
In deep neural networks, better results can often be obtained by increasing the complexity of previously developed basic models. However, it is unclear whether there is a way to boost performance by decreasing the complexity of such models. Intuitively, given a problem, a simpler data structure comes with a simpler algorithm. Here, we investigate the feasibility of improving graph classification performance while simplifying the learning process. Inspired by structural entropy on graphs, we transform the data sample from graphs to coding trees, which is a simpler but essential structure for graph data. Furthermore, we propose a novel message passing scheme, termed hierarchical reporting, in which features are transferred from leaf nodes to root nodes by following the hierarchical structure of coding trees. We then present a tree kernel and a convolutional network to implement our scheme for graph classification. With the designed message passing scheme, the tree kernel and convolutional network have a lower runtime complexity of $O(n)$ than Weisfeiler-Lehman subtree kernel and other graph neural networks of at least $O(hm)$. We empirically validate our methods with several graph classification benchmarks and demonstrate that they achieve better performance and lower computational consumption than competing approaches.