stat.MLSep 27, 2026

Can Tabular Foundation Models Amortize Statistical Inference?

Authors: Kai Ye, Shijin Gong, Hongyi Zhou, Valentina Zangirolami, Chengchun Shi

Organizations: Department of Statistics, London School of Economics and Political Science, London, UK · School of Management, University of Science and Technology of China, Hefei, China · School of Statistics and Data Science, Shanghai University of Finance and Economics, Shanghai, China · Department of Economics, Management and Statistics, University of Milano-Bicocca, Milan, Italy

Abstract

For decades, statistical inference has largely been developed one problem at a time. Given a scientific target, such as a treatment effect or a regression function, statisticians design a problem-specific estimator together with a procedure for quantifying its uncertainty. This paper proposes a different paradigm. We focus on a classical problem in statistical inference, confidence interval construction, and develop TabCon, an amortized inference system built on a tabular foundation model that produces confidence intervals for new datasets through a simple forward pass. The key methodological ingredients of TabCon are a sparse mixture-of-experts architecture and reinforcement-learning-based post-training that calibrate the resulting confidence intervals to a desired coverage level. Across a wide range of benchmark datasets, TabCon attains near-nominal coverage while producing short confidence intervals. At inference time, it also offers considerably greater computational efficiency, running 50 times faster than the classical bootstrap procedure, even when the latter uses only 50 bootstrap samples.

Figures & tables

Appendix figures & tables9 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Conformal Prediction and Conditional Coverage for Tabular Foundation Models

    Sep 28, 2026Sungwoo Park, Sunghee Park, Won ChangTabular Foundation ModelsGroup-Conditional Conformal Prediction

  2. TabPFN-3.5: Technical Report

    Sep 15, 2026Benjamin Jäger, Nick Erickson, Léo Grinsztajn +44Tabular Foundation ModelsInference-Time Optimization

  3. Is One Layer Enough? Understanding Inference Dynamics in Tabular Foundation Models

    May 7, 2026Amir Rezaei Balef, Mykhailo Koshil, Katharina EggenspergerTabular Foundation ModelsTransformer Interpretability