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

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.

Jakub Pele\v{s}ka, Gustav \v{S}\'ir• 2026

Related benchmarks

TaskDatasetResultRank
driver-positionRELBENCH rel-f1 (val)
MAE3.29
10
study-adverseRELBENCH rel-trial (val)
MAE54.37
10
site-successRELBENCH rel-trial (val)
MAE0.45
10
ad-ctrrel-avito (val)
MAE0.04
2
ad-ctrrel-avito (test)
MAE0.04
2
item-ltvrel-amazon (val)
MAE56.89
2
item-ltvrel-amazon (test)
MAE61.53
2
post-votesrel-stack (val)
MAE0.06
2
site-successrel trial (test)
MAE0.47
2
study-adverserel trial (test)
MAE55.79
2
Showing 10 of 11 rows

Other info

Follow for update