mlsanityData to 14 Oct 2026

One Pipeline, Many Transformers: Pattern-Specific Imputation Specialists for Tabular Missing Data

arXiv 2510.02625 ↗First posted 3 Oct 2025cs.LGcited by 4

Not on the map. Why?

Abstract (arXiv)

Missing data in tabular datasets forces practitioners into a hard choice: deploy a general-purpose imputer that may perform poorly for the problem at hand, or wait for someone to design a specialized algorithm. This problem is worsened by the fact that real-world missingness rarely satisfies the textbook missing completely at random (MCAR) assumption, as entries are often missing not at random (MNAR), where the probability of being observed depends on the missing data itself. We collapse this trade-off into a single pre-training pipeline that builds transformer-based imputation specialists through three components: an entry-wise featurization that recasts imputation as supervised prediction over row--column context, a synthetic data generator with pluggable missingness modules, and prior-data fitting on millions of synthetic tables. Swapping only the missingness module, with no changes to architecture, loss, or training, yields a pattern-specific specialist that outperforms methods purpose-built for that pattern. We validate this on MissBench, a new benchmark of 42 OpenML datasets and 11 missingness patterns (including 9 MNAR variants) spanning medicine, finance, and engineering. Remarkably, training exclusively on MCAR yields a default model---TabImpute---robust across all tested patterns. In addition, the pattern-specific specialists produced by our pipeline outperform the 14 established baselines on their target patterns. We open-source the pipeline, models, and benchmark.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 23 of its 66 references that are in the map. First posted on arXiv 3 Oct 2025.

Builds on

23 of 66 references.

References · 23

  1. 2022Generative Adversarial Networkscited by 5,627
  2. 2017Decoupled Weight Decay Regularizationcited by 1,831
  3. 2016Variational Inference: A Review for Statisticianscited by 1,472
  4. 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
  5. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  6. 2014OpenML: networked science in machine learningcited by 526
  7. 2021Transformers Can Do Bayesian Inferencecited by 360
  8. 2018GAIN: Missing Data Imputation using Generative Adversarial Netscited by 352
  9. 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 289
  10. 2016Recommendations as Treatments: Debiasing Learning and Evaluationcited by 248
  11. 2024TabDPT: Scaling Tabular Foundation Models on Real Datacited by 113
  12. 2014Matrix Completion and Low-Rank SVD via Fast Alternating Least Squarescited by 104
11 more
  1. 2022HyperImpute: Generalized Iterative Imputation with Automatic Model Selectioncited by 83
  2. 2020not-MIWAE: Deep Generative Modelling with Missing not at Random Datacited by 62
  3. 2023Generating and Imputing Tabular Data via Diffusion and Flow-based Gradient-Boosted Treescited by 62
  4. 2021Causal Matrix Completioncited by 45
  5. 2023ReMasker: Imputing Tabular Data with Masked Autoencodingcited by 37
  6. 2023Learning from data with structured missingnesscited by 24
  7. 2018Imputation and low-rank estimation with Missing Not At Random datacited by 21
  8. 2024DiffPuter: Empowering Diffusion Models for Missing Data Imputationcited by 19
  9. 2025TabFlex: Scaling Tabular Learning to Millions with Linear Attentioncited by 15
  10. 2025CACTI: Leveraging Copy Masking and Contextual Information to Improve Tabular Data Imputationcited by 4
  11. 2023A Complete Characterisation of Structured Missingnesscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.