arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.10380cs.LGcs.AIcs.CL
收藏

子图级通用提示微调

Subgraph-level Universal Prompt Tuning

Korea University(韩国大学);University of Illinois at Chicago(伊利诺伊大学香槟分校)

Junhyun Lee, Wooseong Yang, Jaewoo Kang

AI总结:

本文提出子图级通用提示微调方法,通过在子图层面分配提示特征,提升模型在不同预训练策略下的泛化能力,在42/45全场景实验中表现优于传统方法。

摘要:

在机器学习不断发展的背景下,通过提示微调来适应预训练模型已成为一个重要趋势。在图领域,多样化的预训练策略给开发有效的基于提示的微调方法带来了独特挑战。以往的方法局限于针对边预测预训练任务的专用提示函数,缺乏对不同预训练策略的泛化能力。最近,一种简单的提示微调方法被设计用于任何预训练策略,在输入图的特征空间内运行,理论上可以模拟任何类型的提示函数,从而显著提高了其在多种下游应用中的通用性。然而,这种简单提示能否充分捕捉图中复杂的上下文仍是一个开放问题,需要进一步研究。针对这一挑战,本文提出了子图级通用提示微调(SUPT)方法,专注于子图内的详细上下文。在SUPT中,提示特征在子图层面分配,保持了方法的通用能力。这需要比基于微调的方法少得多的调参参数,在42/45个全场景实验中表现优于传统方法,平均提升超过2.5%。在少样本场景中,它在41/45个实验中表现优异,平均性能提升超过6.6%。

英文摘要:

In the evolving landscape of machine learning, the adaptation of pre-trained models through prompt tuning has become increasingly prominent. This trend is particularly observable in the graph domain, where diverse pre-training strategies present unique challenges in developing effective prompt-based tuning methods for graph neural networks. Previous approaches have been limited, focusing on specialized prompting functions tailored to models with edge prediction pre-training tasks. These methods, however, suffer from a lack of generalizability across different pre-training strategies. Recently, a simple prompt tuning method has been designed for any pre-training strategy, functioning within the input graph's feature space. This allows it to theoretically emulate any type of prompting function, thereby significantly increasing its versatility for a range of downstream applications. Nevertheless, the capacity of such simple prompts to fully grasp the complex contexts found in graphs remains an open question, necessitating further investigation. Addressing this challenge, our work introduces the Subgraph-level Universal Prompt Tuning (SUPT) approach, focusing on the detailed context within subgraphs. In SUPT, prompt features are assigned at the subgraph-level, preserving the method's universal capability. This requires extremely fewer tuning parameters than fine-tuning-based methods, outperforming them in 42 out of 45 full-shot scenario experiments with an average improvement of over 2.5%. In few-shot scenarios, it excels in 41 out of 45 experiments, achieving an average performance increase of more than 6.6%.

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

↑