Griffin: Towards a Graph-Centric Relational Database Foundation Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Yanbo, Wang, Xiyuan, Gan, Quan, Wang, Minjie, Yang, Qibin, Wipf, David, Zhang, Muhan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912423964835840
author Wang, Yanbo
Wang, Xiyuan
Gan, Quan
Wang, Minjie
Yang, Qibin
Wipf, David
Zhang, Muhan
author_facet Wang, Yanbo
Wang, Xiyuan
Gan, Quan
Wang, Minjie
Yang, Qibin
Wipf, David
Zhang, Muhan
contents We introduce Griffin, the first foundation model attemptation designed specifically for Relational Databases (RDBs). Unlike previous smaller models focused on single RDB tasks, Griffin unifies the data encoder and task decoder to handle diverse tasks. Additionally, we enhance the architecture by incorporating a cross-attention module and a novel aggregator. Griffin utilizes pretraining on both single-table and RDB datasets, employing advanced encoders for categorical, numerical, and metadata features, along with innovative components such as cross-attention modules and enhanced message-passing neural networks (MPNNs) to capture the complexities of relational data. Evaluated on large-scale, heterogeneous, and temporal graphs extracted from RDBs across various domains (spanning over 150 million nodes), Griffin demonstrates superior or comparable performance to individually trained models, excels in low-data scenarios, and shows strong transferability with similarity and diversity in pretraining across new datasets and tasks, highlighting its potential as a universally applicable foundation model for RDBs. Code available at https://github.com/yanxwb/Griffin.
format Preprint
id arxiv_https___arxiv_org_abs_2505_05568
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Griffin: Towards a Graph-Centric Relational Database Foundation Model
Wang, Yanbo
Wang, Xiyuan
Gan, Quan
Wang, Minjie
Yang, Qibin
Wipf, David
Zhang, Muhan
Machine Learning
Artificial Intelligence
Databases
We introduce Griffin, the first foundation model attemptation designed specifically for Relational Databases (RDBs). Unlike previous smaller models focused on single RDB tasks, Griffin unifies the data encoder and task decoder to handle diverse tasks. Additionally, we enhance the architecture by incorporating a cross-attention module and a novel aggregator. Griffin utilizes pretraining on both single-table and RDB datasets, employing advanced encoders for categorical, numerical, and metadata features, along with innovative components such as cross-attention modules and enhanced message-passing neural networks (MPNNs) to capture the complexities of relational data. Evaluated on large-scale, heterogeneous, and temporal graphs extracted from RDBs across various domains (spanning over 150 million nodes), Griffin demonstrates superior or comparable performance to individually trained models, excels in low-data scenarios, and shows strong transferability with similarity and diversity in pretraining across new datasets and tasks, highlighting its potential as a universally applicable foundation model for RDBs. Code available at https://github.com/yanxwb/Griffin.
title Griffin: Towards a Graph-Centric Relational Database Foundation Model
topic Machine Learning
Artificial Intelligence
Databases
url https://arxiv.org/abs/2505.05568