On the Uncertainty Quantification Ability of Tabular Foundation Models

Tyler R. Johnson, Kian Ben-Jacob, Nima Negarandeh, Oriol Vendrell-Gallart, Ramin Bostanabad
5/31/2026
stat.MLcs.LG

Official code: kianswarehouse/GPvsPFN (0 stars, MIT, last commit 2026-05-30).

Abstract

Foundation models (FMs) have achieved substantial success in generalizing across tasks without problemspecific training or fine-tuning. However, many critical applications in mechanics and computational science require not only accurate predictions but also reliable uncertainty quantification (UQ). Herein we investigate the UQ capabilities of tabular FMs in regression tasks through a comprehensive empirical study comparing Tabular Prior-Data Fitted Networks (TabPFN) against Gaussian processes (GPs). We systematically evaluate these two methods across a host of regression problems with varying complexity, dataset sizes, and input dimensionalities. We use a default setting to build all the GPs and for a fair comparison against TabPFN v2.5. Our findings highlight an important trade-off between explicit and learned priors: while TabPFN achieves highly competitive performance for complex, high-dimensional problems with sufficient data, GPs often provide superior predictive accuracy and UQ in data-scarce settings. Moreover, when the chosen kernel constitutes a good prior for the underlying function, GP performance can substantially exceed that of TabPFN. Our results can be reproduced from https://github.com/kianswarehouse/GPvsPFN.

View on arXivView PDF

Code Implementations(1)

MIT

Public repository containing scripts for validating experiments and additional results for the paper "On the Uncertainty Quantification Ability of Tabular Foundation Models" - part of the IEEE Computing in Science and Engineering Special Issue on "Controversies on the Usage of AI/ML for Science and Engineering".

00PythonMay 15, 20264 months agoMIT

Similar Papers

Cite this paper

@article{johnson2026uncertainty,
  title  = {On the Uncertainty Quantification Ability of Tabular Foundation Models},
  author = {Tyler R. Johnson and Kian Ben-Jacob and Nima Negarandeh and Oriol Vendrell-Gallart and Ramin Bostanabad},
  year   = {2026},
  eprint = {2606.01427},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.01427v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.01427.svg)](https://codeofpaper.com/paper/2606.01427)

Discussion