arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2205.09802cs.CVcs.AIcs.LG
收藏

半监督图分类中的标签不变增强算法

Label-invariant Augmentation for Semi-Supervised Graph Classification

Brandeis University(布兰迪斯大学)

Han Yue, Chunhui Zhang, Chuxu Zhang, Hongfu Liu

摘要:

近年来,基于对比度的增强技术在计算机视觉领域掀起了一个新的高潮,旋转、裁剪、翻转等操作与专用算法相结合,极大地提高了模型的泛化能力和鲁棒性。在这种趋势下,一些开创性的尝试将类似的想法用于图形数据。然而,与图像不同,在不改变图的性质的情况下设计合理的增强要困难得多。虽然令人兴奋,但目前的图形对比学习并没有达到视觉对比学习那样有前途的性能。我们推测,图对比学习的当前性能可能受到违反标签不变扩增假设的限制。有鉴于此,我们提出了一个标签不变的增强图结构的数据,以解决这一挑战。与节点/边修改和子图提取不同,我们在表示空间中进行增强,并在最困难的方向生成增强样本,同时保持增强数据的标签与原始样本相同。在半监督的情况下,我们证明了我们提出的方法优于经典的图神经网络的方法和最近的图对比学习的8个基准图结构的数据,其次是几个深入的实验,以进一步探索标签不变的增强在几个方面。

英文摘要:

Recently, contrastiveness-based augmentation surges a new climax in the computer vision domain, where some operations, including rotation, crop, and flip, combined with dedicated algorithms, dramatically increase the model generalization and robustness. Following this trend, some pioneering attempts employ the similar idea to graph data. Nevertheless, unlike images, it is much more difficult to design reasonable augmentations without changing the nature of graphs. Although exciting, the current graph contrastive learning does not achieve as promising performance as visual contrastive learning. We conjecture the current performance of graph contrastive learning might be limited by the violation of the label-invariant augmentation assumption. In light of this, we propose a label-invariant augmentation for graph-structured data to address this challenge. Different from the node/edge modification and subgraph extraction, we conduct the augmentation in the representation space and generate the augmented samples in the most difficult direction while keeping the label of augmented data the same as the original samples. In the semi-supervised scenario, we demonstrate our proposed method outperforms the classical graph neural network based methods and recent graph contrastive learning on eight benchmark graph-structured data, followed by several in-depth experiments to further explore the label-invariant augmentation in several aspects.

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

↑