arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.09239cs.LGcs.IR
收藏

使用基于最近邻的困难负样本进行时序图神经网络的鲁棒训练

Robust Training of Temporal GNNs using Nearest Neighbours based Hard Negatives

Indian Institute of Technology, Delhi(印度理工学院德里分校)

Shubham Gupta, Srikanta Bedathur

AI总结:

针对时序图神经网络因均匀随机负采样导致冗余和次优性能的问题,提出基于重要性的动态负采样方法替代均匀采样,在三个真实数据集上实现了更优的链路预测性能。

摘要:

时序图神经网络(TGNN)在未来链路预测任务中展现出了最先进的性能。这些 TGNN 的训练通过基于均匀随机采样的无监督损失来执行。在训练过程中,在正样本的上下文中,损失是在无信息量的负样本上计算的,这引入了冗余和次优的性能。在本文中,我们提出了 TGNN 的改进无监督学习,通过用基于重要性的负采样替换均匀负采样。我们在理论上激发并定义了用于负样本采样的动态计算分布。最后,通过对三个真实世界数据集的实证评估,我们表明使用基于所提出的负采样的损失训练的 TGNN 提供了一致的卓越性能。

英文摘要:

Temporal graph neural networks Tgnn have exhibited state-of-art performance in future-link prediction tasks. Training of these TGNNs is enumerated by uniform random sampling based unsupervised loss. During training, in the context of a positive example, the loss is computed over uninformative negatives, which introduces redundancy and sub-optimal performance. In this paper, we propose modified unsupervised learning of Tgnn, by replacing the uniform negative sampling with importance-based negative sampling. We theoretically motivate and define the dynamically computed distribution for a sampling of negative examples. Finally, using empirical evaluations over three real-world datasets, we show that Tgnn trained using loss based on proposed negative sampling provides consistent superior performance.

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

↑