arXiv:2505.13650cs.LGcs.AI2025-05

用自增强机制动态筛选高质量图对比样本,提升表示学习效果

Self-Reinforced Graph Contrastive Learning

  • 用模型自身编码器动态评估并选择优质正样本对
  • 在多个图分类任务上超越现有最优GCL方法
  • 适合需要高质量图表示的科研与工业应用

图在社交网络、分子生物学和知识图谱等众多实际场景中广泛存在,能够捕捉实体间的复杂关系。图对比学习(GCL)通过对比正负样本对,实现自监督的鲁棒图表示学习,但其性能受限于正样本对的质量。本文提出SRGCL(自增强图对比学习),利用模型自身编码器动态评估并选择高质量正样本对。设计了融合多种增强策略的统一正样本生成器,并基于流形假设引导的选择器,以保持潜在空间的几何结构。通过概率化机制选择正样本对,使模型在迭代过程中持续优化对样本质量的判断。在多样化的图级分类任务上的实验表明,作为即插即用模块,SRGCL始终优于当前最优的GCL方法,展现出跨领域的适应性与有效性。

原文摘要 · Abstract (English)

Graphs serve as versatile data structures in numerous real-world domains-including social networks, molecular biology, and knowledge graphs-by capturing intricate relational information among entities. Among graph-based learning techniques, Graph Contrastive Learning (GCL) has gained significant attention for its ability to derive robust, self-supervised graph representations through the contrasting of positive and negative sample pairs. However, a critical challenge lies in ensuring high-quality positive pairs so that the intrinsic semantic and structural properties of the original graph are preserved rather than distorted. To address this issue, we propose SRGCL (Self-Reinforced Graph Contrastive Learning), a novel framework that leverages the model's own encoder to dynamically evaluate and select high-quality positive pairs. We designed a unified positive pair generator employing multiple augmentation strategies, and a selector guided by the manifold hypothesis to maintain the underlying geometry of the latent space. By adopting a probabilistic mechanism for selecting positive pairs, SRGCL iteratively refines its assessment of pair quality as the encoder's representational power improves. Extensive experiments on diverse graph-level classification tasks demonstrate that SRGCL, as a plug-in module, consistently outperforms state-of-the-art GCL methods, underscoring its adaptability and efficacy across various domains.

图对比学习自增强表示学习

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