OpenRFM: Dissecting Relational In-Context Learning
About
Relational Foundation Models (RFMs) promise a single pre-trained predictor that, given any relational database, returns predictions in one forward pass via relational in-context learning (ICL). Yet a substantial gap separates open RFMs from their commercial counterparts, and the origin of this gap has not been systematically understood. We dissect a representative framework, the Relational Transformer (RT), from two perspectives. Model side: we show that RT performs relation-level ICL, and a kernel regression view shows it fails when sparse label-cell coverage yields an underdetermined regression. Data side: we ablate RT's pre-training source and find that existing synthetic-only pre-training and in-distribution pre-training drive the same architecture into different regimes, lazy vs. feature-learning. Probing this gap reveals that the missing ingredient is a support-identifiable relational latent in the label-generation process. These two diagnoses translate into (1) a dual-stage ICL architecture that combines the relational backbone with a batch-level ICL layer lifted from a pre-trained tabular foundation model to overcome relation-level label scarcity, and (2) a homophily-aware synthetic plus continual real-data pre-training mixture, augmented with a prototype-based regularization. These choices define OpenRFM, a simple yet effective RFM that improves average task performance by approximately 30% over the RT backbone and surpasses the commercial model KumoRFMv1 on a large set of evaluation tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Study Outcome Prediction | rel (trial) | ROC-AUC0.634 | 65 | |
| Entity Classification | RelBench rel-avito user-visits | AUC64.6 | 49 | |
| Entity Classification | RelBench rel-stack user-badge | AUC86.4 | 40 | |
| site-succ regression | rel (trial) | R2 Score0.224 | 21 | |
| item-sales regression | rel-hm | R20.533 | 21 | |
| ad-ctr regression | rel-avito | R^20.111 | 21 | |
| Entity Classification | RelBench rel-stack user-engage | AUC88.5 | 21 | |
| Binary Classification | RelBench V1 | Avito Click AUROC63.9 | 20 | |
| Classification | rel-f1 d-dnf | AUC0.812 | 13 | |
| Classification | rel-f1 d-top3 | AUC89.9 | 13 |