AI for science needs calibrated models and uncertainty, but we show standard test-time diffusion guidance methods miss the Bayesian posterior. Our Calibrated Bayesian Guidance (CBG) fixes this, enabling Bayesian inference and setting a new SOTA for black-hole imaging.
Felix Draxler
@drrelax.bsky.social
Postdoc @ UCI with Stephan Mandt. PhD @ Heidelberg University. Generative models, Parallel Token Prediction, Free-Form Flows, Normalizing Flows
LLMs are autoregressive and slow? No! Parallel Token Prediction decodes multiple consistent tokens in one model call. PTP allows arbitrary dependencies in one call, unlike discrete diffusion. Practical: 2.4x speedup github.com/mandt-lab/ptp ICLR: Apr 23, morning poster P3-#608