5 papers
TabMGP: Martingale Posterior with TabPFN
Kenyon Ng, Edwin Fong, David T. Frazier +2
Bayesian inference provides principled uncertainty quantification but is often limited by the challenges of prior and likelihood elicitation. The martingale posterior (MGP) (Fong e…
PFN-TS: Thompson Sampling for Contextual Bandits via Prior-Data Fitted Networks
Yan Shuo Tan, Kenyon Ng, Ruizhe Deng +3
Thompson sampling is a widely used strategy for contextual bandits: at each round, it samples a reward function from a Bayesian posterior and acts greedily under that sample. Prior…
Uncertainty Decomposition for Bayes-Filtered Transformers via Bayesian Predictive Inference
Sandra Fortini, Kenyon Ng, Sonia Petrone +2
Bayes-filtered transformers are transformers meta-learned on sequences from a prior predictive distribution to approximate the corresponding posterior predictive distribution. They…
Temperature Optimization for Bayesian Deep Learning
Kenyon Ng, Chris van der Heide, Liam Hodgkinson +1
The Cold Posterior Effect (CPE) is a phenomenon in Bayesian Deep Learning (BDL), where tempering the posterior to a cold temperature often improves the predictive performance of th…
Pathwise Gradient Variance Reduction with Control Variates in Variational Inference
Kenyon Ng, Susan Wei
Variational inference in Bayesian deep learning often involves computing the gradient of an expectation that lacks a closed-form solution. In these cases, pathwise and score-functi…