mlsanityData to 14 Oct 2026

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

arXiv 2605.16085 ↗First posted 15 May 2026cs.DB cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 27 of its 35 references that are in the map. First posted on arXiv 15 May 2026.

Builds on

27 of 35 references.

References · 27

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  4. 2017Inductive Representation Learning on Large Graphscited by 6,320
  5. 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,893
  6. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  7. 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
  8. 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,434
  9. 2019A Comprehensive Survey on Transfer Learningcited by 1,160
  10. 2021Revisiting Deep Learning Models for Tabular Datacited by 740
  11. 2020Benchmarking Graph Neural Networkscited by 680
  12. 2021Deep Neural Networks and Tabular Data: A Surveycited by 491
15 more
  1. 2022GraphMAE: Self-Supervised Masked Graph Autoencoderscited by 448
  2. 2020TabTransformer: Tabular Data Modeling Using Contextual Embeddingscited by 409
  3. 2020TaBERT: Pretraining for Joint Understanding of Textual and Tabular Datacited by 398
  4. 2021SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Trainingcited by 319
  5. 2021Self-Supervised Representation Learning: Introduction, Advances and Challengescited by 148
  6. 2024RelBench: A Benchmark for Deep Learning on Relational Databasescited by 96
  7. 2025Griffin: Towards a Graph-Centric Relational Database Foundation Modelcited by 47
  8. 2020RPT: Relational Pre-trained Transformer Is Almost All You Need towards Democratizing Data Preparationcited by 27
  9. 2025Relational Deep Learning: Challenges, Foundations and Next-Generation Architecturescited by 23
  10. 2024Continual Learning on Graphs: Challenges, Solutions, and Opportunitiescited by 21
  11. 2023Temporal Fact Reasoning over Hyper-Relational Knowledge Graphscited by 10
  12. 2024Predictive Multiplicity of Knowledge Graph Embeddings in Link Predictioncited by 9
  13. 2025Certainty in Uncertainty: Reasoning over Uncertain Knowledge Graphs with Statistical Guaranteescited by 9
  14. 2023Towards Foundation Models for Relational Databases [Vision Paper]cited by 7
  15. 2025Seeing and Knowing in the Wild: Open-domain Visual Entity Recognition with Large-scale Knowledge Graphs via Contrastive Learningcited by 4

Built on by

Not cited yet.