arXiv:2511.08287cs.LGcs.AI2025-11AAAI被引 1

提出双核图社区对比学习,提升大图训练效率与效果

Dual-Kernel Graph Community Contrastive Learning

  • 将图转化为节点集合网络,降低计算复杂度
  • 线性复杂度损失函数,有效捕捉图层级结构
  • 结合知识蒸馏加速推理,适合大规模图应用

图对比学习(GCL)已成为无标签场景下训练图神经网络的有力方法。然而,其在大规模图上的可扩展性受限于GNN的密集消息传递机制以及正负节点对间对比损失的二次计算复杂度。为此,我们提出一种高效GCL框架,将输入图转换为保留社区结构信息的紧凑节点集合网络。首先引入具有线性复杂度的核化图社区对比损失,实现节点集合间有效信息传递,以捕捉图的层级结构;随后在解耦的GNN架构中引入知识蒸馏技术,加速推理同时保持强泛化能力。在十六个不同规模的真实世界数据集上的实验表明,该方法在有效性与可扩展性上均优于当前最优的GCL基线。

原文摘要 · Abstract (English)

Graph Contrastive Learning (GCL) has emerged as a powerful paradigm for training Graph Neural Networks (GNNs) in the absence of task-specific labels. However, its scalability on large-scale graphs is hindered by the intensive message passing mechanism of GNN and the quadratic computational complexity of contrastive loss over positive and negative node pairs. To address these issues, we propose an efficient GCL framework that transforms the input graph into a compact network of interconnected node sets while preserving structural information across communities. We firstly introduce a kernelized graph community contrastive loss with linear complexity, enabling effective information transfer among node sets to capture hierarchical structural information of the graph. We then incorporate a knowledge distillation technique into the decoupled GNN architecture to accelerate inference while maintaining strong generalization performance. Extensive experiments on sixteen real-world datasets of varying scales demonstrate that our method outperforms state-of-the-art GCL baselines in both effectiveness and scalability.

图对比学习社区检测GNN加速知识蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。