arXiv:2605.16085cs.DBcs.AI2026-05中稿 · the 1st Workshop o…

用语言模型+图神经网络构建关系数据库的通用基础模型。

Towards Foundation Models for Relational Databases with Language Models and Graph Neural Networks

论文配图:Towards Foundation Models for Relational Databases with Language Models and Graph Neural Networks
图 1 · 摘自论文原文
  • 结合BART与GraphSAGE,同时捕捉行内语义和表间关系。
  • 在driver-dnf任务上达到67.40的ROC-AUC,接近监督模型表现。
  • 轻量级设计适合资源受限场景,是迈向通用数据库模型的一步。

关系型数据库存储了全球大量结构化信息,对复杂预测应用至关重要。然而,传统深度学习方法常通过人工特征工程将数据库扁平化为单张表,丢失了关系上下文。关系深度学习(RDL)通过将数据库建模为关系实体图(REGs)并使用图神经网络(GNN)来缓解此问题,但通常任务和数据库特定。为融合两种范式优势,我们提出一种混合架构:使用微调的BART编码器捕获行内语义,再通过基于GraphSAGE的GNN注入关系上下文。在RelBench上的实验表明,GNN显著丰富了BART的行嵌入,在rel-f1数据集的driver-dnf任务上达到67.40的ROC-AUC。该结果优于多数基线,接近LightGBM(68.86),仅比先进RDL模型(72.62)低5.22点,但与顶尖基础模型KumoRFM(82.63)仍有差距。这些结果表明,轻量级的LM-GNN混合架构为构建关系数据库的基础模型提供了一条有前景且资源高效的道路。

原文摘要 · Abstract (English)

Relational databases store much of the world's structured information, and they are essential for driving complex predictive applications. However, deep learning progress on relational data remains limited, as conventional approaches flatten databases into single tables via manual feature engineering, discarding relational context. Relational deep learning (RDL) addresses this by modeling databases as relational entity graphs (REGs) for graph neural networks (GNNs), but remains task- and database-specific. To combine the strengths of both paradigms, we propose a hybrid architecture combining a fine-tuned BART encoder to capture intra-row semantics with a GraphSAGE-based GNN over REGs to inject relational context. Experiments on RelBench show that the GNN substantially enriches BART's row embeddings, achieving a ROC-AUC of 67.40 on the driver-dnf task from the rel-f1 dataset. This performance is competitive with supervised baselines such as LightGBM (68.86) and narrows the gap to RDL (72.62) to within 5.22 points, though a substantial gap remains to state-of-the-art foundation models such as KumoRFM (82.63). These results suggest that lightweight hybrid LM-GNN architectures offer a promising and resource-efficient path towards foundation models for relational databases.

数据库图神经网络基础模型

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