Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

No Need to Train Your RDB Foundation Model

About

Relational databases (RDBs) contain vast amounts of heterogeneous tabular information that can be exploited for predictive modeling purposes. But since the space of potential targets is vast across enterprise settings, how can we \textit{avoid retraining} a new model each time we wish to predict a new quantity of interest? Foundation models based on in-context learning (ICL) offer a convenient option, but so far are largely restricted to single-table operability. In generalizing to multiple interrelated tables, it is essential to compress variably-sized RDB neighborhoods into fixed-length ICL samples for consumption by the decoder. However, the details here are critical: unlike existing supervised learning RDB pipelines, we provide theoretical and empirical evidence that ICL-specific compression should be constrained \emph{within} high-dimensional RDB columns where all entities share units and roles, not \textit{across} columns where the relevance of heterogeneous data types cannot possibly be determined without label information. Conditioned on this restriction, we then demonstrate that encoder expressiveness is actually not compromised by excluding trainable parameters. Hence we arrive at a principled family of RDB encoders that can be seamlessly paired with already-existing single-table ICL foundation models, whereby no training or fine-tuning is required. From a practical standpoint, we develop scalable SQL primitives to implement the encoder stage, resulting in an easy-to-use open-source RDB foundation model\footnote{\label{foot: RDBLearn_learn} https://github.com/HKUSHXLab/rdblearn} capable of robust performance on unseen datasets out of the box.

Linjie Xu, Yanlin Zhang, Quan Gan, Minjie Wang, David Wipf• 2026

Related benchmarks

TaskDatasetResultRank
Entity ClassificationRELBENCH rel-amazon item-churn (test)
AUROC82.07
14
Churn PredictionAmazon 4DBInfer (test)
AUC0.7741
9
user-churnStackExchange 4DBInfer (test)
AUC0.8796
9
CVRRetailrocket 4DBInfer (test)
AUC0.8469
9
post-upvoteStackExchange 4DBInfer (test)
AUC0.8845
9
CTR-100KOutbrain 4DBInfer (test)
AUC0.5447
9
Entity ClassificationRelBench rel-avito user-clicks
AUC69.04
8
Entity ClassificationRelBench rel-trial study-outcome
AUC71.58
8
Entity ClassificationRelBench rel-avito user-churn
AUC0.6757
8
Entity ClassificationRelBench rel-avito user-visits
AUC65.49
8
Showing 10 of 16 rows

Other info

Follow for update