Universal Encoders for Modular Relational Deep Learning
About
Relational Deep Learning (RDL) models multi-tabular databases as temporal heterogeneous graphs for end-to-end representation learning. While RDL is evolving rapidly, existing approaches face significant generalization obstacles. They are either schema-specific, requiring training from scratch for every new database, or they rely on monolithic architectures that entangle feature encoding with graph message-passing. Analyzing these limitations, we establish four core pillars for building foundational relational models: semantic granularity, structural topology, temporal causality, and unified optimization. Addressing these pillars, we propose a modular approach that decouples row encoding from graph message-passing. We introduce the Universal Row Encoder, a transformer-based module that integrates raw cell data with schema metadata$-$including column semantics, table names, and global distribution statistics$-$to produce table-width invariant row embeddings. By explicitly feeding global statistics to an intra-row self-attention mechanism, the encoder natively contextualizes unseen features and handles sparse data. Serving as a flexible "backend" for any downstream graph architecture, our pretrained encoder enhances cross-database knowledge transfer on the established RelBench benchmarks while improving learning convergence and memory footprint.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| driver-position | RELBENCH rel-f1 (val) | MAE3.29 | 10 | |
| study-adverse | RELBENCH rel-trial (val) | MAE54.37 | 10 | |
| site-success | RELBENCH rel-trial (val) | MAE0.45 | 10 | |
| ad-ctr | rel-avito (val) | MAE0.04 | 2 | |
| ad-ctr | rel-avito (test) | MAE0.04 | 2 | |
| item-ltv | rel-amazon (val) | MAE56.89 | 2 | |
| item-ltv | rel-amazon (test) | MAE61.53 | 2 | |
| post-votes | rel-stack (val) | MAE0.06 | 2 | |
| site-success | rel trial (test) | MAE0.47 | 2 | |
| study-adverse | rel trial (test) | MAE55.79 | 2 |