How Well Do Prior-Data Fitted Networks Amortize Bayesian Inference?
stat.ML, cs.LG
Submitted: 2026-10-09
Updated: 2026-10-09
Code: https://github.com/jax-ml/jax
License: http://creativecommons.org/licenses/by/4.0/
The gist: Prior-data fitted networks (PFNs) are trained to approximate the Bayesian posterior predictive distribution (PPD) of a synthetic prior, yet results about the approximation quality are surprisingly
Terminology
Abstract
Prior-data fitted networks (PFNs) are trained to approximate the Bayesian posterior predictive distribution (PPD) of a synthetic prior, yet results about the approximation quality are surprisingly thin. Taking an empirical risk minimization perspective, we decompose the deployment error of PFNs into prior, context, and amortization error, and measure the latter against a reference PPD under a realistic prior. Results show that PFNs generally recover the targeted PPD very well. Nevertheless, we find that parts of the residual amortization error persist, especially in rare settings under the prior pretraining distribution, even when scaling the model size, which manifests in disagreement between independently pretrained PFNs and leaves uncertainty about the predictor at inference time. As a remedy, we study multiple types of Bayesian model averages and find that these can consistently reduce this uncertainty and improve predictions up to full-size TabICLv2 models.
Sources
- Mind the Gap? A Distributional Comparison of Real and Synthetic Priors for Tabular Foundation Models
- Fine-tuned In-Context Learning Transformers are Excellent Tabular Data Classifiers
- EXAONE Tabular 1.0 : Technical Report
- TabArena: A Living Benchmark for Machine Learning on Tabular Data
- Deep Ensembles: A Loss Landscape Perspective
- Uncertainty Decomposition for Bayes-Filtered Transformers via Bayesian Predictive Inference
- Real-TabPFN: Improving Tabular Foundation Models via Continued Pre-training With Real-World Data
- TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Models
- TabPFN-3.5: Technical Report
- TabFM: A Zero-Shot Foundation Model for Tabular Data
- Transformers Can Do Bayesian Inference
- Bayes' Power for Explaining In-Context Learning Generalizations
- Uncertainty Quantification for Prior-Data Fitted Networks using Martingale Posteriors
- TabMGP: Martingale Posterior with TabPFN
- nanoTabPFN: A Lightweight and Educational Reimplementation of TabPFN
- TabH2O: A Unified Foundation Model for Tabular Prediction
- Beyond IID: How General Are Tabular Foundation Models, Really?
- Towards Evaluating Data Priors for Tabular Foundation Models
- Xiaomi-TabLDM: A Tabular Foundation Model Technical Report
- LimiX: Unleashing Structured-Data Modeling Capability for Generalist Intelligence
Related papers
- Behavior of prediction performance metrics with rare events
- Optimal Estimation of Generic Dynamics by Path-Dependent Neural Jump ODEs
- A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors
- Efficient quadratic entropy with distance sketches
- Embedding-Bias in Conditional Independence Testing
- From Log-Odds to Shapley Values: An Explanatory Geometry for the Weighted Naive Bayes Classifier