LimiX-2: A Contextual Mechanism Network Towards General Structured-Data Intelligence
Abstract
We introduce LimiX-2, a new model in the LimiX family, developed through model and data scaling guided by our previously established scaling laws. LimiX-2 adopts the Contextual Mechanism Networks (CMNs) paradigm and is pretrained with Context-Conditional Masked Modeling (CCMM). CMNs shifts the organizing principle of in-context learning from target-centric prediction to mechanism-oriented joint modeling. Rather than centering the network on the p(y mid x, D_{context}) objective of conventional tabular PFNs, it is designed around learning p(x, y mid D_{context}), a context-dependent representation of the joint structure underlying data generation. Pretraining uses synthetic datasets generated by structural causal models (SCMs) spanning diverse graph structures, functional mechanisms, and observation processes. Evaluations on TabArena, TALENT, and BCCO show that LimiX-2 outperforms current dataset-specific models and tabular foundation models. Beyond predictive performance, the CMN paradigm also promotes causal awareness in LimiX-2: its feature attention encodes direct causal relationships, enabling accurate causal skeleton recovery.
Community
🧩 LimiX-2: Towards General Structured-Data Intelligence
One pretrained model. Multiple structured-data tasks. #1 across three major tabular benchmarks.
LimiX-2 is a pretrained foundation model for structured data that can support
🎯 Classification
📈 Regression
🧩 Missing-value imputation
🔗 Causal skeleton recovery
— with no task-specific parameter updates for downstream prediction tasks.
🏆 #1 across all three benchmarks
LimiX-2 takes the top spot on all three broad tabular benchmarks we evaluate:
🏆 TabArena — 1935 Elo
🏆 TALENT — 1506 Elo
🏆 BCCO — 1432 Elo
LimiX-2 ranks #1 overall across all three, ahead of leading tabular foundation models and strong dataset-specific systems.
And the gains continue as we scale the model from 12.5M → 406.2M parameters: performance improves consistently across all five evaluated scaling series, with no clear sign of saturation within the measured range.
So how can one pretrained model handle such different structured-data problems?
A table is more than a target column.
Most tabular foundation models are organized around one question:
given ( x ), predict ( y ).
LimiX-2 takes a broader view:
Can a model learn the relationships among variables in a table, so that different unknown quantities can be inferred from different observed ones?
We introduce Contextual Mechanism Networks (CMNs), a new paradigm that shifts structured-data in-context learning from target-centric prediction toward context-dependent joint modeling.
Instead of focusing only on
CMNs aim to capture the broader dependency structure behind
LimiX-2 instantiates this idea with Context-Conditional Masked Modeling (CCMM).
During pretraining, different variables are masked under different observation patterns, turning the same table into many conditional prediction problems. Rather than learning only how to predict one designated target, the model learns to infer different variables from different available evidence.
LimiX-2 is pretrained exclusively on synthetic datasets generated from diverse structural causal models (SCMs), spanning different graph structures, functional mechanisms, distributions, and observation processes.
This broader training objective gives a single pretrained model a unified basis for classification, regression, and missing-value reconstruction.
Beyond predictive performance, we also find that LimiX-2's learned feature attention contains structural information that can be used for causal skeleton recovery—suggesting that learning conditional relationships across variables can capture structure beyond a single supervised target.
This points to a broader possibility:
structured-data foundation models may move beyond predicting predefined targets toward learning reusable mechanisms of how variables relate and interact.
** A table is more than a target column. Prediction may be only the beginning.**
The next question is:
How far can general structured-data intelligence scale?
LimiX-2 is just the beginning. Follow the project as we push toward larger models, broader capabilities, and more general structured-data intelligence. 🚀
📄 Paper — https://arxiv.org/abs/2609.17488
💻 Code — https://github.com/limix-ldm-ai/LimiX
🤗 Model — https://huggingface.co/stable-ai/LimiX-2
Reproductions, independent evaluations, integrations, and feedback are very welcome.
Great work on general structured‑data intelligence! Shifting from target‑centric prediction to joint mechanism‑oriented modeling looks like a promising direction for tabular foundation models. Impressive results across TabArena, TALENT and BCCO benchmarks. Really appreciate that the model also shows causal skeleton recovery capability beyond pure prediction performance. Looking forward to seeing further real‑world validations!
Get this paper in your agent:
hf papers read 2609.17488 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 1
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 1
Collections including this paper 0
No Collection including this paper