作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2020, Vol. 46 ›› Issue (10): 67-73. doi: 10.19678/j.issn.1000-3428.0055764

• 人工智能与模式识别 • 上一篇    下一篇

基于矩阵分解的属性网络表示学习

张潘1,2, 卢光跃1,2, 吕少卿1,2, 赵雪莉1,2   

  1. 1. 西安邮电大学 通信与信息工程学院, 西安 710121;
    2. 陕西省信息通信网络及安全重点实验室, 西安 710121
  • 收稿日期:2019-08-19 修回日期:2019-10-02 发布日期:2019-10-25
  • 作者简介:张潘(1991-),男,硕士研究生,主研方向为网络表示学习、数据挖掘;卢光跃,教授;吕少卿,讲师、博士;赵雪莉,硕士研究生。
  • 基金资助:
    陕西省教育厅科研计划项目(17JK0703)。

Attributed Network Representation Learning Based on Matrix Factorization

ZHANG Pan1,2, LU Guangyue1,2, Lü Shaoqing1,2, ZHAO Xueli1,2   

  1. 1. School of Communications and Information Engineering, Xi'an University of Posts and Telecommunications, Xi'an 710121, China;
    2. Shaanxi Provincial Key Laboratory of Information Communication Network and Security, Xi'an 710121, China
  • Received:2019-08-19 Revised:2019-10-02 Published:2019-10-25

摘要: 为融合网络拓扑结构与节点属性信息以提高网络表示学习质量,提出一种新的属性网络表示学习算法(ANEMF)。引入余弦相似性概念,定义网络二阶结构相似度矩阵和属性相似度矩阵,通过对网络结构相似度和属性相似度损失函数进行联合优化学习,并利用矩阵分解的形式实现网络拓扑结构与节点属性信息的融合,同时应用乘法更新规则计算得到节点表示向量。在3个公开数据集上的实验结果表明,与DeepWalk和TADW算法相比,ANEMF算法得到的节点表示向量能够保留网络拓扑结构与节点属性信息,有效提升其在节点分类任务中的综合性能。

关键词: 机器学习, 网络分析, 数据挖掘, 网络表示学习, 矩阵分解, 网络嵌入

Abstract: To combine the information of network topological structure and node attribute to improve the quality of network representation learning,this paper proposes a new attributed network representation learning algorithm,named ANEMF.The algorithm introduces the idea of cosine similarity to define the second-order structural similarity matrix and the attribute similarity matrix of the network.Through the cooperative optimized learning of network structure similarity and attribute similarity functions,the information of network topological structure and node attribute is fused in the form of matrix factorization.Finally,the node representation vectors are obtained through the multiplication update rules.Experimental results on three public datasets show that compared with DeepWalk and TADW algorithms,the proposed algorithm can keep the information of network topological structure and node attribute in obtained node representation vectors.It can significantly improve the overall performance in the node classification tasks.

Key words: machine learning, network analysis, data mining, network representation learning, matrix factorization, network embedding

中图分类号: