Jul, 2022

SizeShiftReg: 一种用于提高图神经网络尺寸泛化能力的正则化方法

TL;DR本研究提出一种基于规则化策略的方法,能够在只有训练数据而没有测试数据的情况下,提高图神经网络在更大规模的图上的泛化能力,从而改善图分类的性能。实验结果表明,流行的 GNN 模型在我们的规则化策略下,在 50% 规模最小的图上进行训练,在 10% 规模最大的图上进行测试时,性能提高了达到 30%。