Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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.

Elliot Pickens, Chiraag Gohel, Sidharth Satya• 2026

Related benchmarks

TaskDatasetResultRank
Simulation-Based InferenceSBIBM Lotka–Volterra
C2ST Score0.931
15
Posterior EstimationSBIBM SLCP
Joint C2ST82
10
Posterior EstimationSBIBM Two Moons
Joint C2ST58
9
Posterior EstimationSBIBM Gaussian Mixture
Joint C2ST62
9
Posterior EstimationSBIBM SIR
Joint C2ST61
9
Posterior EstimationSBIBM Lotka–Volterra
Joint C2ST0.93
9
Posterior EstimationSBIBM Gaussian Linear
Joint C2ST0.55
8
Posterior EstimationSBIBM Bernoulli GLM
Joint C2ST66
6
Posterior Estimationsolar_dynamo
Joint C2ST0.868
6
Posterior Estimationou
Joint C2ST0.866
6
Showing 10 of 39 rows

Other info

Follow for update