3 个月前

NodeNet:一种用于节点分类的图正则化神经网络

NodeNet:一种用于节点分类的图正则化神经网络

摘要

现实世界中的事件表现出高度的相互依赖性和关联性,因此所产生的数据点也继承了这些关联关系。然而,大多数人工智能/机器学习(AI/ML)技术在建模时忽略了数据点之间的这种关联。近年来,基于图的AI/ML技术受到广泛关注,其核心目标正是充分利用数据之间的关联性。图-based学习算法能够有效利用数据及其相关关系,构建性能更优的模型。神经图学习(Neural Graph Learning, NGL)便是其中一种代表性方法,它通过在传统机器学习算法中引入改进的损失函数,以充分利用图结构中的边信息。本文提出一种基于NGL的模型——NodeNet,用于解决引文图中的节点分类任务。我们详细阐述了模型的改进之处,并讨论了这些改进与任务需求之间的关联性。此外,我们将实验结果与当前最先进的方法进行了对比,并深入分析了NodeNet表现出优越性能的原因。

基准测试

基准方法指标
node-classification-on-citeseerNodeNet
Accuracy: 80.09%
node-classification-on-coraNodeNet
Accuracy: 86.80%
node-classification-on-pubmedNodeNet
Accuracy: 90.21%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
NodeNet:一种用于节点分类的图正则化神经网络 | 论文 | HyperAI超神经