Josh McClellan (@JoshMcClellan0) profile picture

I research generalization and multi-agent systems in reinforcement learning

Joined April 2020
Photo by Josh McClellan (@JoshMcClellan0)
Josh McClellan retweeted
ICLR 2025 just gave an Outstanding Paper Award to a method that fixes model editing with one line of code 🤯 here's the problem it solves: llms store facts in their parameters. sometimes those facts are wrong or outdated. "model editing" lets you surgically update specific facts without retraining the whole model. the standard approach: find which parameters encode the fact (using causal tracing), then nudge those parameters to store the new fact. works great for one edit. but do it a hundred times in sequence and the model starts forgetting everything else. do it a thousand times and it degenerates into repetitive gibberish. every edit that inserts new knowledge corrupts old knowledge. you're playing whack-a-mole with the model's memory. AlphaEdit reframes the problem. instead of asking "how do we update knowledge with less damage?" the authors ask "how do we make edits mathematically invisible to preserved knowledge?" the trick: before applying any parameter change, project it onto the null space of the preserved knowledge matrix. in plain english: find the directions in parameter space where you can move freely without affecting anything the model already knows. only move in those directions. it's like remodeling one room in a house by only touching walls that aren't load-bearing. the rest of the structure doesn't even know anything changed. the results from Fang et al. across GPT2-XL, GPT-J, and LLaMA3-8B: > average 36.7% improvement over existing editing methods > works as a plug-and-play addition to MEMIT, ROME, and others > models maintain 98.48% of general capabilities after 3,000 sequential edits > prevents the gibberish collapse that kills other methods at scale and the implementation is literally one line of code added to existing pipelines. what i find genuinely elegant: the paper proves mathematically that output remains unchanged when querying preserved knowledge. this isn't "it works better in practice." it's "we can prove it doesn't touch what it shouldn't." the honest caveats: largest model tested was LLaMA3-8B. nobody's shown this works at 70B+ scale yet. a follow-up paper (AlphaEdit+) flagged brittleness when new knowledge directly conflicts with preserved knowledge, which is exactly the hardest case in production. and the whole approach assumes causal tracing correctly identifies where facts live, which isn't always clean. but as a core insight, this is the kind of work that deserves the award. not because it solves everything. because it changes the question. the era of "edit and pray" for llm knowledge updates might actually be ending.
19
78
4
608
51,292
⏰⏰ New Science of Robot Learning Paper: "Much Ado About Noising." TL;DR we answer why generative models, like flow and diffusion models, actually work for robotic control tasks🤖🤖 (hint: its not multimodality). This leads to a new minimal iterative policy (MIP) that matches flow models with much faster inference🚄🚄 Check out @ChaoyiPan 's thread and paper to find out more. Amazing work by @ChaoyiPan, together with @GuanyaShi , @nmboffi , @guannanqu. Come find us at NeurIPS to chat more!
Generative models (diffusion/flow) are taking over robotics 🤖. But do we really need to model the full action distribution to control a robot? We suspected the success of Generative Control Policies (GCPs) might be "Much Ado About Noising." We rigorously tested the myths. 🧵👇
4
27
1
199
19,806
Josh McClellan retweeted
I still don't believe that Spatial AI is a big data problem. Massive data won't unlock robotics. Representation is still the hard bit. We need efficient, composable representation of 3D physical scenes to enable mental simulation like humans use to plan creative uses of objects.
18
48
4
353
43,981
Josh McClellan retweeted
My team at Boston Dynamics has open Research Scientist positions! We are hiring full-time and interns. If you are excited about RL & humanoids, we want to get to know you! Research Scientist: bostondynamics.wd1.myworkday… Research Scientist Intern: bostondynamics.wd1.myworkday…
8
34
314
34,432
Josh McClellan retweeted
Much of the field obsesses over end-to-end learning. But strong generalization requires compositionality: building modular, reusable abstractions, and reassembling them on the fly when faced with novelty. The models of the future won't be just pipes, they will be Lego castles.
58
160
25
1,295
136,996
I'm excited to share that our paper, "Boosting Sample Efficiency and Generalization in Multi-agent Reinforcement Learning via Equivariance," has been accepted to NeurIPS 2024! 🎉 arxiv.org/abs/2410.02581 @ptokekar @furongh
2
1
2
5
1,202
This robustness stems directly from its symmetry guarantees, allowing it to lose less performance when adapting to new scenarios. If you'll be at Neurips come visit our poster next week to learn more and discuss the exciting future of MARL!
1
62
If you work for a US university, you have probably noticed the rollout of strict new policies mandating disclosures and approvals for funding, consulting, and COIs, and also threats of legal action for non-compliance. Here’s why this is happening now 🧵
2
24
7
103
57,532
Josh McClellan retweeted
Why do latest language transformers (LLMs like ChatGPT etc.) use reinforcement learning (RL) for finetuning instead of regular supervised learning (SL)? There are at least 5 reasons ... [1/10]
32
301
20
1,947
506,848
Josh McClellan retweeted
I remember NeurIPS 2016 (it was called NIPS back then). Deep RL was the big thing. Deep RL everywhere. Every conversation I had there was about how Deep RL trained in environments like OpenAI Universe (remember that) was going to lead to AGI within 5 to 10 years.
I was not concerned about AI last week and I am concerned about AI today. Not in the "they're going to kill us" way, but in the "societies can adapt pretty quickly, but maybe not quite /this/ quickly" kind of way.
26
150
57
992
Josh McClellan retweeted
All the best examples of ChatGPT, from OpenAI:
361
2,780
564
13,615
If you want others to care about what you think, then start by caring yourself. Get a notebook, write your thoughts down, challenge them, and develop them into something worth sharing.
10
86
7
925
Josh McClellan retweeted
Read a bit about Grokking recently. Here's some learnings: "Grokking" is a curious neural net behavior observed ~1 year ago (arxiv.org/abs/2201.02177). Continue optimizing a model long after perfect training accuracy and it suddenly generalizes. Figure:
37
297
65
2,026