李辉,赵宇红
1:内蒙古科技大学信息工程学院
2:内蒙古电力(集团)有限责任公司
摘要(Abstract):
注意力机制模型可通过自主学习,通过注意力聚焦学习节点的重要性,基于重要性为不同的节点赋予不同的权重,从而更充分且精确地提取节点特征向量.为了提高链路预测的准确性,提出一种基于双层注意力机制的链路预测方法(DAMLP).该方法设计了双层注意力机制,即节点级注意力机制可学习用户(项目)特征潜在的相关特征信息,聚合级注意力机制通过聚合相关特征信息,自动学习不同相关特征信息的重要性.通过双层注意力机制对相关特征信息的学习,使得链路预测更加准确.采取RMSE和MAE评价指标,通过与一些主流算法进行对比实验,在2个真实数据集的实验结果表明,所提方法有更好的链路预测效果.
关键词(KeyWords): 链路预测;注意力机制;特征信息;准确性
基金项目(Foundation):
作者(Author): 李辉,赵宇红
DOI: 10.16559/j.cnki.2095-2295.2023.03.014