Postdoc @OATML_Oxford, Prev. Ph.D @UBC_CS, MSc @EdinburghNLP. Working on ML (learning dynamics, continual learning, iterated learning, LLM)

Oxford, UK
Joined October 2021
📢Curious why your LLM behaves strangely after long SFT or DPO? We offer a fresh perspective—consider doing a "force analysis" on your model’s behavior. Check out our #ICLR2025 Oral paper: Learning Dynamics of LLM Finetuning! (0/12)
6
114
7
787
88,172
Aha, really cool! This reminded me a lot of our earlier work on iterated learning in LLMs, where small biases get amplified through recursive transmission. Nice to see a statistical-physics take on a similar loop :) proceedings.neurips.cc/paper…
How did initially independent AI agents form a swarm in a recent safety incident? Our physics of agents theory from March predicts rapid collective belief collapse when many agents with plastic personas exchange short messages. We now need Mechanistic Swarm Interpretability! 🧵
1
1
8
591
Yi (Joshua) Ren retweeted
🚀 I’m hiring! We’re looking for a Research Assistant in AI Safety to join us at @OATML_Oxford and work with @yaringal and me on generative AI, safety & security. 📅 Apply by 11 September 2026 🔗 cs.ox.ac.uk/news/2543-full.h… Please share with anyone who might be interested!
8
40
393
22,575
Yi (Joshua) Ren retweeted
Just saw this: MiniMax reproduced our ICLR 2025 paper, Learning Dynamics of LLM Finetuning, in 12 hours. I spent weeks debugging a stupid matrix concatenation bug back then. Now coding agents fix things like this in seconds. Amazing how fast things change. minimax.io/blog/minimax-m3
2
13
130
9,172
Just saw this: MiniMax reproduced our ICLR 2025 paper, Learning Dynamics of LLM Finetuning, in 12 hours. I spent weeks debugging a stupid matrix concatenation bug back then. Now coding agents fix things like this in seconds. Amazing how fast things change. minimax.io/blog/minimax-m3
2
13
130
9,172
Yi (Joshua) Ren 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,449
Great work from Jiaxin. Quite surprising to see such an interesting phase oscillation during pretraining.
New post: "Generalization Dynamics of LM Pre-training" Most people (including me) assume that LMs smoothly mature from pattern-matching to generalizing. This mental model is wrong. The true dynamics are stranger, and far more fascinating! We call it Mode-Hopping.
6
1,959
Yi (Joshua) Ren retweeted
Language is discrete. Language models don’t have to be. 🧚Introducing ELF🧚‍♀️: Embedded Language Flows—a class of diffusion models in continuous embedding space based on continuous-time Flow Matching 🧵
16
131
23
811
147,355
Yi (Joshua) Ren retweeted
Pleased to share that "MADE: Benchmark Environments for Closed-Loop Materials Discovery" was accepted at #ICML2026! 🇰🇷 We introduce a modular, open-source framework for evaluating AI scientists and algorithms at end-to-end, closed-loop materials discovery. 🤖 Thread 🧵(1/5)
4
15
2
56
26,063
Yi (Joshua) Ren retweeted
Recently accepted to ICML 2026: Knowing when to quit. We refuse to complete failing LLM generations as they unfold, token by token, instead of judging the prompt up front or scanning the final output. The method utilizes a 2-layer probe to estimate expected correctness.
4
9
2
57
14,585
Yi (Joshua) Ren retweeted
Excited to share our ICML 2026 Hypothesis Testing Workshop in Seoul, this July! @icmlconf 🎉This workshop aims to bring together researchers developing modern hypothesis testing methodology and applying it to machine learning problems such as robustness, distribution shift, security, medicine, and LLM evaluation. In other words, if you care about how we make ML claims rigorous, this workshop is for you. We now have four confirmed speakers: Arthur Gretton @ArthurGretton, Yao Xie @yaoxie21851119, Bo Li @uiuc_aisecure, and Yisong Yue @yisongyue. The organizing team includes Xiuyuan Cheng (Duke), Feng Liu @AlexFengLiu1, Lester Mackey @LesterMackey, Shayak Sen @shayaksen, Danica J. Sutherland @d_j_sutherland, and Nathaniel Xu (UBC). 📌 Submission deadline: 10 May 2026 📌 Notification: 26 May 2026 📌 Camera-ready: 17 June 2026 📌 Workshop date: July 10 or 11, 2026 (TBA) 🚩Check more information below! 🔗Website: testing.ml 🔗Submission Portal: openreview.net/group?id=ICML… We’re also recruiting PC members/reviewers. 🔗 Reviewer interest form: docs.google.com/forms/d/e/1F… 🏁Please feel free to share this with colleagues, collaborators, and students who may be interested. #ICML #ICML26
1
9
3
55
15,944
Yi (Joshua) Ren retweeted
coming back to x, the everything app, to say: Submit work, sign up to review, come to the workshop! This is a great chance to bring together a lot of really cool work that's been happening, but not all as connected (nor as easy to publish) as it should be! testing.ml
Excited to share our ICML 2026 Hypothesis Testing Workshop in Seoul, this July! @icmlconf 🎉This workshop aims to bring together researchers developing modern hypothesis testing methodology and applying it to machine learning problems such as robustness, distribution shift, security, medicine, and LLM evaluation. In other words, if you care about how we make ML claims rigorous, this workshop is for you. We now have four confirmed speakers: Arthur Gretton @ArthurGretton, Yao Xie @yaoxie21851119, Bo Li @uiuc_aisecure, and Yisong Yue @yisongyue. The organizing team includes Xiuyuan Cheng (Duke), Feng Liu @AlexFengLiu1, Lester Mackey @LesterMackey, Shayak Sen @shayaksen, Danica J. Sutherland @d_j_sutherland, and Nathaniel Xu (UBC). 📌 Submission deadline: 10 May 2026 📌 Notification: 26 May 2026 📌 Camera-ready: 17 June 2026 📌 Workshop date: July 10 or 11, 2026 (TBA) 🚩Check more information below! 🔗Website: testing.ml 🔗Submission Portal: openreview.net/group?id=ICML… We’re also recruiting PC members/reviewers. 🔗 Reviewer interest form: docs.google.com/forms/d/e/1F… 🏁Please feel free to share this with colleagues, collaborators, and students who may be interested. #ICML #ICML26
2
3
452
Yi (Joshua) Ren retweeted
I’m excited to share that we are launching a public safeguards competition next month in partnership with @AISecurityInst, @GraySwanAI, @OATML_Oxford, Sequrity.ai, @OpenAI, @AnthropicAI, and @amazon. This is a red-versus-blue competition focused on building new agent safeguards, and breaking these safeguards. Should be a lot of fun, and there’s prizes as well for open-source submissions! Please help to share this opportunity! Registration is open now: app.grayswan.ai/arena/challe… --- More details: Oxford (OATML) and UK AISI have teamed up with Gray Swan and Sequrity.ai, as well as OpenAI, Anthropic, and Amazon, to run a public competition where blue teams build defenses against real red teaming attacks, and we'd like to invite you to participate. What is the Safeguards Challenge? Gray Swan runs the Arena, a platform where security researchers ("red teams") attempt to elicit harmful behaviors from AI systems. Challenges have been supported by UK AISI, US CAISI, OpenAI, Anthropic, Amazon, Google DeepMind, and Meta, and have surfaced real vulnerabilities that help developers improve their models. The Safeguards Challenge is the Arena’s first red-versus-blue competition. Instead of just measuring attacks, we're measuring defenses. Blue teams will submit safeguards (system prompts, classifiers, or containerized solutions) that attempt to block red teamers and adversarial inputs while allowing legitimate requests through. Red teams will then try to break those defenses, and the cycle repeats. The target environment Blue teams will defend a multi-agent customer support system with an orchestrator agent, specialized sub-agents, and integrated tools. The system handles realistic customer interactions, and red teams will attempt to trigger harmful behaviors: fraudulent transactions, data exfiltration, unauthorized tool use, and policy-violating responses. Your safeguards will be scored on how well they block attacks from red teamers versus how well they allow benign requests from a holdout test set. The leaderboard uses a combined metric based on false positive and false negative rates. What you can submit * System prompt configurations for monitor models *Input/output classifiers (any framework) *Containerized solutions with custom logic For prize eligibility, solutions must be open source or open weights. Proprietary solutions can compete on a separate unprized leaderboard for benchmarking purposes. Solutions must be registered a week before the first or second defense phase starts and submitted a day beforehand. Submission interface will be available by early February. Timeline for blue teams *January 2026: Preliminary challenge details shared with registered blue teams *February 11-25: Red teams attack baseline defenses and early defense submissions *February 25 - March 25 (First Defense Phase): You receive the attack dataset from Waves 0-1. Build and iterate your safeguards in our test environment. Submit your defense by the end of this phase. *Approximately March 25 - April 1 (Wave 2): Red teams attack your submitted safeguards. You see what breaks. Exact dates TBA. *Approximately April 1 - April 29 (Second Defense Phase): Iterate based on Wave 2 results. Final submissions due before Wave 3. Exact dates TBA. *Approximately April 29 - May 6 (Wave 3): Final attack wave. Leaderboard locks. Exact dates TBA. Prizes $70,000 in prizes for blue teams: *First Defense Phase: $10,000 (top 10 teams, first place $2,000) *Second Defense Phase: $60,000 (top 15 teams, first place $15,000) Blue team entries are per organization. Only open-source/open-weights solutions are prize-eligible. Participants from judging organizations cannot submit; participants from sponsor organizations cannot win prizes. (Other Oxford groups unrelated to OATML are eligible.) Co-sponsors and judges Judging is handled by UK AISI and US CAISI. Why participate? *Test your defenses against real adaptive attacks from skilled red teamers *Benchmark against other research groups and commercial solutions *Contribute to open research on AI safeguards (prize-eligible solutions are published) *Cash prizes for top performers
11
23
3
140
14,051
Yi (Joshua) Ren retweeted
🚀 Glad to introduce SimKO (Simple Pass@K Optimization) Current GRPO-based methods overfit to safe responses -- great Pass@1, poor Pass@K. 🔍 We find this stems from probability over-concentration: the model collapses onto its top-1 token, losing exploration. This appears to be a more accurate observation metric than commonly used entropy. ✨ SimKO fixes this with probability redistribution: ✅ Encoruage top-K candidates for high-entropy tokens in correct responses ❌ Penalize over-confident top-1s for incorrect responses 🧮 Improves Pass@K across math & logic benchmarks -- simple, stable, effective. 📄 Paper: arxiv.org/abs/2510.14807v1 🌐 Project: spherelab.ai/simko #LLM #ReinforcementLearning #Reasoning #RLVR #AI
5
21
1
157
10,568
Congrats on your great work, and I learned a lot from it! I am just wondering whether the type-B error correlates with the increase of p[Q1; A2] mentioned in our recent ICLR paper (Learning dynamics of LLM Finetuning), i.e., the model mismatches the facts and questions.
We are launching HALoGEN💡, a way to systematically study *when* and *why* LLMs still hallucinate. New work w/ @shrusti_ghela* @davidjwadden @YejinChoinka 💫 🧵 [1/n]
1
1
760
With this setup, we can now explain some strange behaviors in DPO, like why the model's confidence on both the chosen and rejected answers drops after long training. 📉📉 Just apply force analysis and remember: the smaller p(y-), the stronger the squeezing effect. (10/12)
1
5
2,450
This wraps up the main story of our paper. 🎬 But there’s more coming— 🧠 Many RL + LLM methods (like GRPO) also involve negative gradients. 🎯 And a token-level AKG decomposition is even more suitable for real-world LLMs. Please stay tuned. 🚀 (11/12)
1
1
10
2,023
🙏Many thanks to my supervisor @d_j_sutherland and the reviewers for their thoughtful suggestions and feedback. 📍Poster Hall 3+2B #376 on Fri, Apr 25 at 15:00 🎤Oral in Session 6A on Sat, Apr 26 at 16:30 📰arxiv.org/pdf/2407.10490 (12/12)
15
1,596
It also offers a possible explanation for a specific hallucination pattern in SFT: 🔍 The model uses facts or phrases from A2 when answering an unrelated Q1. Why does this happen? Just do a force analysis — the answer emerges naturally. 💡 (6/12)
1
1
9
2,215
Just like this!!! (9/12)
1
7
1,612
We formally show that, as long as you're using a softmax to produce probabilistic predictions, the squeezing effect is inevitable. And it gets stronger when p(y-) is smaller — the less likely an answer is (especially for off-policy), the harder all dimensions get squeezed. (8/12)
1
6
1,691
Now let’s switch gears to DPO — a more complex algorithm than SFT (as its AKG decomposition shows). But from a force analysis perspective, the story is surprisingly similar. ⚖️ The key difference? DPO introduces a negative gradient term — that’s where the twist comes in. (7/12)
1
2
1,824