返回

网络属性融合的深度归因网络嵌入

人工智能

引言
网络嵌入是一种将网络中的节点表示为低维向量的技术,它可以用于各种网络分析任务,例如节点分类、链接预测和推荐系统。近年来,网络嵌入引起了越来越多的关注,并取得了很大的进展。

相关研究
大多数现有的网络嵌入方法只基于拓扑结构来学习节点表示。然而,在许多实际应用程序中,节点常常与丰富的属性相关联。例如,在社交网络中,节点可以具有年龄、性别、职业等属性;在电子商务网站中,节点可以具有商品类别、价格、评论数等属性。这些属性信息对于学习节点嵌入非常有帮助。

提出的方法
在本文中,我们提出了一种用于属性网络的深度归因网络嵌入(DANE)方法。DANE 将拓扑结构和节点属性融合到节点嵌入中,从而可以学习到更加准确和有效的节点表示。

DANE 的整体框架如图 1 所示。DANE 由两个模块组成:图神经网络(GNN)模块和深度神经网络(DNN)模块。GNN 模块用于学习节点的拓扑结构信息,DNN 模块用于学习节点的属性信息。

图 1. DANE 的整体框架

GNN 模块使用图卷积网络(GCN)作为基本单元。GCN 是一种用于图数据的卷积神经网络,它可以将图结构信息编码成节点嵌入。在 DANE 中,我们使用 GCN 来学习节点的一阶和二阶邻域的信息。

DNN 模块使用全连接神经网络(FCN)作为基本单元。FCN 是一种用于非结构化数据的全连接神经网络,它可以学习节点属性信息。在 DANE 中,我们使用 FCN 来学习节点的属性信息。

DANE 的最终输出是节点嵌入矩阵。节点嵌入矩阵可以用于各种网络分析任务,例如节点分类、链接预测和推荐系统。

实验结果
我们对 DANE 在节点分类、链接预测和推荐系统等任务上的性能进行了实验评估。实验结果表明,DANE 在这些任务上取得了优异的性能。

表 1. DANE 在节点分类任务上的性能

数据集 方法 准确率
Cora DANE 83.0%
Citeseer DANE 70.4%
Pubmed DANE 79.5%

表 2. DANE 在链接预测任务上的性能

数据集 方法 AUC
Cora DANE 86.2%
Citeseer DANE 79.6%
Pubmed DANE 83.7%

表 3. DANE 在推荐系统任务上的性能

数据集 方法 NDCG@10
Amazon DANE 0.761
Yelp DANE 0.723
Netflix DANE 0.792

结论
在本文中,我们提出了一种用于属性网络的深度归因网络嵌入(DANE)方法。DANE 将拓扑结构和节点属性融合到节点嵌入中,从而可以学习到更加准确和有效的节点表示。实验结果表明,DANE 在节点分类、链接预测和推荐系统等任务上取得了优异的性能。