首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于Attention机制的链接预测算法
引用本文:程华,张林,房一泉.基于Attention机制的链接预测算法[J].华中科技大学学报(自然科学版),2019(2):109-114.
作者姓名:程华  张林  房一泉
作者单位:华东理工大学信息科学与工程学院
摘    要:针对社会网络中链接预测问题,提出了基于注意力(Attention)机制的链接表示及其预测算法.基于待预测节点的共邻关系构建其链接局部网络,设计了基于紧密游走的网络拓扑序列化方法.采用双向循环神经网络(Bi-RNN)对链接序列进行向量编码,以充分挖掘序列相关节点间的上下文依赖信息.通过Attention机制对链接中的节点进行关注和加权,强化重要节点对链接预测任务的贡献,实现链接拓扑特征的自动提取与准确分类预测.实验结果表明,在4种不同类型的社会网络数据集中,该算法的准确率和运算效率都有较大提高且普适性较强.

关 键 词:注意力机制  双向循环神经网络  拓扑序列化  链接预测  局部网络

Link prediction algorithm based on attention mechanism
Abstract:
Keywords:
本文献已被 CNKI 等数据库收录!
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号