School of Software, Dalian University of Technology, China; Key Laboratory for Ubiquitous Network and Service Software of Liaoning Province, China.
Neural Netw. 2022 Aug;152:224-233. doi: 10.1016/j.neunet.2022.04.018. Epub 2022 Apr 20.
Attributed graph clustering is challenging as it needs to effectively combine both graph structure and node feature information to accomplish node clustering. Recent studies mostly adopt graph neural networks to learn node embeddings, then apply traditional clustering methods to obtain clusters. However, their node embeddings are not specifically designed for clustering. Moreover, most of their loss functions only rely on either structure or feature information, making both kinds of information not fully retained in node embeddings. In this paper, we propose a multi-task embedding learning method (MTEL) for attributed graph clustering, which constructs two prediction tasks in terms of structure and feature based adjacency matrices respectively. To make the node embeddings helpful for the downstream clustering, in each task, we predict the minimum hop number between each pair of nodes in the adjacency matrix, so that the correlation degrees among nodes can be encoded into node embeddings. To improve the performance of the prediction task, we regularize the model parameters in these two tasks via ℓ norm, through which the model parameters can be jointly learned. Experiments on real attributed graphs show that MTEL is superior for attributed graph clustering over state-of-the-art methods.
归因图聚类具有挑战性,因为它需要有效地结合图结构和节点特征信息来完成节点聚类。最近的研究大多采用图神经网络来学习节点嵌入,然后应用传统的聚类方法来获得聚类。然而,它们的节点嵌入并不是专门为聚类设计的。此外,它们的大多数损失函数仅依赖于结构或特征信息,使得这两种信息在节点嵌入中都没有得到充分保留。在本文中,我们提出了一种用于归因图聚类的多任务嵌入学习方法 (MTEL),该方法分别基于结构和特征构建了两个预测任务。为了使节点嵌入有助于下游聚类,在每个任务中,我们预测邻接矩阵中每对节点之间的最小跳数,以便将节点之间的相关度编码到节点嵌入中。为了提高预测任务的性能,我们通过 ℓ 范数正则化这两个任务中的模型参数,通过这种方式可以共同学习模型参数。在真实的归因图上的实验表明,MTEL 优于最先进的归因图聚类方法。