Tabular foundation models have mostly just been predicting a single target variable. A new paradigm learns the relationships across an entire table instead.
Title: LimiX-2: A Contextual Mechanism Network Towards General Structured-Data Intelligence
URL:
📝 Overview
Contextual Mechanism Networks (CMN) learn the full joint distribution p(x, y) across a table, not just the target y, letting a single model handle classification, regression, missing-value imputation, and causal discovery.
❗ Problem it solves
Existing PFN-based methods specialize in conditional prediction of the target only, without explicitly modeling the dependency structure between variables.
⚙️ Methodology
Cell-level embeddings feed a 24-layer dual-axis transformer, trained via "Context-Conditional Masked Modeling" that masks part of the query rows and jointly predicts both features and targets. It's pretrained at scale on synthetic data generated from structural causal models.
📊 Results
On TabArena (51 datasets), it hits an Elo of 1935 vs. 1818 for the baseline; on BCCO (156 datasets), cumulative wins are 2.06x higher. It also took 1st place on F1 score for causal skeleton recovery on all 6 tested datasets, all while using 4x fewer parameters than TabFM.
🔬 Use cases
Since the same model handles classification, regression, and causal discovery, it could replace much of a data analysis pipeline with a single foundation model.
#
TabularData# #
MachineLearning#