
摘要
消息传递神经网络(MPNNs)在图学习任务中得到了广泛应用,但受限于信息交换范围狭窄的问题,即在每轮消息传递过程中仅限于邻近节点之间的信息交换。为了解决这些限制,已提出了多种策略,包括引入虚拟节点以促进全局信息交换。在本研究中,我们提出了一种层次支持图(Hierarchical Support Graph, HSG),这是通过递归粗化原始图而扩展的虚拟节点概念。该方法提供了一个灵活的框架,用于增强图中的信息流动,且独立于所使用的具体MPNN层。我们对HSG进行了理论分析,探讨了其经验性能,并展示了HSG能够超越其他使用虚拟节点的方法,在多个数据集上取得了最先进的结果。
代码仓库
carlosinator/support-graphs
官方
pytorch
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| graph-classification-on-peptides-func | GatedGCN-HSG | AP: 0.6866±0.0038 |
| graph-property-prediction-on-ogbg-molpcba | GatedGCN-HSG | Test AP: 0.3129±0.0020 |
| graph-regression-on-peptides-struct | GatedGCN-HSG | MAE: 0.2421±0.0007 |
| node-classification-on-coco-sp | GatedGCN-HSG | macro F1: 0.3535±0.0032 |
| node-classification-on-pascalvoc-sp-1 | GatedGCN-HSG | macro F1: 0.4604±0.0059 |