Benchmarking Tabular Foundation Models for Conditional Density Estimation in Regression
Not on the map. Why?
Abstract (arXiv)
Conditional density estimation (CDE) - recovering the full conditional distribution of a response given tabular covariates - is essential in settings with heteroscedasticity, multimodality, or asymmetric uncertainty. Recent tabular foundation models, such as TabPFN and TabICL, naturally produce predictive distributions, but their effectiveness as general-purpose CDE methods has not been systematically evaluated, unlike their performance for point prediction, which is well studied. We benchmark three tabular foundation model variants against a diverse set of parametric, tree-based, and neural CDE baselines on 39 real-world datasets, across training sizes from 50 to 20,000, using six metrics covering density accuracy, calibration, and computation time. Across all sample sizes, foundation models achieve the best CDE loss, log-likelihood, and CRPS on the large majority of datasets tested. Calibration is competitive at small sample sizes but, for some metrics and datasets, lags behind task-specific neural baselines at larger sample sizes, suggesting that post-hoc recalibration may be a valuable complement. In a photometric redshift case study using SDSS DR18, TabPFN exposed to 50,000 training galaxies outperforms all baselines trained on the full 500,000-galaxy dataset. Taken together, these results establish tabular foundation models as strong off-the-shelf conditional density estimators.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 11 of its 63 references that are in the map. First posted on arXiv 27 Mar 2026.
Builds on
11 of 63 references.
References · 11
- 2014NICE: Non-linear Independent Components Estimationcited by 1,058
- 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 289
- 2025TabArena: A Living Benchmark for Machine Learning on Tabular Datacited by 176
- 2026TabICLv2: A better, faster, scalable, and open tabular foundation modelcited by 144
- 2024A Closer Look at Deep Learning Methods on Tabular Datasetscited by 41
- 2024TabReD: Analyzing Pitfalls and Filling the Gaps in Tabular Deep Learning Benchmarkscited by 33
- 2018Gaussian Process Conditional Density Estimationcited by 23
- 2017The Kernel Mixture Network: A Nonparametric Method for Conditional Density Estimation of Continuous Random Variablescited by 15
- 2021Density Regression with Bayesian Additive Regression Treescited by 6
- 2025CP4SBI: Local Conformal Calibration of Credible Sets in Simulation-Based Inferencecited by 6
- 2026Distributional Regression with Tabular Foundation Models: Evaluating Probabilistic Predictions via Proper Scoring Rulescited by 3
Built on by
Citing papers per year: 2026: 3.
3 papers.