RelBench: A Benchmark for Deep Learning on Relational Databases

We present RelBench, a public benchmark for solving predictive tasks in relational databases with deep learning. RelBench provides databases and tasks spanning diverse domains, scales, and database dimensions, and is intended to be a foundational infrastructure for future research in this direction. We use RelBench to conduct the first comprehensive empirical study of graph neural network (GNN) based predictive models on relational data, as recently proposed by Fey et al. 2024. End-to-end learned GNNs are capable fully exploiting the predictive signal encoded in links between entities, marking a significant shift away from the dominant paradigm of manual feature engineering combined with tabular machine learning. To thoroughly evaluate GNNs against the prior gold-standard we conduct a user study, where an experienced data scientist manually engineers features for each task. In this study, GNNs learn better models whilst reducing human work needed by more than an order of magnitude. This result demonstrates the power of GNNs for solving predictive tasks in relational databases, opening up new research opportunities.

Authors

• Joshua Robinson
• Rishabh Ranjan Indian Institute of Technology Jodhpur, India
• Weihua Hu
• Kexin Huang
• Jiaqi Han
• Alejandro Dobles
• Matthias Fey
• Jan E. Lenssen
• Yiwen Yuan Carnegie Mellon University
• Zecheng Zhang
• Xinwei He
• Jure Leskovec Stanford University, Department of Computer Science

Keywords

No keywords are indexed for this paper.