arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.04768cs.LGcs.AI
收藏

十亿级图基础模型

Billion-Scale Graph Foundation Models

Meta

Maya Bechler-Speicher, Yoel Gottlieb, Andrey Isakov, David Abensur, Ami Tavory, Daniel Haimovich, Ido Guy, Udi Weinsberg

AI总结:

本文提出GraphBFF,一种用于构建大规模异构图的十亿参数图基础模型的端到端方法,通过引入GraphBFF Transformer架构,揭示了异构图的神经缩放定律,并在多个下游任务中展示了其优越的性能。

摘要:

图结构数据支撑了许多关键应用。尽管基础模型通过大规模预训练和轻量级适应改变了语言和视觉领域,但将其扩展到一般、现实世界的图结构却具有挑战性。在本文中,我们提出了Graph Billion-Foundation-Fusion(GraphBFF):一种用于构建大规模异构图的十亿参数图基础模型(GFMs)的端到端方法。该方法的核心是GraphBFF Transformer,一种灵活且可扩展的架构,专为实际的十亿级GFMs设计。利用GraphBFF,我们提出了异构图的神经缩放定律,并显示损失随着模型容量或训练数据规模的增加而减少,取决于哪个因素是瓶颈。GraphBFF框架提供了具体的方法论,用于数据分批、预训练和微调,以构建大规模的GFMs。我们通过一个现实世界中的十亿级图展示了该框架的有效性,评估了一个十亿参数的GraphBFF Transformer,按照所提出的配方。在十个不同的现实世界下游任务上,涵盖节点和链接级别的分类和回归,GraphBFF在训练过程中未见过的图上始终优于基线,最大差距达到31个PRAUC点,包括在少样本设置中。最后,我们讨论了使GFMs成为工业规模图学习实际和原则性基础的关键挑战和开放机会。

英文摘要:

Graph-structured data underpins many critical applications. While foundation models have transformed language and vision via large-scale pretraining and lightweight adaptation, extending this paradigm to general, real-world graphs is challenging. In this work, we present Graph Billion-Foundation-Fusion (GraphBFF): an end-to-end recipe for building billion-parameter Graph Foundation Models (GFMs) for large-scale heterogeneous graphs. Central to the recipe is the GraphBFF Transformer, a flexible and scalable architecture designed for practical billion-scale GFMs. Using the GraphBFF, we present neural scaling laws for heterogeneous graphs and show that loss decreases predictably as either model capacity or training data scales, depending on which factor is the bottleneck. The GraphBFF framework provides concrete methodologies for data batching, pretraining, and fine-tuning for building GFMs at scale. We demonstrate the effectiveness of the framework over a real-world billion-scale graph, with an evaluation of a billion-parameter GraphBFF Transformer following the proposed recipe. Across ten diverse, real-world downstream tasks on graphs unseen during training, spanning node- and link-level classification and regression, GraphBFF consistently outperforms baselines, with large margins of up to 31 PRAUC points, including in few-shot settings. Finally, we discuss key challenges and open opportunities for making GFMs a practical and principled foundation for graph learning at industrial scale.

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

↑