计算机工程与设计2024,Vol.45Issue(9) :2805-2811.DOI:10.16208/j.issn1000-7024.2024.09.033

基于图自编码器多尺度特征的自监督群体发现

Self-supervised community detection based on graph auto-encoders with multi-scale features

沈国栋 汪晓锋 毛岱波 王栽胜 张增杰 全大英
计算机工程与设计2024,Vol.45Issue(9) :2805-2811.DOI:10.16208/j.issn1000-7024.2024.09.033

基于图自编码器多尺度特征的自监督群体发现

Self-supervised community detection based on graph auto-encoders with multi-scale features

沈国栋 1汪晓锋 1毛岱波 2王栽胜 1张增杰 1全大英1
扫码查看

作者信息

  • 1. 中国计量大学信息工程学院,浙江杭州 310000
  • 2. 宁波市科技信息与发展战略研究院信息技术部,浙江宁波 315000
  • 折叠

摘要

现有基于图自编码器的群体发现方法通常忽略了编码层多尺度特征对群体发现的影响,同时由于缺少统一的优化目标函数导致次优结果.为此,提出一种基于图自编码器多尺度特征融合的自监督群体发现方法.在图自编码器的基础上引入一种多尺度自表达模块,从不同编码层获取具有区分性的节点关系矩阵表示,并与节点潜在表示进行融合;通过节点聚类模块获得初步的群体识别结果;引入一种自监督模块监督节点表示学习过程,获得更优结果,构建一种端对端的网络群体发现模型.在多个公开数据集上进行对比实验,验证了所提方法的有效性,与现有方法相比,其在群体识别准确度上有了明显提升.

Abstract

Existing methods based on graph autoencoder(GAE)usually ignore the influence of multi-scale feature information of encoding layers on community detection,and get sub-optimal results due to the lack of a unified optimization objective function.To this end,a self-supervised method for community detection based on GAE with multi-scale features was presented.A multi-scale self-expression module was introduced,discriminative node representations were obtained from different encoding layers and fused.A node clustering module was introduced to obtain rough clustering results.At the same time,a self-supervised node rep-resentation learning process was introduced to achieve better results,thereby constructing an end-to-end network group discovery model.Through comparative experiments on multiple datasets,the effectiveness of the proposed method is verified,and the accuracy of community detection is significantly improved compared with that of the existing methods.

关键词

图自编码器/群体发现/多尺度特征/自监督学习/特征融合/端到端/统一优化

Key words

graph auto-encoder/community detection/multi-scale features/self-supervised learning/feature fusion/end-to-end/unified optimization

引用本文复制引用

基金项目

国家重点研发计划基金项目(2019YFB1707104)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
参考文献量3
段落导航相关论文