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

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.

Zhikai Chen, Junyu Yin, Jialiang Gu, Siheng Xiong, Xiaoze Liu, Ruowang Zhang, Keren Zhou, Kai Guo• 2026

Related benchmarks

TaskDatasetResultRank
Study Outcome Predictionrel (trial)
ROC-AUC0.634
65
Entity ClassificationRelBench rel-avito user-visits
AUC64.6
49
Entity ClassificationRelBench rel-stack user-badge
AUC86.4
40
site-succ regressionrel (trial)
R2 Score0.224
21
item-sales regressionrel-hm
R20.533
21
ad-ctr regressionrel-avito
R^20.111
21
Entity ClassificationRelBench rel-stack user-engage
AUC88.5
21
Binary ClassificationRelBench V1
Avito Click AUROC63.9
20
Classificationrel-f1 d-dnf
AUC0.812
13
Classificationrel-f1 d-top3
AUC89.9
13
Showing 10 of 31 rows

Other info

Follow for update