3 个月前

GraphRARE:基于相对熵增强的强化学习图神经网络

GraphRARE:基于相对熵增强的强化学习图神经网络

摘要

图神经网络(Graph Neural Networks, GNNs)在基于图的分析任务中展现出显著优势。然而,现有大多数方法均基于同质性假设,在异质性图(heterophilic graphs)上表现不佳。在异质性图中,相连节点的特征差异较大且类别标签不同,语义相关的节点可能相隔多跳。为解决这一局限性,本文提出GraphRARE——一种基于节点相对熵与深度强化学习的通用框架,旨在增强GNN的表达能力。该框架引入一种创新的节点相对熵度量方法,综合考虑节点特征与结构相似性,用于衡量节点对之间的互信息。此外,为避免远距离节点带来的有用信息与噪声混合所导致的次优解问题,本文设计了一种基于深度强化学习的图拓扑优化算法。该算法依据定义的节点相对熵,智能选择具有信息量的节点,同时剔除噪声节点。在七个真实世界数据集上开展的大量实验表明,GraphRARE在节点分类任务中具有显著优势,并具备优化原始图拓扑结构的能力。

基准测试

基准方法指标
node-classification-on-chameleonGraphSAGE-RARE (λ=1.0)
Accuracy: 69.28±1.90
node-classification-on-cornellH2GCN-RARE (λ=1.0)
Accuracy: 87.84±4.05
node-classification-on-squirrelGCN-RARE (λ=1.0)
Accuracy: 55.90±1.39
node-classification-on-texasH2GCN-RARE (λ=1.0)
Accuracy: 86.76±5.80
node-classification-on-wisconsinH2GCN-RARE (λ=1.0)
Accuracy: 90.00±2.97

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
GraphRARE:基于相对熵增强的强化学习图神经网络 | 论文 | HyperAI超神经