国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:张元钧, 张曦煌
单位:江南大学 人工智能与计算机学院, 江苏 无锡 214002
关键词:链路预测,动态网络,降噪自编码器,图卷积网络,长短期记忆网络
基金:国家自然科学基金资助项目(61673193);中国博士后科学基金资助项目(2017M621625);江苏省产学研合作项目(BY2015019-30)。
针对动态网络节点之间链路预测的准确率低和运行时间长的情况,提出了一种以降噪自编码器(dAE)为框架,结合图卷积网络(GCN)和长短期记忆(LSTM)网络的动态网络表示学习模型dynGAELSTM。首先,该模型的前端采用GCN捕获动态图节点的高阶图邻域的特征信息;其次,将提取到的信息输入dAE的编码层以获取低维特征向量,并在LSTM网络上获取动态网络的时空依赖特征;最后,经dAE的解码层重建预测图,并与真实图对比来构建损失函数,从而优化模型完成链路预测。理论分析和仿真实验表明,dynGAELSTM模型相较于预测性能第二的模型在三个数据集上的预测性能分别提升了0.79、1.19和3.13个百分点,模型的运行时间降低了0.92%和1.73%。可见dynGAELSTM模型在链路预测任务中相较于现有模型精度提升,复杂度降低。
来源:2021年第7期
《计算机应用》期刊编辑部