arXiv:2512.08763cs.LG2025-12KDD被引 4

用强化学习统一优化图提示,让模型更通用且效果更好

Learning and Editing Universal Graph Prompt Tuning via Reinforcement Learning

  • 在所有节点加提示,确保理论上的通用性
  • 通过强化学习筛选节点并动态调整提示,提升性能
  • 适用于多种预训练策略和少样本场景

早期图提示调优方法依赖任务特定设计,限制了在不同预训练策略间的适应性。相比之下,通用图提示调优直接在图特征空间操作,理论上可等价任意提示函数,摆脱对特定预训练的依赖。近期工作提出选择性节点提示以追求更优提示,但我们指出这会损害通用性的理论基础。本文通过引入更严格约束,证明在所有节点添加提示是实现通用性的必要条件。为此,我们提出新模型LEAP:先构建基础通用提示,再用演员-评论家强化学习选择节点并编辑提示。在多种预训练策略、全样本与少样本场景下的图级和节点级任务中,LEAP持续优于微调及其他提示方法。

原文摘要 · Abstract (English)

Early graph prompt tuning approaches relied on task-specific designs for Graph Neural Networks (GNNs), limiting their adaptability across diverse pre-training strategies. In contrast, another promising line of research has investigated universal graph prompt tuning, which operates directly in the input graph's feature space and builds a theoretical foundation that universal graph prompt tuning can theoretically achieve an equivalent effect of any prompting function, eliminating dependence on specific pre-training strategies. Recent works propose selective node-based graph prompt tuning to pursue more ideal prompts. However, we argue that selective node-based graph prompt tuning inevitably compromises the theoretical foundation of universal graph prompt tuning. In this paper, we strengthen the theoretical foundation of universal graph prompt tuning by introducing stricter constraints, demonstrating that adding prompts to all nodes is a necessary condition for achieving the universality of graph prompts. To this end, we propose a novel model and paradigm, Learning and Editing Universal GrAph Prompt Tuning (LEAP), which preserves the theoretical foundation of universal graph prompt tuning while pursuing more ideal prompts. Specifically, we first build the basic universal graph prompts to preserve the theoretical foundation and then employ actor-critic reinforcement learning to select nodes and edit prompts. Extensive experiments on graph- and node-level tasks across various pre-training strategies in both full-shot and few-shot scenarios show that LEAP consistently outperforms fine-tuning and other prompt-based approaches.

图神经网络提示调优强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。