arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.00643cs.LGcs.AI

谱增强用于图上的自监督学习

Spectral Augmentation for Self-Supervised Learning on Graphs

The Pennsylvania State University(宾夕法尼亚州立大学);University of Virginia(弗吉尼亚大学)

Lu Lin, Jinghui Chen, Hongning Wang

AI总结:

本文从谱域视角提出谱增强方法,通过最大化谱变化指导拓扑增强,提升图对比学习在自监督表示学习、迁移学习及对抗鲁棒性上的性能。

摘要:

图对比学习(GCL)作为一种新兴的图自监督学习技术,旨在通过实例判别来学习表示。其性能在很大程度上依赖于图增强,以反映对小扰动鲁棒的不变模式;然而,关于GCL应捕获何种图不变性仍不清楚。近期研究主要在空间域中以均匀随机方式进行拓扑增强,忽略了其对谱域中固有结构性质的影响。在本工作中,我们旨在通过从谱角度探索图的不变性,找到一种有原则的拓扑增强方式。我们开发了谱增强方法,通过最大化谱变化来指导拓扑增强。在图和节点分类任务上的大量实验证明了我们的方法在自监督表示学习中的有效性。所提出的方法在迁移学习中也带来了有前景的泛化能力,并在对抗攻击下具有引人注目的鲁棒性。我们的研究为图拓扑增强揭示了一个通用原则。

英文摘要:

Graph contrastive learning (GCL), as an emerging self-supervised learning technique on graphs, aims to learn representations via instance discrimination. Its performance heavily relies on graph augmentation to reflect invariant patterns that are robust to small perturbations; yet it still remains unclear about what graph invariance GCL should capture. Recent studies mainly perform topology augmentations in a uniformly random manner in the spatial domain, ignoring its influence on the intrinsic structural properties embedded in the spectral domain. In this work, we aim to find a principled way for topology augmentations by exploring the invariance of graphs from the spectral perspective. We develop spectral augmentation which guides topology augmentations by maximizing the spectral change. Extensive experiments on both graph and node classification tasks demonstrate the effectiveness of our method in self-supervised representation learning. The proposed method also brings promising generalization capability in transfer learning, and is equipped with intriguing robustness property under adversarial attacks. Our study sheds light on a general principle for graph topology augmentation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑