融合GCN、自编码器与图Transformer,提升复杂图数据聚类效果
Tri-Learn Graph Fusion Network for Attributed Graph Clustering
- 三模块协同学习,通过特征融合增强全局与局部信息
- 在ACM、Reuters、USPS数据集上分别提升0.87%、14.14%、7.58%准确率
- 适合处理异构图数据,可用于新闻分类与主题检索
近年来,基于图卷积网络(GCN)的模型在图数据分析领域取得显著进展。然而,在处理大规模复杂图数据时,仍面临过平滑和过压缩等问题,导致聚类质量下降。尽管图Transformer架构缓解了部分问题,但在异构图数据上的表现仍受限。为此,本文提出一种新型深度聚类框架——三学习图融合网络(Tri-GFN),融合GCN、自编码器(AE)与图Transformer。该框架通过独特的三学习机制与特征融合增强策略,提升全局与局部信息的区分度与一致性。三个模块由三通道增强模块精细融合,充分挖掘节点属性与拓扑结构,确保鲁棒的聚类表示。三学习机制实现模块间相互学习,特征融合策略捕捉复杂关系,生成高度判别性表示。实验显示,其在ACM数据集上准确率提升约0.87%,在Reuters数据集上提升14.14%,在USPS数据集上提升7.58%。由于在Reuters数据集上的优异表现,Tri-GFN可应用于自动新闻分类、主题检索等场景。
原文摘要 · Abstract (English)
In recent years, models based on Graph Convolutional Networks (GCN) have made significant strides in the field of graph data analysis. However, challenges such as over-smoothing and over-compression remain when handling large-scale and complex graph datasets, leading to a decline in clustering quality. Although the Graph Transformer architecture has mitigated some of these issues, its performance is still limited when processing heterogeneous graph data. To address these challenges, this study proposes a novel deep clustering framework that comprising GCN, Autoencoder (AE), and Graph Transformer, termed the Tri-Learn Graph Fusion Network (Tri-GFN). This framework enhances the differentiation and consistency of global and local information through a unique tri-learning mechanism and feature fusion enhancement strategy. The framework integrates GCN, AE, and Graph Transformer modules. These components are meticulously fused by a triple-channel enhancement module, which maximizes the use of both node attributes and topological structures, ensuring robust clustering representation. The tri-learning mechanism allows mutual learning among these modules, while the feature fusion strategy enables the model to capture complex relationships, yielding highly discriminative representations for graph clustering. It surpasses many state-of-the-art methods, achieving an accuracy improvement of approximately 0.87% on the ACM dataset, 14.14 % on the Reuters dataset, and 7.58 % on the USPS dataset. Due to its outstanding performance on the Reuters dataset, Tri-GFN can be applied to automatic news classification, topic retrieval, and related fields.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。