Nearly Optimal Bayesian Inference for Structural Missingness
Not on the map. Why?
Abstract (arXiv)
Structural missingness breaks 'just impute and train': values can be undefined by causal or logical constraints, and the mask may depend on observed variables, unobserved variables (MNAR), and other missingness indicators. It simultaneously brings (i) a catch-22 situation with causal loop, prediction needs the missing features, yet inferring them depends on the missingness mechanism, (ii) under MNAR, the unseen are different, the missing part can come from a shifted distribution, and (iii) plug-in imputation, a single fill-in can lock in uncertainty and yield overconfident, biased decisions. In the Bayesian view, prediction via the posterior predictive distribution integrates over the full model posterior uncertainty, rather than relying on a single point estimate. This framework decouples (i) learning an in-model missing-value posterior from (ii) label prediction by optimizing the predictive posterior distribution, enabling posterior integration. This decoupling yields an in-model almost-free-lunch: once the posterior is learned, prediction is plug-and-play while preserving uncertainty propagation. It achieves SOTA on 43 classification and 15 imputation benchmarks, with finite-sample near Bayes-optimality guarantees under our SCM prior.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 52 references that are in the map. First posted on arXiv 26 Jan 2026.
Builds on
28 of 52 references.
References · 28
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
- 2016Distribution-Free Predictive Inference For Regressioncited by 668
- 2021Transformers Can Do Bayesian Inferencecited by 360
- 2018GAIN: Missing Data Imputation using Generative Adversarial Netscited by 352
- 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 289
- 2020Handling Missing Data with Graph Representation Learningcited by 133
- 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 132
- 2020Missing Data Imputation using Optimal Transportcited by 100
- 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 89
16 more
- 2022HyperImpute: Generalized Iterative Imputation with Automatic Model Selectioncited by 83
- 2023PFNs4BO: In-Context Learning for Bayesian Optimizationcited by 73
- 2023Functional Flow Matchingcited by 67
- 2023Generating and Imputing Tabular Data via Diffusion and Flow-based Gradient-Boosted Treescited by 62
- 2025Do-PFN: In-Context Learning for Causal Effect Estimationcited by 51
- 2025CausalPFN: Amortized Causal Effect Estimation via In-Context Learningcited by 49
- 2020MCFlow: Monte Carlo Flow Models for Data Imputationcited by 37
- 2023ReMasker: Imputing Tabular Data with Masked Autoencodingcited by 37
- 2025TabPFN Unleashed: A Scalable and Effective Solution to Tabular Classification Problemscited by 36
- 2023Transformed Distribution Matching for Missing Value Imputationcited by 31
- 2023Learning from data with structured missingnesscited by 24
- 2025Position: The Future of Bayesian Prediction Is Prior-Fittedcited by 20
- 2024DiffPuter: Empowering Diffusion Models for Missing Data Imputationcited by 19
- 2022Data Imputation with Iterative Graph Reconstructioncited by 15
- 2024Interpretable Machine Learning for TabPFNcited by 11
- 2024Optimal Transport for Structure Learning Under Missing Datacited by 6
Built on by
Not cited yet.