Capability research on foundation models. What makes one
good in the first place; what it was pretrained on; how you would know,
from its behavior, which part of the data was responsible. Most of this is
with the Tabular Foundation Models Lab at Lexsi Labs.
Tabular foundation models
When Does Synthetic Relational Data Teach Models to Use Relations? Tracing Predictive Structure from Pretraining Data to Model Behavior
NeurIPS 2026 · ATTRIB Workshop
Shivam Dubey, Mohamed Bouadi, Nassim Bouarour, Varun Kulkarni, Aditya Tanna, Vinay Kumar Sankarapu
Code
Synthetic relational pretraining data is supposed to teach a tabular foundation model to use relations. This traces the predictive structure from the pretraining distribution through to model behavior, and separates the cases where the model genuinely learns to use relations from the cases where it only appears to.
From Behavior to Provenance: Attributing Tabular Foundation Models to Synthetic Pretraining Data
NeurIPS 2026 · ATTRIB Workshop
Mohamed Bouadi, Nassim Bouarour, Shivam Dubey, Aditya Tanna, Vinay Kumar Sankarapu
Attributing what a tabular foundation model does at inference back to the synthetic data it was pretrained on: behavior to provenance, rather than the usual direction.
Context Poisoning in Tabular ICL: Geometric Hijacking and Inverse Scaling
2026 · mechanistic interpretability
Shivam Dubey
The first logit-lens implementation for tabular in-context learning: a classification phase transition at Layer 8, representation hijacking as early as Layer 4, and total hijacking from 1.2% poisoned context. Larger models are more vulnerable, not less. QK/OV decomposition and direct logit attribution show that mechanistically-informed defenses (PCA subspace scrubbing, head ablation) fail because of manifold redundancy.
Code
Pretraining and inductive bias
ICML 2026 · FMSD Workshop
Mohamed Bouadi, Nassim Bouarour, Varun Kulkarni, Shivam Dubey, Aditya Tanna, Vinay Kumar Sankarapu
arXiv
Introduces O'Prior, a compositional realism prior for synthetic pretraining of tabular foundation models. I ran the interpretability and data-quality ablations, which confirm that mechanism diversity, realism composition, and shift-aware stress each contribute independently to downstream quality.
AI for science
Physics-Informed Neural Operators with Multi-Method Explainability for Ocean Temperature Prediction
AAAI 2026 · XAI4Science Workshop
Shivam Dubey
Neural operators for ocean temperature prediction, with the physics constraint and the explanation methods carrying equal weight: a forecast you cannot interrogate is not much use to the people who have to act on it.