LimiX-2: A Contextual Mechanism Network Towards General Structured-Data Intelligence
TL;DR - LimiX-2 is a foundation model for structured data that jointly models context-dependent data-generating mechanisms rather than focusing only on target prediction. It reportedly outperforms dataset-specific and tabular foundation models while also supporting causal skeleton recovery.
- Uses Contextual Mechanism Networks to learn (p(x, y \mid D_{\mathrm{context}})), shifting in-context learning toward joint mechanism modeling.
- Pretrains via Context-Conditional Masked Modeling on synthetic datasets generated by diverse structural causal models.
- Evaluations on TabArena, TALENT, and BCCO show gains over existing dataset-specific models and tabular foundation models.
- Feature attention captures direct causal relationships, enabling accurate recovery of causal graph skeletons.