arXiv:2412.12155cs.LGcs.AI2024-12被引 5

用提示词微调让无符号图预训练模型高效适配少标签有符号图任务

SGPT: Few-Shot Prompt Tuning for Signed Graphs

  • 基于平衡理论设计图模板,分离负边带来的混合关系
  • 在7个数据集上显著超越现有方法,少样本下表现优异
  • 适合标签稀缺的工业有符号图场景,通用性强

有符号图神经网络(SGNN)在学习有符号图表示方面表现优异,但通常需要大量特定任务标签,限制了其在标签稀缺工业场景中的应用。相比之下,无符号图结构丰富,可直接用于预训练图神经网络(GNN),有望降低下游有符号图任务的监督需求。然而,从无符号图到有符号图的知识迁移具有挑战性,因两者在图类型和任务目标上存在根本差异。为此,我们提出签名图提示微调(SGPT),一种将预训练无符号GNN适配至少样本有符号图任务的新框架。我们首先基于平衡理论设计图模板,解耦负边引入的混合节点关系,缓解无符号与有符号图间的结构不匹配。进一步引入任务模板,将下游有符号任务统一为链接预测目标,对齐优化目标。此外,我们设计特征提示以对齐下游语义空间与预训练特征空间,并引入语义提示以任务感知方式融合链接符号语义。我们在七个基准有符号图数据集上进行了广泛实验,结果表明SGPT显著优于现有最先进方法,建立了一种强大且通用的少样本有符号图学习方案。

原文摘要 · Abstract (English)

Signed Graph Neural Networks (SGNNs) are effective in learning expressive representations for signed graphs but typically require substantial task-specific labels, limiting their applicability in label-scarce industrial scenarios. In contrast, unsigned graph structures are abundant and can be readily leveraged to pre-train Graph Neural Networks (GNNs), offering a promising solution to reduce supervision requirements in downstream signed graph tasks. However, transferring knowledge from unsigned to signed graphs is non-trivial due to the fundamental discrepancies in graph types and task objectives between pre-training and downstream phases. To address this challenge, we propose Signed Graph Prompt Tuning (SGPT), a novel graph prompting framework that adapts pre-trained unsigned GNNs to few-shot signed graph tasks. We first design a graph template based on balance theory to disentangle mixed node relationships introduced by negative links, mitigating the structural mismatches between unsigned and signed graphs. We further introduce a task template that reformulates downstream signed tasks into a unified link prediction objective, aligning their optimization goals with the pre-training task. Furthermore, we develop feature prompts that align downstream semantic spaces with the feature spaces learned during pre-training, and semantic prompts to integrate link sign semantics in a task-aware manner. We conduct extensive experiments on seven benchmark signed graph datasets, demonstrating that SGPT significantly outperforms existing state-of-the-art methods, establishing a powerful and generalizable solution for few-shot signed graph learning.

有符号图提示微调少样本学习图神经网络

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