提出解释保持的图数据增强,提升半监督图学习效果
Explanation-Preserving Augmentation for Semi-Supervised Graph Representation Learning
- 用图解释器提取关键子图,生成保留语义的增强数据
- 在多个基准数据集上超越现有最优方法,性能显著提升
- 适合需要可解释性与高效学习的图神经网络研究者
自监督图表示学习通常对每张图生成一对增强图,使同一图的增强结果表示相似,不同图的表示相异。然而,有效增强需兼顾语义保持与数据扰动,多数现有方法仅关注数据扰动,导致性能受限。本文提出解释保持增强(EPA),利用图解释器实现语义保持:先用少量标签训练图解释器,识别解释图标签的关键子图;再以此生成语义保持的增强数据,用于提升自监督图学习。整个过程为半监督范式。理论分析与大量实验表明,EPA-GRL 在多个基准数据集上优于使用语义无关增强的现有最先进方法。代码已开源。
原文摘要 · Abstract (English)
Self-supervised graph representation learning (GRL) typically generates paired graph augmentations from each graph to infer similar representations for augmentations of the same graph, but distinguishable representations for different graphs. While effective augmentation requires both semantics-preservation and data-perturbation, most existing GRL methods focus solely on data-perturbation, leading to suboptimal solutions. To fill the gap, in this paper, we propose a novel method, Explanation-Preserving Augmentation (EPA), which leverages graph explanation for semantics-preservation. EPA first uses a small number of labels to train a graph explainer, which infers the subgraphs that explain the graph's label. Then these explanations are used for generating semantics-preserving augmentations for boosting self-supervised GRL. Thus, the entire process, namely EPA-GRL, is semi-supervised. We demonstrate theoretically, using an analytical example, and through extensive experiments on a variety of benchmark datasets, that EPA-GRL outperforms the state-of-the-art (SOTA) GRL methods that use semantics-agnostic augmentations. The code is available at https://github.com/realMoana/EPA-GRL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。