Distilling Tabular Foundation Models for Structured Health Data
On the map: Knowledge distillation → General knowledge distillation methods
Abstract (arXiv)
Tabular foundation models (TFMs) achieve strong performance on health datasets, but their inference cost and infrastructure requirements limit practical use. We study whether their predictive behavior can be transferred to lightweight tabular models through knowledge distillation. Since in-context TFMs condition on the training set at inference time, naive distillation can introduce context leakage; we address this with stratified out-of-fold teacher labeling. Across 19 healthcare datasets, 6 TFM teachers, 4 student families, and several multi-teacher ensembles, we find that distilled students retain at least 90% of teacher AUC, outperforming teachers in some cases, while running at least 26× faster on CPU and preserving calibration and fairness critical for health applications. Moreover, multi-teacher averaging does not consistently improve over the best single teacher. Leakage-aware distillation is thus a viable route for bringing TFM-quality predictions into inference-constrained health settings.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in General knowledge distillation methods, one of 12 directions in Knowledge distillation. Builds on 16 of its 26 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
16 of 26 references.
Closest ancestors · 1
Same direction, newest first
Other references · 15
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2016Equality of Opportunity in Supervised Learningcited by 2,007
- 2026TabICLv2: A better, faster, scalable, and open tabular foundation modelcited by 144
- 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 132
- 2024TabDPT: Scaling Tabular Foundation Models on Real Datacited by 113
- 2020Understanding and Improving Knowledge Distillationcited by 83
- 2025LimiX: Unleashing Structured-Data Modeling Capability for Generalist Intelligencecited by 79
- 2025Orion-MSP: Multi-Scale Sparse Attention for Tabular In-Context Learningcited by 19
- 2025TabTune: A Unified Library for Inference and Fine-Tuning Tabular Foundation Modelscited by 11
- 2025Dataset distillation for memorized data: Soft labels can leak held-out teacher knowledgecited by 10
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Distillation of Tabular Foundation Models into Efficient Predictors
- 2026TabKD: Tabular Knowledge Distillation through Interaction Diversity of Learned Feature Bins
- 2026TabGift: A Context-Efficient Foundation Model for Tabular Data Generation
- 2026Pocket Foundation Models: Distilling TFMs into CPU-Ready Gradient-Boosted Trees
- 2026VIP-COP: Context Optimization for Tabular Foundation Models
- 2026TabSwift: An Efficient Tabular Foundation Model with Row-Wise Attention
- 2026TabFM-Auto: Self-Evolving Pipelines for Tabular Foundation Models
- 2026Test-Time Compute for Tabular Foundation Models: Mechanisms, Gains, and Limits