节点属性增强的图自编码器

作     者：张芳 王祺 刘彦北 ZHANG Fang;WANG Qi;LIU Yan-bei

作者机构：天津工业大学生命科学学院天津300387 天津工业大学电子与信息工程学院天津300387 

出 版 物：《天津工业大学学报》 (Journal of Tiangong University)

年 卷 期：2021年第40卷第5期

页      面：76-80页

学科分类：0810[工学-信息与通信工程] 08[工学] 081001[工学-通信与信息系统] 

基　　金：天津市科学技术与工程重大专项资助项目(17ZXSCSY00060) 天津市教委科研计划资助项目(2017KJ087) 

主　　题：图自编码器 结构重建 节点属性重建 

摘      要：针对传统图自编码器的解码方法忽略节点属性作用的问题,提出一种联合重建图结构和属性信息的节点属性增强的图自编码器(NEGAE)模型。模型在编码器部分,采用图卷积神经网络进行图节点数据的特征提取,获得其节点表示;在解码器部分,一方面采用内积方式对图结构进行重建,另一方面采用反卷积的方式对节点属性进行重建;最后,将结构信息和节点属性信息的重建误差融合到一个统一的损失函数中进行优化。在Cora、Citeseer、Pubmed数据集上的结果表明:该模型在链路预测任务中的ROC曲线下面积(AUC)分别达到91.19%、90.27%、96.69%;聚类任务中的聚类准确度(ACC)分别达到60.31%、50.60%、66.79%,说明NEGAE方法在各种学习任务上均取得了良好的性能。