arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2210.03561cs.LGcs.AIstat.ML
收藏

基于测试时图变换的图表示学习

Empowering Graph Representation Learning with Test-Time Graph Transformation

Michigan State University(密歇根州立大学);Snap Inc.(Snap公司)

Wei Jin, Tong Zhao, Jiayuan Ding, Yozen Liu, Jiliang Tang, Neil Shah

摘要:

作为图上表示学习的有力工具,图神经网络(GNN)促进了从药物发现到推荐系统的各种应用。然而,GNN的有效性受到与数据质量相关的问题的极大挑战,例如分布偏移,异常特征和对抗攻击。最近的努力已经从建模的角度来解决这些问题,这需要改变模型架构或重新训练模型参数的额外成本。在这项工作中,我们提供了一个以数据为中心的视图来解决这些问题,并提出了一个名为GTrans的图转换框架,该框架在测试时适应和细化图数据,以实现更好的性能。我们提供的框架设计的理论分析,并讨论为什么适应图形数据比适应模型更好。大量的实验已经证明了GTrans在三个不同的场景下的有效性,八个基准数据集,其中次优数据。值得注意的是,GTrans在大多数情况下表现最好,在三个实验设置中比最佳基线提高了2.8%,8.2%和3.8%。代码发布于https://github.com/ChandlerBang/GTrans。

英文摘要:

As powerful tools for representation learning on graphs, graph neural networks (GNNs) have facilitated various applications from drug discovery to recommender systems. Nevertheless, the effectiveness of GNNs is immensely challenged by issues related to data quality, such as distribution shift, abnormal features and adversarial attacks. Recent efforts have been made on tackling these issues from a modeling perspective which requires additional cost of changing model architectures or re-training model parameters. In this work, we provide a data-centric view to tackle these issues and propose a graph transformation framework named GTrans which adapts and refines graph data at test time to achieve better performance. We provide theoretical analysis on the design of the framework and discuss why adapting graph data works better than adapting the model. Extensive experiments have demonstrated the effectiveness of GTrans on three distinct scenarios for eight benchmark datasets where suboptimal data is presented. Remarkably, GTrans performs the best in most cases with improvements up to 2.8%, 8.2% and 3.8% over the best baselines on three experimental settings. Code is released at https://github.com/ChandlerBang/GTrans.

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

↑