mlsanityData to 14 Oct 2026

GEAR: Generative Expansion and Real Anchoring for Two-Stage Distillation of Tabular Foundation Models

arXiv 2608.18849 ↗First posted 19 Aug 2026cs.LG stat.ME stat.MLcited by 1

Not on the map. Why?

Abstract (arXiv)

Tabular foundation models (TFMs) achieve strong performance through in-context learning, but context-dependent inference imposes substantial latency and memory costs, hindering large-scale deployment. We propose GEAR (Generative Expansion and Real Anchoring), a modular two-stage framework that distills TFMs into lightweight MLP or tree-based predictors that can be deployed on commodity CPUs. Stage 1 uses synthetic covariates solely as teacher-query locations and trains the student on soft TFM targets, expanding coverage beyond observed rows. Stage 2 re-anchors the student to the target distribution using real labels and out-of-fold teacher predictions, whitch avoids self-labeling leakage. We further derive a risk certificate characterizing the trade-off between generated-query volume and generator fidelity. Experiments on TALENT and TabArena demonstrate the broad applicability of GEAR. Two-stage MLPs outperform supervised MLPs by 1.81--2.00 AUC points on binary tasks and 1.19--1.35 points on multiclass tasks, with additional gains over real-data-only distillation of 1.76--2.19 and 2.09--2.40 points, respectively. On binary tasks, the gains also transfer to LightGBM and XGBoost, and all three student families outperform CatBoost, the strongest non-TFM baseline, in mean AUC. Ablations show gains beyond longer training or alternative warm starts, greater stability from staged than mixed optimization, and generator-dependent diminishing returns as query volume increases. Finally, GEAR reduces median inference time by 57--2866 times and peak prediction memory by 1.9--3.3 times, while retaining higher AUC than matched supervised baselines.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 37 of its 44 references that are in the map. First posted on arXiv 19 Aug 2026.

Builds on

37 of 44 references.

References · 37

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  3. 2019A Comprehensive Survey on Transfer Learningcited by 1,160
  4. 2021Revisiting Deep Learning Models for Tabular Datacited by 740
  5. 2019Modeling Tabular data using Conditional GANcited by 699
  6. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  7. 2022TabDDPM: Modelling Tabular Data with Diffusion Modelscited by 377
  8. 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 289
  9. 2023Mixed-Type Tabular Data Synthesis with Score-based Diffusion in Latent Spacecited by 186
  10. 2025TabArena: A Living Benchmark for Machine Learning on Tabular Datacited by 176
  11. 2026TabICLv2: A better, faster, scalable, and open tabular foundation modelcited by 144
  12. 2024TabDPT: Scaling Tabular Foundation Models on Real Datacited by 113
25 more
  1. 2024Why Tabular Foundation Models Should Be a Research Prioritycited by 106
  2. 2024Better by Default: Strong Pre-Tuned MLPs and Boosted Trees on Tabular Datacited by 92
  3. 2020Transfer Learning for High-dimensional Linear Regression: Prediction, Estimation, and Minimax Optimalitycited by 90
  4. 2024TabDiff: a Mixed-type Diffusion Model for Tabular Data Generationcited by 63
  5. 2024Retrieval & Fine-Tuning for In-Context Tabular Modelscited by 59
  6. 2024TALENT: A Tabular Analytics and Learning Toolboxcited by 52
  7. 2025A Closer Look at TabPFN v2: Understanding Its Strengths and Extending Its Capabilitiescited by 41
  8. 2025TabPFN Unleashed: A Scalable and Effective Solution to Tabular Classification Problemscited by 36
  9. 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
  10. 2024Mixture of In-Context Prompters for Tabular PFNscited by 33
  11. 2024TabPFGen -- Tabular Data Generation with TabPFNcited by 28
  12. 2025A Comprehensive Survey of Synthetic Tabular Data Generationcited by 28
  13. 2024In-Context Data Distillation with TabPFNcited by 19
  14. 2026LimiX-2M: Mitigating Low-Rank Collapse and Attention Bottlenecks in Tabular Foundation Modelscited by 9
  15. 2025Chunked TabPFN: Exact Training-Free In-Context Learning for Long-Context Tabular Datacited by 8
  16. 2025Synthetic Tabular Data Generation: A Comparative Survey for Modern Techniquescited by 6
  17. 2026End-to-End Compression for Tabular Foundation Modelscited by 6
  18. 2025TabDistill: Distilling Transformers into Neural Nets for Few-Shot Tabular Classificationcited by 3
  19. 2026Causal Data Augmentation for Robust Fine-Tuning of Tabular Foundation Modelscited by 3
  20. 2026TabFORGE: A Tabular Foundation Model for Generative Modellingcited by 3
  21. 2026VIP-COP: Context Optimization for Tabular Foundation Modelscited by 3
  22. 2026Selecting Feature Interactions for Generalized Additive Models by Distilling Foundation Modelscited by 2
  23. 2026Pocket Foundation Models: Distilling TFMs into CPU-Ready Gradient-Boosted Treescited by 2
  24. 2026Distilling Tabular Foundation Models for Structured Health Datacited by 2
  25. 2026Context-Constrained Transfer Learning for Tabular Foundation Models via Data Distillationcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.