基于融合变分图注意自编码器的深度聚类模型

• Published in: 计算机科学 Vol. 48; no. z2; pp. 81 - 116
• Main Authors: 康雁, 寇勇奇, 谢思宇, 王飞, 张兰, 吴志伟, 李浩
• Format: Journal Article
• Language: Chinese
• Published: 云南大学软件学院 昆明 650504 2021
• Subjects: 深度聚类;表征学习;自编码器;变分图注意自编码器;自监督聚类
• ISSN: 1002-137X
• DOI: 10.11896/jsjkx.210300036

TP181; 聚类作为数据挖掘和机器学习中最基本的任务之一,在各种现实世界任务中已得到广泛应用.随着深度学习的发展,深度聚类成为一个研究热点.现有的深度聚类算法主要从节点表征学习或者结构表征学习两个方面入手,较少考虑同时将这两种信息进行融合以完成表征学习.提出一种融合变分图注意自编码器的深度聚类模型FVGTAEDC(Deep Clustering Model Based on Fusion Varitional Graph Attention Self-encoder),此模型通过联合自编码器和变分图注意自编码器进行聚类,模型中自编码器将变分图注意自编码器从网络中学习(低阶和高阶)结构表示进行集成,随后从原始数据中学习特征表示.在两个模块训练的同时,为了适应聚类任务,将自编码器模块融合节点和结构信息的表示特征进行自监督聚类训练.通过综合聚类损失、自编码器重构数据损失、变分图注意自编码器重构邻接矩阵损失、后验概率分布与先验概率分布相对熵损失,该模型可以有效聚合节点的属性和网络的结构,同时优化聚类标签分配和学习适合于聚类的表示特征.综合实验证明,该方法在5个现实数据集上的聚类效果均优于当前先进的深度聚类方法.