Jun, 2024

ProG:图灵提示学习基准

TL;DR人工智能在图上的通用智能在各种应用中取得了显著的进展,然而传统的 “预训练和微调” 范式在复杂和小样本场景下存在低效和负迁移问题。图提示学习作为一种有希望的替代方案出现,利用轻量级提示来操作数据,通过重新构思下游任务来填补任务间隙。然而,仍存在几个关键挑战:如何统一各种图提示模型,如何评估图提示的质量,以及提高其实际应用中的可用性和选择性。为了应对这些挑战,我们推出了图提示学习的首个全面基准。我们的基准集成了六种预训练方法和五种最先进的图提示技术,在十五个多样化的数据集上进行评估,以评估性能、灵活性和效率。我们还介绍了一个名为 'ProG' 的易于使用的开源库,简化了各种图提示模型的执行,有助于客观评估。此外,我们提出了一个统一的框架,将现有的图提示方法分为两个主要方法:图提示和标记提示。该框架增强了图提示技术的适用性和比较性。代码可在此 https URL 中获取。