cs.LGMay 8, 2026

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

Authors: Elliot PickensChiraag GohelSidharth Satya

Organizations: University of Pennsylvania · The George Washington University · Brown University

Abstract

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.

Explore similar work

CardsList
  1. Tokenised Flow Matching for Hierarchical Simulation Based Inference

    Apr 22, 2026Giovanni Charles, Cosmo Santoni, Seth Flaxman +1PosteriorBayesian

  2. TabPFN-3.5: Technical Report

    Sep 15, 2026Benjamin Jäger, Nick Erickson, Léo Grinsztajn +44Tabular Foundation ModelsReal-Time Inference