提出新图卷积网络,提升多标签节点分类准确性
Correlation-Aware Graph Convolutional Networks for Multi-Label Node Classification
- 通过关联感知分解模块构建每标签的关联图
- 在消息传递中显式建模标签间相关性,提升分类效果
- 适合处理具有复杂标签关系的图数据任务
多标签节点分类是图挖掘中重要但研究不足的领域,因现实世界中许多节点属于多个类别而非单一类别。尽管已有工作利用图卷积网络(GCNs)学习节点表示并建模嵌入空间中的标签相关性,但仍受多重标签带来的特征模糊和拓扑模糊影响,削弱了图中信息传递的可信度,并忽略了图数据上的标签相关性。因此,减少模糊性并增强GCN的分类能力至关重要。然而,这极具挑战性,因需同时保持各标签的独特性并充分利用标签间的相关性。为此,本文提出关联感知图卷积网络(CorGCN)。通过引入新颖的关联感知图分解模块,CorGCN可为每个标签学习包含丰富标签相关性信息的图;随后采用关联增强图卷积,在消息传递过程中建模标签间关系以进一步提升分类性能。在五个数据集上的大量实验验证了所提CorGCN的有效性。
原文摘要 · Abstract (English)
Multi-label node classification is an important yet under-explored domain in graph mining as many real-world nodes belong to multiple categories rather than just a single one. Although a few efforts have been made by utilizing Graph Convolution Networks (GCNs) to learn node representations and model correlations between multiple labels in the embedding space, they still suffer from the ambiguous feature and ambiguous topology induced by multiple labels, which reduces the credibility of the messages delivered in graphs and overlooks the label correlations on graph data. Therefore, it is crucial to reduce the ambiguity and empower the GCNs for accurate classification. However, this is quite challenging due to the requirement of retaining the distinctiveness of each label while fully harnessing the correlation between labels simultaneously. To address these issues, in this paper, we propose a Correlation-aware Graph Convolutional Network (CorGCN) for multi-label node classification. By introducing a novel Correlation-Aware Graph Decomposition module, CorGCN can learn a graph that contains rich label-correlated information for each label. It then employs a Correlation-Enhanced Graph Convolution to model the relationships between labels during message passing to further bolster the classification process. Extensive experiments on five datasets demonstrate the effectiveness of our proposed CorGCN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。