通过信息瓶颈原理学习更鲁棒的图结构,提升节点分类效果
GaGSL: Global-augmented Graph Structure Learning via Graph Information Bottleneck
- 利用全局特征与结构增强生成新数据,缓解原始图噪声影响
- 结合信息瓶颈优化,学习最小必要且信息丰富的图结构
- 在多个数据集上优于现有方法,尤其适合噪声图数据
图神经网络在半监督节点分类任务中表现优异,但通常假设观测结构能准确反映真实节点关系。现实中图结构常存在噪声或缺失,导致表示质量下降。因此,学习一个兼具性能与鲁棒性的清洁图结构至关重要。本文提出一种基于图信息瓶颈(GIB)原则的新方法——全局增强图结构学习(GaGSL)。核心思想是学习紧凑且富有信息量的图结构。为减少仅依赖原始结构带来的偏差,首先通过全局特征增强和全局结构增强生成增强特征与增强结构;随后将二者输入具有不同参数的结构估计器,分别优化并重构图结构;最终融合重构结构形成最终图结构。最后基于互信息的GIB机制指导图结构优化,获得最小充分图结构。跨多个数据集的全面评估表明,相比当前最优方法,GaGSL展现出卓越性能与鲁棒性。
原文摘要 · Abstract (English)
Graph neural networks (GNNs) are prominent for their effectiveness in processing graph data for semi-supervised node classification tasks. Most works of GNNs assume that the observed structure accurately represents the underlying node relationships. However, the graph structure is inevitably noisy or incomplete in reality, which can degrade the quality of graph representations. Therefore, it is imperative to learn a clean graph structure that balances performance and robustness. In this paper, we propose a novel method named \textit{Global-augmented Graph Structure Learning} (GaGSL), guided by the Graph Information Bottleneck (GIB) principle. The key idea behind GaGSL is to learn a compact and informative graph structure for node classification tasks. Specifically, to mitigate the bias caused by relying solely on the original structure, we first obtain augmented features and augmented structure through global feature augmentation and global structure augmentation. We then input the augmented features and augmented structure into a structure estimator with different parameters for optimization and re-definition of the graph structure, respectively. The redefined structures are combined to form the final graph structure. Finally, we employ GIB based on mutual information to guide the optimization of the graph structure to obtain the minimum sufficient graph structure. Comprehensive evaluations across a range of datasets reveal the outstanding performance and robustness of GaGSL compared with the state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。