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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Simulation-Based Inference | SBIBM Lotka–Volterra | C2ST Score0.931 | 15 | |
| Posterior Estimation | SBIBM SLCP | Joint C2ST82 | 10 | |
| Posterior Estimation | SBIBM Two Moons | Joint C2ST58 | 9 | |
| Posterior Estimation | SBIBM Gaussian Mixture | Joint C2ST62 | 9 | |
| Posterior Estimation | SBIBM SIR | Joint C2ST61 | 9 | |
| Posterior Estimation | SBIBM Lotka–Volterra | Joint C2ST0.93 | 9 | |
| Posterior Estimation | SBIBM Gaussian Linear | Joint C2ST0.55 | 8 | |
| Posterior Estimation | SBIBM Bernoulli GLM | Joint C2ST66 | 6 | |
| Posterior Estimation | solar_dynamo | Joint C2ST0.868 | 6 | |
| Posterior Estimation | ou | Joint C2ST0.866 | 6 |