← Search

Yee-Whye Teh

4 accepted papers

2026

From Backward Spreading to Forward Replay: Revisiting Target Construction in LLM Parameter Editing

ICML 2026poster

LLM parameter editing methods commonly rely on computing an ideal target hidden-state at a target layer (referred as anchor point) and distributing the target vector to multiple preceding layers (commonly known as backward spreading) for cooperative editing. Although widely used for a long time, its…

Cited by 0SourceScholar
2026

Meta Flow Maps enable scalable reward alignment

ICML 2026poster

Controlling generative models—whether via inference-time steering or fine-tuning—is expensive. Control relies on estimating the value function—typically necessitating costly trajectory simulations. To eliminate this bottleneck, we introduce *Meta Flow Maps (MFMs)*, stochastic extensions of consisten…

Cited by 0SourceScholar
2026

Variational Flow Maps: Make Some Noise for One-Step Conditional Generation

ICML 2026poster

Flow maps enable high-quality image generation in a single forward pass. However, unlike iterative diffusion models, their lack of an explicit sampling trajectory impedes incorporating external constraints for conditional generation and solving inverse problems. We put forth _Variational Flow Maps_,…

Cited by 0SourceScholar
2016

DR-ABC: Approximate Bayesian Computation with Kernel-Based Distribution Regression

ICML 2016poster

Performing exact posterior inference in complex generative models is often difficult or impossible due to an expensive to evaluate or intractable likelihood function. Approximate Bayesian computation (ABC) is an inference framework that constructs an approximation to the true likelihood based on the…