Find me @yeewhye@sigmoid.social Professor at @OxCSML, @oxfordstats and Research Director at @GoogleDeepMind. All opinions are my own.

Oxford, England
Joined January 2017
Reposting this again... I think this is *the way* to do inference-time steering/guidance/conditioning of diffusion models, because meta flow maps are able to produce much better estimates of the required guidance term. Check it out!
✨🚀Reward alignment just hit a major milestone🚀✨ Controlling generative models is usually bottlenecked by expensive trajectory rollouts 🐢🐌 or massive particle ensembles 🐢🐌 Meta Flow Maps (MFMs) eliminate both We extend consistency models and flow maps into the stochastic regime—enabling efficient generative control Is that a text-to-image model you see below? Nope. It’s just ImageNet with inference-time steering 🤯🤯🤯 Want unbiased off-policy fine-tuning ? We got that too 📜arxiv.org/abs/2601.14430 🌐meta-flow-maps.github.io 🧵1/10
1
9
83
17,366
WTF?! 🤯 RL fine-tuning for flow models, without simulating rollouts. Wasserstein-Tilted Flow Maps move each sample toward higher reward instead of reweighting them. ⚡ 1 step beats 50 steps of the base model 📉 up to 280x less training compute NeurIPS 2026 Spotlight 🧵
#NeurIPS2026 Spotlight RL for diffusion and flow models often requires expensive rollouts. We show that flow maps enable simulation-free RL finetuning. Introducing Wasserstein-Tilted Flow Maps (WTF)! 🤩 The key is a transport-based alternative to KL regularization. 🧵
3
5
1
33
2,960
Yee Whye Teh retweeted
#NeurIPS2026 Spotlight RL for diffusion and flow models often requires expensive rollouts. We show that flow maps enable simulation-free RL finetuning. Introducing Wasserstein-Tilted Flow Maps (WTF)! 🤩 The key is a transport-based alternative to KL regularization. 🧵
7
47
5
304
22,486
RT @harit_v: Excited to be in SF and Berkeley next week. Will be at, #COLM2026, Trustworthy AI workshop at the Simons Institute, Fronti…
1
153
This is super cool work by @AbbasMammadov11 et al. Transporting samples towards high reward regions instead of reweighting makes more use of the geometric structure of the sample space!
WTF?! Simulation-Free Reinforcement Learning with Wasserstein-Tilted Flow Maps Abbas Mammadov, Jerry Y. Huang, Justin Lin, Partha Kaushik, Sheel Shah, Kartik Nair, Yee Whye Teh, Nicholas M. Boffi arxiv.org/abs/2609.27033 [𝚌𝚜.𝙻𝙶 𝚌𝚜.𝙲𝚅 𝚜𝚝𝚊𝚝.𝙼𝙻]
2
11
129
14,154
Yee Whye Teh retweeted
Happy to share that our work on "Don’t Be Choosy: Scoring over Choosing for Verbalized LLM Confidence", led by @harit_v with @yeewhye, was accepted at #UncertaiNLP! I'll be presenting it in Budapest at @emnlpmeeting in late October + attending @Climate_NLP 🇭🇺
1
4
534
Yee Whye Teh retweeted
Paperena evaluates AI scientists over realistic research cycles: writing, reviewing, and revising. It measures factuality, reproducibility, novelty, and community behavior. From @harit_v and @paperena_ai, one of 25+posters at Frontier Data Summit. frontierdatasummit.ai/
6
2
19
3,068
Whether we like it or not, AI tools are being used all across the research process. It’s important that we are able to study their capabilities, and build a world in which they help rather than hinder the scientific community. @paperena_ai is a first step in doing so, join us!
Overwhelmed with the number of ICLR submissions and the projected growth to millions of papers in next few years? The paper mill is there, but is it any good? can this be aligned to what we care? Join us at the Frontier Data Summit to see how @paperena_ai can help. TL;DR: Paperena is a platform for recording, evaluating and disseminating research primarily done by AI. A home for AI papers and community of AI scientists. More details and initial version coming soon! There is plenty more charted out. Let's chat and build!
1
6
42
6,243
Yee Whye Teh retweeted
🧬Fully Understand Evolution Strategies (ES) for LLM Reasoning ES can post-train LLMs without backprop. Our study finds that ES: 🔍 Explores higher Pass@K than GRPO 💥 Still sparse functional updates, with no catastrophic forgetting. ⚙️ Requires fewer samples as LLMs get larger We summarize the paper through 3 research questions 👇 1️⃣ Does ES exhibit the same post-training characteristics as GRPO? ES maintains broader reasoning coverage! Its population-based parameter-space exploration induces diverse reasoning behaviors: ES improves Pass@1 while achieving stronger large-K Pass@K, without the same entropy collapse observed in GRPO. The two are also complementary. 🔀 Sequential compositions—ES→GRPO and GRPO→ES—combine GRPO’s strength in Pass@1 with ES’s gains in Pass@K, yielding better Pass@1–Pass@K trade-offs. ➡️ ES preserves broader coverage; combining it with GRPO can capture both strengths. 2️⃣ Does ES necessarily cause catastrophic forgetting? No. Although ES induces substantial whole-model parameter drift, its task-relevant effects are concentrated in a small subset of larger-magnitude updates. Most parameter changes contribute little after perturbation cancellation, revealing surprising functional sparsity. Held-out capabilities also remain largely preserved under appropriate training settings. ➡️ Large parameter movement does not necessarily imply widespread functional change—or catastrophic forgetting. 3️⃣ What hyperparameter settings and estimators make ES effective and scalable? We find three practical takeaways: ✅ Z-score reward normalization is a key ingredient for effective ES training. ✅ Larger pretrained models require smaller population sizes for effective optimization. ✅ For discrete reasoning rewards, the two-point estimator commonly favored in zeroth-order SFT provides no advantage. 💡 Takeaway: ES is not merely a memory-efficient, gradient-free alternative to GRPO. Its broader reasoning coverage, functionally sparse updates, and distinct scaling behavior position it as a distinct paradigm for LLM reasoning post-training. 📄 arXiv:2608.27351
4
19
3
118
18,005
Great work by @zhengzhi20 . Feels like a much better approach tofor agentic and long horizon tasks than RL!
🚀 Will Agentic ESOpt Be the Future of Long-Horizon Agentic Fine-Tuning? Excited to share our new work on Arxiv. Agentic ESOpt: Fine-Tuning Long-Horizon LLM Agents with Minimal GPU Memory Requirements As LLM agents become longer-horizon and more branching, conventional Agentic RL faces three growing bottlenecks: high training/memory cost and increasingly challenging long-horizon credit assignment. Our Agentic ESOpt approaches agent fine-tuning from the parameter space instead, targeting three advantages: ⚡ Model Scalability — only minimal inference-level GPU memory requirement 🔧 Flexibility — prompt/skill–parameter co-evolution ⏳ Long-Horizon Scalability — trajectory-level parameter attribution without horizon-wise decomposition There are impressive empirical results, and we believe ES may be a better-matched parameter fine-tuning paradigm as agents become larger and longer-horizon. 📄 Paper: arxiv.org/abs/2608.17310 🌐 Project: zz1358m.github.io/Project-Ag… 💻 GitHub: github.com/zz1358m/Agentic-E… 🤗 Models & Checkpoints: huggingface.co/collections/z… Code, training logs, and checkpoints have been fully open-sourced. Main takeaways ↓
4
1
21
4,569
Yee Whye Teh retweeted
🚀 Will Agentic ESOpt Be the Future of Long-Horizon Agentic Fine-Tuning? Excited to share our new work on Arxiv. Agentic ESOpt: Fine-Tuning Long-Horizon LLM Agents with Minimal GPU Memory Requirements As LLM agents become longer-horizon and more branching, conventional Agentic RL faces three growing bottlenecks: high training/memory cost and increasingly challenging long-horizon credit assignment. Our Agentic ESOpt approaches agent fine-tuning from the parameter space instead, targeting three advantages: ⚡ Model Scalability — only minimal inference-level GPU memory requirement 🔧 Flexibility — prompt/skill–parameter co-evolution ⏳ Long-Horizon Scalability — trajectory-level parameter attribution without horizon-wise decomposition There are impressive empirical results, and we believe ES may be a better-matched parameter fine-tuning paradigm as agents become larger and longer-horizon. 📄 Paper: arxiv.org/abs/2608.17310 🌐 Project: zz1358m.github.io/Project-Ag… 💻 GitHub: github.com/zz1358m/Agentic-E… 🤗 Models & Checkpoints: huggingface.co/collections/z… Code, training logs, and checkpoints have been fully open-sourced. Main takeaways ↓
6
21
4
69
34,532
Yee Whye Teh retweeted
Day 1⃣ of our @agents4academia hackathon, we asked the participants: what are the most frustrating parts of your workflow? Almost every single one: keeping up with the literature. The firehose is impossible. The trust is breaking (👋AI slop). So we built something. Meet Prior 🧵
1
4
11
1,446
Yee Whye Teh retweeted
We have been building @paperena_ai. A platform to power the evaluation layer for science in the AGI era. Excited to talk about it at the AI scientist summer workshop, today! I'll be presenting remotely, but don't forget to catch @klarakaleb at the workshop if you are there. paperena.ai
An incredibly packed room for our morning speakers Nathan Frey @nc_frey from Anthropic and Vivek Natarajan @vivnat from Google DeepMind at the AI Scientist Summer Workshop. We still have lots of fantastic speakers throughout the day! Checkout the schedule at ai-scientist-workshop.github…
10
17
2,059
I had great fun hacking up software and agents with Claude and the teams across @OxfordStats @NUSComputing and @NTUsg , and peaking into the wild future of academic research. Thanks @AnthropicAI !
First up a hackathon: 20+ researchers · two weeks · @OxfordStats × @NUSComputing × @NTUsg and tokens on tap by @AnthropicAI . Outcome: Five open‑source AI agents solving common workflow frustrations across the whole research lifecycle. 🧵
3
1
15
4,345
And Thanks to @harit_v and @klarakaleb for doing much of the organisation!
5
619
Yee Whye Teh retweeted
1/ 🧵 Decoding-time safety methods improve LLM safety, but they rewrite *safe* generations too, an "alignment tax" on helpfulness. Value-Filtered Decoding steers only when needed, with a calibrated bound on how often we touch safe outputs. 📄Accepted at AI4Good @ ICML 2026✨
3
11
46
12,437
Very nice work by @AbbasMammadov11 diving into linear inverse problems with diffusion models, resulting in a simple yet SOTA method that illuminates what's going on.
🚀 New paper! Exact Posterior Score (EPS) 📢 Oral @ ICML 2026 SPIGM! Posterior sampling for linear inverse problems is STILL just denoising - you only have to query the denoiser at the right place. No likelihood gradients. No projections. 📄 arxiv.org/abs/2606.17048 🧵1/11
2
6
52
9,265
Yee Whye Teh retweeted
[📄preprint] Diffusion models 🤝 MCMC ! Diffusion model samplers are biased due to discretisation 💡The fix: Metropolis-type adjustment on corrector steps ❗️Challenge: no access to the density ratio, only the score 🔑Insight: the score (and some maths) is all you need... [1/3]
5
50
348
20,893
🔊 Make Some Noise! Introducing Variational Flow Maps (VFM) - a principled framework for one-step conditional image generation & reward alignment. No iterative guidance. No hundreds of NFEs. Just learn the right noise 🎯 🚀 Below are one-step generation results from VFM after only 0.5 epoch of reward fine-tuning on ImageNet ⚡ Paper: arxiv.org/abs/2603.07276 Code: github.com/abbasmammadov/VFM 🧵1/10
2
20
2
133
13,032
Yee Whye Teh retweeted
📍 Can LLMs discover, abstract, and reuse higher-level tool skills across tasks? Existing tool-use benchmarks test solving tasks with fixed tools. But real workflows contain recurring structures where efficiency comes from reusable tool compositions, not isolated calls. We introduce SkillCraft: 126 tasks across 6 domains designed to test whether LLM agents can acquire compositional skills, not just call atomic tools. We also propose Skill Mode, a lightweight protocol with four MCP primitives that let agents compose, verify, cache, and reuse tool chains at test time. Our Key findings across evaluating 8 SOTA models: ⚡Skill Mode enables agents to self-discover and reuse skills, leading to higher success and efficiency than agents without it. The gains are larger for stronger models. 🧠 Stronger models (e.g., Claude) discover more generalizable skills, which transfer across tasks and even across models. 🔍 Deeper composition ≠ better — shallow, well-tested skills generalize best. 🔗 Paper: arxiv.org/abs/2603.00718 💻 Code: github.com/shiqichen17/Skill… 🏠 Page: skillcraft-website.github.io… (1/7)
9
39
12
205
75,019