TabMGP: Martingale Posterior with TabPFN
Kenyon Ng, Edwin Fong, David Frazier, Jeremias Knoblauch, Susan Wei
Abstract
Bayesian inference provides principled uncertainty quantification but is often limited by the challenges of prior and likelihood elicitation. The martingale posterior (MGP) (Fong et al., 2023) offers an alternative by replacing these requirements with a predictive rule. In addition, the MGP focuses inference on parameters defined through a loss function. This framework is especially resonant in the era of foundation transformers; practitioners increasingly leverage models like TabPFN for their state-of-the-art capabilities, yet often require epistemic uncertainty for a scientific estimand that need not parameterise the implicit latent model. The MGP provides a mechanism to recover these posterior distributions. We introduce TabMGP, an MGP built on TabPFN for tabular data. TabMGP produces credible sets with near-nominal coverage and often outperforms both handcrafted MGP constructions and standard Bayesian baselines.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext cbfe371f-ad00-476c-bbba-292bef630f18Builds on5
- Transformers Can Do Bayesian InferenceSamuel Müller, Noah Hollmann, Sebastian Pineda-Arango, Josif Grabocka et al.ICLR 2022 · 287 citations
- Is In-Context Learning in Large Language Models Bayesian? A Martingale PerspectiveFabian Falck, Ziyu Wang, Christopher C. HolmesICML 2024 · 46 citations
- A Rigorous Link between Deep Ensembles and (Variational) Bayesian MethodsVeit David Wild, Sahra Ghalebikesabi, Dino Sejdinovic, Jeremias KnoblauchNeurIPS 2023 · 40 citations
- Quasi-Bayes meets VinesDavid Huk, Yuanhe Zhang, Ritabrata Dutta, Mark SteelNeurIPS 2024 · 6 citations
- Martingale Posterior Neural ProcessesHyungi Lee, Eunggu Yun, Giung Nam, Edwin Fong et al.ICLR 2023
Related papers
- Effortless, Simulation-Efficient Bayesian Inference using Tabular Foundation ModelsJulius Vetter, Manuel Glöckler, Daniel Gedon, Jakob H. MackeNeurIPS 2025 · 14 citations
- Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-LearningAnish Dhir, Cristiana Diaconu, Valentinian Lungu, James Requeima et al.NeurIPS 2025 · 16 citations
- Using maximal information auxiliary variables to improve synthetic data generation based on TabPFN foundation modelsElias Chaibub NetoICLR 2026
- Foundation Models for Causal Inference via Prior-Data Fitted NetworksYuchen Ma, Dennis Frauen, Emil Javurek, Stefan FeuerriegelICLR 2026 · 37 citations
- FIRE: Multi-fidelity Regression with Distribution-conditioned In-context Learning using Tabular Foundation ModelsRosen Yu, Nicholas Sung, Faez AhmedICML 2026 · 1 citation
