arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.13630cs.LG

UniGraph:学习面向文本属性图的统一跨域基础模型

UniGraph: Learning a Unified Cross-Domain Foundation Model for Text-Attributed Graphs

National University of Singapore(新加坡国立大学)

Yufei He, Yuan Sui, Xiaoxin He, Bryan Hooi

AI总结:

本文提出UniGraph,一个基于文本属性图统一跨域基础模型,采用语言模型与图神经网络级联架构及掩码图建模预训练,实现零样本和少样本迁移,性能超越监督训练的GNNs。

摘要:

ChatGPT和GPT-4等基础模型已经彻底改变了人工智能领域,展现出在超出其初始训练目标的各种任务和应用中卓越的泛化能力。然而,图学习主要集中于针对特定任务或数据集定制的单图模型,缺乏将所学知识迁移到不同领域的能力。这一局限性源于图结构固有的复杂性和多样性,以及图数据特有的不同特征和标签空间。在本文中,我们认识到文本是一种有效的统一媒介,并利用文本属性图(TAGs)来发挥这一潜力。我们提出了UniGraph框架,旨在学习一个面向TAGs的基础模型,该模型能够泛化到不同领域中的未见图和任务。与使用预计算的不同维度节点特征作为输入的单图模型不同,我们的方法利用文本特征来统一节点表示,即使是对于分子图等天然不具有文本特征的图也是如此。我们提出了一种新颖的语言模型(LMs)和图神经网络(GNNs)级联架构作为骨干网络。此外,我们提出了第一个专门为TAGs上的大规模自监督学习设计的预训练算法,该算法基于掩码图建模。我们引入使用大语言模型(LLMs)的图指令微调,以实现零样本预测能力。我们在各种图学习任务和领域上的综合实验证明了该模型在未见图上的自监督表示学习、少样本上下文迁移和零样本迁移方面的有效性,甚至超越或匹配了在目标数据集上经过监督训练的GNNs的性能。

英文摘要:

Foundation models like ChatGPT and GPT-4 have revolutionized artificial intelligence, exhibiting remarkable abilities to generalize across a wide array of tasks and applications beyond their initial training objectives. However, graph learning has predominantly focused on single-graph models, tailored to specific tasks or datasets, lacking the ability to transfer learned knowledge to different domains. This limitation stems from the inherent complexity and diversity of graph structures, along with the different feature and label spaces specific to graph data. In this paper, we recognize text as an effective unifying medium and employ Text-Attributed Graphs (TAGs) to leverage this potential. We present our UniGraph framework, designed to learn a foundation model for TAGs, which is capable of generalizing to unseen graphs and tasks across diverse domains. Unlike single-graph models that use pre-computed node features of varying dimensions as input, our approach leverages textual features for unifying node representations, even for graphs such as molecular graphs that do not naturally have textual features. We propose a novel cascaded architecture of Language Models (LMs) and Graph Neural Networks (GNNs) as backbone networks. Additionally, we propose the first pre-training algorithm specifically designed for large-scale self-supervised learning on TAGs, based on Masked Graph Modeling. We introduce graph instruction tuning using Large Language Models (LLMs) to enable zero-shot prediction ability. Our comprehensive experiments across various graph learning tasks and domains demonstrate the model's effectiveness in self-supervised representation learning on unseen graphs, few-shot in-context transfer, and zero-shot transfer, even surpassing or matching the performance of GNNs that have undergone supervised training on target datasets.

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

↑