4 个月前

基于层次支持图的下一代消息传递

基于层次支持图的下一代消息传递

摘要

消息传递神经网络(MPNNs)在图学习任务中得到了广泛应用,但受限于信息交换范围狭窄的问题,即在每轮消息传递过程中仅限于邻近节点之间的信息交换。为了解决这些限制,已提出了多种策略,包括引入虚拟节点以促进全局信息交换。在本研究中,我们提出了一种层次支持图(Hierarchical Support Graph, HSG),这是通过递归粗化原始图而扩展的虚拟节点概念。该方法提供了一个灵活的框架,用于增强图中的信息流动,且独立于所使用的具体MPNN层。我们对HSG进行了理论分析,探讨了其经验性能,并展示了HSG能够超越其他使用虚拟节点的方法,在多个数据集上取得了最先进的结果。

代码仓库

基准测试

基准方法指标
graph-classification-on-peptides-funcGatedGCN-HSG
AP: 0.6866±0.0038
graph-property-prediction-on-ogbg-molpcbaGatedGCN-HSG
Test AP: 0.3129±0.0020
graph-regression-on-peptides-structGatedGCN-HSG
MAE: 0.2421±0.0007
node-classification-on-coco-spGatedGCN-HSG
macro F1: 0.3535±0.0032
node-classification-on-pascalvoc-sp-1GatedGCN-HSG
macro F1: 0.4604±0.0059

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于层次支持图的下一代消息传递 | 论文 | HyperAI超神经