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

Mitra: Mixed Synthetic Priors for Enhancing Tabular Foundation Models

About

Since the seminal work of TabPFN, research on tabular foundation models (TFMs) based on in-context learning (ICL) has challenged long-standing paradigms in machine learning. Without seeing any real-world data, models pretrained on purely synthetic datasets generalize remarkably well across diverse datasets, often using only a moderate number of in-context examples. This shifts the focus in tabular machine learning from model architecture design to the design of synthetic datasets, or, more precisely, to the prior distributions that generate them. Yet the guiding principles for prior design remain poorly understood. This work marks the first attempt to address the gap. We systematically investigate and identify key properties of synthetic priors that allow pretrained TFMs to generalize well. Based on these insights, we introduce Mitra, a TFM trained on a curated mixture of synthetic priors selected for their diversity, distinctiveness, and performance on real-world tabular data. Mitra consistently outperforms state-of-the-art TFMs, such as TabPFNv2 and TabICL, across both classification and regression benchmarks, with better sample efficiency.

Xiyuan Zhang, Danielle C. Maddix, Junming Yin, Nick Erickson, Abdul Fatir Ansari, Boran Han, Shuai Zhang, Leman Akoglu, Christos Faloutsos, Michael W. Mahoney, Cuixiong Hu, Huzefa Rangwala, George Karypis, Bernie Wang• 2025

Related benchmarks

TaskDatasetResultRank
User Clicks Predictionrel-avito
ROC-AUC58.12
84
User Engagement Predictionrel-stack
ROC-AUC83.37
69
Driver Top 3 Predictionrel-f1
ROC-AUC81.47
54
Driver DNF Predictionrel-f1
ROC-AUC0.7125
54
Item Churn Predictionrel-amazon
ROC-AUC76.39
54
User Churn PredictionAmazon Rel
ROC-AUC0.6129
54
User Churn Predictionrel-hm
ROC-AUC63.37
52
Study Outcome Predictionrel (trial)
ROC-AUC0.5373
52
User Repeat PredictionRel Event
ROC-AUC64.38
50
User Ignore PredictionRel Event
ROC-AUC0.767
50
Showing 10 of 31 rows

Other info

Follow for update