用自监督对比学习提升图模型跨域迁移能力
GraphCLIP: Enhancing Transferability in Graph Foundation Models for Text-Attributed Graphs
- 利用大模型生成图摘要对,自监督预训练图模型
- 零样本和少样本场景下均优于现有方法
- 适合需要强泛化能力的图数据下游任务
近年来,由于真实场景中普遍存在自由文本节点特征,以及大语言模型的发展,文本属性图(TAGs)研究受到广泛关注。然而当前方法面临两大挑战:(i) 对标签信息依赖过重;(ii) 跨领域零/少样本迁移能力有限。这限制了数据与模型规模的扩展,因人力成本高及缩放定律复杂,阻碍了具备强迁移能力的图基础模型发展。本文提出GraphCLIP框架,通过自监督对比图摘要预训练方法,提升图基础模型的跨领域零/少样本迁移能力。具体地,借助大模型生成并筛选大规模图摘要对数据,提出新型图摘要预训练方法,并结合不变性学习,增强模型在跨域零样本下的表现。针对少样本学习,设计一种与预训练目标对齐的图提示调优技术,缓解灾难性遗忘并降低学习成本。大量实验表明,GraphCLIP在零样本和少样本设置中均具优势,多下游任务评估验证其通用性。代码已公开于:https://github.com/ZhuYun97/GraphCLIP
原文摘要 · Abstract (English)
Recently, research on Text-Attributed Graphs (TAGs) has gained significant attention due to the prevalence of free-text node features in real-world applications and the advancements in Large Language Models (LLMs) that bolster TAG methodologies. However, current TAG approaches face two primary challenges: (i) Heavy reliance on label information and (ii) Limited cross-domain zero/few-shot transferability. These issues constrain the scaling of both data and model size, owing to high labor costs and scaling laws, complicating the development of graph foundation models with strong transferability. In this work, we propose the GraphCLIP framework to address these challenges by learning graph foundation models with strong cross-domain zero/few-shot transferability through a self-supervised contrastive graph-summary pretraining method. Specifically, we generate and curate large-scale graph-summary pair data with the assistance of LLMs, and introduce a novel graph-summary pretraining method, combined with invariant learning, to enhance graph foundation models with strong cross-domain zero-shot transferability. For few-shot learning, we propose a novel graph prompt tuning technique aligned with our pretraining objective to mitigate catastrophic forgetting and minimize learning costs. Extensive experiments show the superiority of GraphCLIP in both zero-shot and few-shot settings, while evaluations across various downstream tasks confirm the versatility of GraphCLIP. Our code is available at: https://github.com/ZhuYun97/GraphCLIP
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。