Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

RelBench: A Benchmark for Deep Learning on Relational Databases

About

We present RelBench, a public benchmark for solving predictive tasks over relational databases with graph neural networks. RelBench provides databases and tasks spanning diverse domains and scales, and is intended to be a foundational infrastructure for future research. We use RelBench to conduct the first comprehensive study of Relational Deep Learning (RDL) (Fey et al., 2024), which combines graph neural network predictive models with (deep) tabular models that extract initial entity-level representations from raw tables. End-to-end learned RDL models fully exploit the predictive signal encoded in primary-foreign key links, marking a significant shift away from the dominant paradigm of manual feature engineering combined with tabular models. To thoroughly evaluate RDL against this prior gold-standard, we conduct an in-depth user study where an experienced data scientist manually engineers features for each task. In this study, RDL learns better models whilst reducing human work needed by more than an order of magnitude. This demonstrates the power of deep learning for solving predictive tasks over relational databases, opening up many new research opportunities enabled by RelBench.

Joshua Robinson, Rishabh Ranjan, Weihua Hu, Kexin Huang, Jiaqi Han, Alejandro Dobles, Matthias Fey, Jan E. Lenssen, Yiwen Yuan, Zecheng Zhang, Xinwei He, Jure Leskovec• 2024

Related benchmarks

TaskDatasetResultRank
Study Outcome Predictionrel (trial)
ROC-AUC0.674
65
Entity ClassificationRelBench rel-avito user-visits
AUC64.6
49
Entity RegressionRelBench v1.0 (test)
CTR (Avito Ad)18.3
45
Entity ClassificationRelBench rel-stack user-badge
AUC87.9
40
Binary ClassificationRelBench 1.0 (test)
Relational Amazon User Churn72
26
ad-ctr regressionrel-avito
R^20.19
21
Entity ClassificationRelBench rel-stack user-engage
AUC89.6
21
site-succ regressionrel (trial)
R2 Score-0.028
21
item-sales regressionrel-hm
R20.227
21
item-salesRELBENCH rel-hm (test)
MAE0.056
16
Showing 10 of 66 rows

Other info

Follow for update