3 个月前

CensNet:图神经网络中的卷积与边-节点切换

CensNet:图神经网络中的卷积与边-节点切换

摘要

本文提出了一种名为CensNet(Convolution with Edge-Node Switching Graph Neural Network)的图神经网络模型,用于具有节点和边特征的图结构数据的半监督分类与回归任务。CensNet是一种通用的图嵌入框架,能够将节点和边共同映射至潜在特征空间。通过利用原始无向图的线图(line graph)结构,实现节点与边角色的互换,并据此提出了两种新颖的图卷积操作,以支持特征传播。在真实世界中的学术引用网络与量子化学图数据集上的实验结果表明,所提出方法在性能上达到了或超越了当前最先进的水平。

基准测试

基准方法指标
graph-regression-on-lipophilicityRandom Forests
RMSE@80%Train: 1.16
graph-regression-on-lipophilicityCensNet
RMSE@80%Train: 0.93
graph-regression-on-lipophilicityLogistic Regression
RMSE@80%Train: 1.15
graph-regression-on-tox21Logistic Regression
AUC@80%Train: 0.71
graph-regression-on-tox21Random Forest
AUC@80%Train: 0.71
graph-regression-on-tox21CensNet
AUC@80%Train: 0.78

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CensNet:图神经网络中的卷积与边-节点切换 | 论文 | HyperAI超神经