atlas

Research map / Graph learning / Knowledge and temporal graphs

Position: Relational Deep Learning - Graph Representation Learning on Relational Databases

ICML 2024 · Graph learning › Knowledge and temporal graphs · approach: Heterogeneous graph learning · 3 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Position paper proposing Relational Deep Learning, which represents relational databases as temporal heterogeneous graphs and trains GNNs end-to-end, plus the RelBench benchmark with two databases and four tasks.

Problem

Machine learning on multi-table relational databases currently requires manual feature engineering (joins and aggregations) into a single table, which is slow, error-prone, leaks temporal information, and suboptimal.

Approach

View a relational database as a temporal heterogeneous graph with one node per row and edges from primary-foreign keys, then learn predictions end-to-end with message passing GNNs that only pass messages from earlier timestamps, with labels generated from historical data via training tables.

Key results

Limitations

Experimental setup

Datasets: rel-amazon rel-stackex

Benchmarks: RelBench rel-amazon-ltv RelBench rel-amazon-churn RelBench rel-stackex-engage RelBench rel-stackex-votes

Metrics: MAE average precision

Baselines: XGBoost LightGBM

Builds on: message passing neural networks heterogeneous graph neural networks PyTorch Frame PyTorch Geometric XGBoost

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy