mlsanityData to 14 Oct 2026

Pre-trained Tabular Foundation Models as Versatile Summary Networks for Neural Posterior Estimation

arXiv 2605.07765 ↗First posted 8 May 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

In this work, we study TabPFN as a training-free, modular summary network for simulation-based Bayesian inference (SBI). Tabular foundation models such as TabPFN are pretrained on broad families of synthetic tabular data-generating processes and adapt at test time through in-context learning, making them natural candidates for SBI, where posterior estimation often depends on learning informative summaries of simulated observations. We propose PFN-NPE: a general recipe that uses a pretrained TabPFN encoder as a fixed summary network for simulator outputs, then pairs the resulting summaries with a downstream inference head chosen for the problem. With normalizing flows as the default inference head, PFN-NPE matches established posterior approximation methods and sometimes outperforms them. More importantly, diagnostic probes show that the TabPFN-derived summaries often preserve useful posterior location and marginal information. These analyses also reveal a limitation in that TabPFN-derived summaries may struggle to represent the joint posterior structure even when the marginals are well recovered. Still, our experiments show that TabPFN can serve as an effective summary network across a diverse set of SBI settings, with the inference network left modular and task-dependent.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 17 of its 23 references that are in the map. First posted on arXiv 8 May 2026.

Builds on

17 of 23 references.

References · 17

  1. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  2. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  3. 2019Neural Spline Flowscited by 596
  4. 2019The frontier of simulation-based inferencecited by 502
  5. 2021Transformers Can Do Bayesian Inferencecited by 360
  6. 2016Revisiting Classifier Two-Sample Testscited by 268
  7. 2018Sequential Neural Likelihood: Fast Likelihood-free Inference with Autoregressive Flowscited by 221
  8. 2019Automatic Posterior Transformation for Likelihood-Free Inferencecited by 219
  9. 2018Validating Bayesian Inference Algorithms with Simulation-Based Calibrationcited by 124
  10. 2023Flow Matching for Scalable Simulation-Based Inferencecited by 101
  11. 2020On Contrastive Learning for Likelihood-free Inferencecited by 98
  12. 2023Statistical Foundations of Prior-Data Fitted Networkscited by 65
5 more
  1. 2023Sampling-Based Accuracy Testing of Posterior Estimators for General Inferencecited by 46
  2. 2025Amortized In-Context Bayesian Posterior Estimationcited by 15
  3. 2025Effortless, Simulation-Efficient Bayesian Inference using Tabular Foundation Modelscited by 13
  4. 2023Simulation-based Inference for High-dimensional Data using Surjective Sequential Neural Likelihood Estimationcited by 8
  5. 2026TabPFN Through The Looking Glass: An interpretability study of TabPFN and its internal representationscited by 2

Built on by

Not cited yet.