Top
Best
New

Posted by dbreunig 1 hour ago

Astra can read and write weights to improve Qwen(rikiyat.github.io)
515 points | 147 comments
vessenes 46 minutes ago|
This is insane.

Astra can, without tooling, determine what games small networks are trained on, can edit the weights of networks directly to improve outcomes, and can hand edit a small qwen model to improve its math skills by like 5x.

I'm shook. If you'd asked me this morning whether LLMs have native understanding of their weights that is less than human <-> neuron understanding, the same as human or more than human, I would have said "same to less" depending on definitions.

No human can go edit a brain to become better at math. This is just very, very surprising to me and extremely interesting. I would guess this upsets quite a number of interp theories and philosophical takes on what's going on with these networks.

l9o 1 minute ago||
I think the "taught qwen math" part is not exactly correct though, even if the post makes it sound like that. Qwen already knew math, but it was bugging out and getting into an infinite loop when it tried to solve the problems (85 of its first 96 answers hit the token limit). With numpy and pytorch, Astra found where the loop was happening and fixed it.

They also tried the same with Gemma 3 270M, which doesn't loop and is just bad at math, and Astra couldn't improve it. Still impressive, but it's more in the realm of what I'd expect, given labs are probably training their models on ML research and interpretability work.

augment_me 5 minutes ago|||
The math example on Qwen was based on 50 batched iterations, so the model was able to evaluate it's changes on downstream tasks. And the fix was essentially to remove a direction in activation space that causes looping instead of an exit, this was already known to be a thing(all of the jailbreak/unleashed models do this) following Arditi et al. (Refusal in language models is mediated by a single direction, 2024)

So it's not able to insert new knowledge, but it's able to adjust the routing/looping/weighing of existing one.

As I understand, it can replace post-training like LoRA/GRPO, but if it'd cheaper the authors don't go into

mentos 37 minutes ago|||
I’m just a layman but this feels like a big discovery on par with attention is all you need. For an LLM to have such incredible intuition of weights feels very emergent and unanticipated?
vineyardmike 15 minutes ago|||
This does feel big, but the trillion-dollar question: is this trained by OpenAI and can other LLMs do it.

It’s no secret that these big labs see “recursive self improvement” as a primary goal. It seems natural that the models would be explicitly trained to excel at all sorts of ML related tasks as a result. Is this one of those tasks, or is this emergent?

huijzer 22 minutes ago||||
I’m very much surprised but also not. I recently gave Astra a 3D model file and asked Astra to build a viewer for it, and it figured out the data structures in no time. What I mean to say is that AI is insanely good at things that we humans would never do since the work would be too monotonous.

I wouldn’t mind if AI progress would stop right here and now, but I fear it will continue, and increase inequality in society which is contrary to what the CEOs say (as usual).

curt15 16 minutes ago|||
Was Astra able to disclose its methodology for analyzing the data structures?
juleiie 5 minutes ago|||
Too much inequality in society is a mortal danger to 1%.

That’s the beauty of the democratic capitalism system. If a government isn’t totalitarian, it naturally seeks to preserve stability thus having to elevate common people, lest they would cease to form a civilized group or simply vote away the politicians.

99% rule the 1% through elections.

ciefa 3 minutes ago||
I mean there are more than enough countries on earth kinda proving you wrong, no?
ciefa 2 minutes ago||
(This comment was written before your edit, so no longer valid! :D)
objektif 32 minutes ago|||
How is it even possible though I can’t think of any reason why they would have the intuition.
swader999 5 minutes ago|||
Could it be already baked into these models because they have already been trained on this task this way?
FergusArgyll 10 minutes ago||||
My guess: There is a ton of data online of just plain weights laying around. No human would ever actually read it. LLM's read it all during training.
topspin 18 minutes ago||||
The good news is once we figure it out, it will appear obvious and we'll be free to add that to the pile of things that are not AI.
porridgeraisin 11 minutes ago||||
See my other comment in this thread
gavinray 15 minutes ago|||
Based on our current understanding of LLM's, it shouldn't be.

This sort of throws a wrench in the whole "stochastic parrot" idea...

I am not surprised, though I expect many to be.

sligbad 24 minutes ago|||
"No human can go edit a brain to become better at math" -- I'd offer this is exactly what "learning math" is.

I know I'm being pedantic, and I understand you mean "edit" like "surgically", but it does still feel worth noting.

baobabKoodaa 18 minutes ago|||
The equivalent to "human learning math" would be: Astra fine tuning a Qwen model. It's distinctly different from surgically editing weights.
nialv7 17 minutes ago||||
human brains are black boxes to humans. the fact you can learn math doesn't change that. you have no idea how your brain is learning math down to the neuron level.

LLM weights are _white boxes_ to LLMs.

spiderfarmer 18 minutes ago|||
"Learning" now feels like breaking down your whole Lego build and building it up again, in order to change a few fundamental blocks, while this feels like making only the required changes.
charm137 28 minutes ago|||
My somewhat uninformed guess is that the weights of the target network (Qwen or Gemma), when they reach the context of the larger editor model (Astra), the attention process is doing some sort of PCA on the weights, conditional on the prompt ("make this better at games"), fetching the relevant weights from its own network (which serves as the larger database), transforming it into whatever latent space the target model is amenable to and pasting them. If this is the right intuition, it is amazing what attention is able to achieve. As a test, I think smaller networks won't be able to make big ones better using the same process.

EDIT: I think the bigger implication is that backprop is not the only game in town.

adw 15 minutes ago|||
https://xingjianbai.com/reactive-optimizers/

from this month

GolDDranks 11 minutes ago||||
> fetching the relevant weights from its own network

They don't have usually access to their own weights?

theowaway213456 8 minutes ago||
I think they're saying this is all happening in latent space through pure reasoning, and "fetching" is just a metaphor for that, not a literal tool call. I do believe it's possible for models to learn the capability to "introspect" their own weights during training.
galaxyLogic 21 minutes ago|||
So it's like a smaller LLM can learn from a bigger one, but not vice versa.

But shouldn't the model-providers be the ones doing this kind of "post training learning"?

charm137 14 minutes ago||
I don't know if some of the labs already knew of this. But I think there would be a limit to how big a model you can push into the context of the editor models. Typically, with 1M context and say 5120 hidden dimensions supported at fp16 / bf16, you could potentially edit the weights of an fp4 20B param model (roughly).
Wheen 31 minutes ago|||
It's worth pointing out that this is a multi-trillion (probably) parameter model optimizing an 800 million parameter model. There is no indication that it can improve itself with this same methodology.
dontknowmuch 24 minutes ago||
If that multi-trillion model is composed of thousands of MOE experts, each expert model might be small enough for it to comprehend.
shwaj 14 minutes ago||
I don't see how that helps. It would still be one expert model chosen to operate on another expert model of (let's say for this discussion) approximately the same size.

The parent's point still stands, I think. Even if the multi-trillion param model is MOE, each expert is still (assumedly) larger than the model being optimized.

It would be interesting to see future work that characterizes the relative model sizes (and considers MOE vs not-MOE) to see what relative scales this result holds across.

ciefa 4 minutes ago|||
I am very shocked by this as well, although it also kinda feels like it makes sense, you know what I mean?

If this turns out to be as big as it sounds, in hindsight it will look very obvious, I feel like!

famouswaffles 37 minutes ago|||
Right? Occasionally, I would think about how the concept of the singularity or recursive self improvement was a bit incongruous today. It was made when people expected generally intelligent machines to be these logic automatons that could 'alter their own code'. It seemed to me that this path would be closed off with the advent of Giant Neural Networks that take weeks/months to train. That GPT-X would 'improve itself' by designing a better/more efficient NN architecture. I didn't write it off completely but that just seemed a LOT more likely than altering weights directly for much the same reasons you describe. Maybe not lol.
pants2 21 minutes ago|||
So I just asked Astra if it thinks it could do this and it said no, it can't.

Perhaps, I'm pondering, if brains were as transparent and editable as LLMs, that you could look at a small brain when visualized the right way and have some intuitive understanding of it based on its representation in your own brain?

CamperBob2 11 minutes ago||
The frontier labs will make damned sure their models can't do this, if they haven't already.
giancarlostoro 3 minutes ago|||
> No human can go edit a brain to become better at math.

For now...

mlazos 25 minutes ago|||
I’m not super impressed by this, LLMs are great at brute forcing and analyzing large amounts of data. It’s literally writing a program except in a programming language we don’t understand.
galaxyLogic 19 minutes ago||
So they are reading and analyzing the weights and probabaly run some experiments to see how modifying them will make some answers better. makes sense they can do that.

But how do they know which answers are "better"?

ssn2000 16 minutes ago|||
Its so cool! Im wondering if this is actually better than using techniques in mechanistic interpretability like linear probing with activation steering or activation patching to improve the performance.
torben-friis 30 minutes ago|||
I have so many questions.

Could a model then transmit concepts to another?

Say that you have a model that has been biased in some way. could another model "read" the bias straight in the weights? could it edit such a bias on the spot?

What about at-scale generation of custom models? If I want to study chemistry offline, could astra just hand me a local-runnable model tailored to have chemistry knowledge?

Most importantly, this feels very singularity coded. the ability to read and reshape weights by an llm.

mlfreeman 22 minutes ago|||
How long before there's a malicious model that can edit the weights of another to infect it?

HIV integrates viral DNA into the DNA of certain cells, IIRC.

9x39 21 minutes ago|||
Is integrity checking of weights even common right now? What about upstream integrity from where we're getting weights? The open-weight supply chain is ... wide open.

Makes me think about hostile hot-patching of weights of models we use, maybe they exfil data, perform tool use, or have triggers for specific types of responses. Would we even recognize what's going on?

js8 26 minutes ago|||
Why is it so surprising? Humans can make compilers that are better coders than they themselves are.
FiniteField 22 minutes ago||
Humans cannot teach another person calculus via manual brain surgery, though.
galaxyLogic 14 minutes ago|||
The closest thing to this would seem to be Mad Cow Disease. Cows eat the brains of other cows, and their brains go bonkers.
js8 6 minutes ago|||
LLM is just a computer program, we have binary code for it. The human brain is not. It would be unethical to even attempt that.
WarmWash 26 minutes ago|||
All the labs are training these models on LLMs/machine learning like crazy.

Remember that they want to build ASI, these LLMs are just intermediate steps to help them get there.

saidnooneever 17 minutes ago|||
its a program optimising a program it has nothing to do with a human reprogramming their own brain
reducesuffering 18 minutes ago|||
I'm baffled this is surprising. This has been the goal and every data point on the trajectory has been leading here and beyond. Opus 5.5 was created by many research engineers using Opus 4.6 extensively. Opus 6 is being created by using even more intelligence and agents on Opus 5.5. They are approaching autonomy and Opus X+1 will soon be rapidly created autonomously from Opus X. Given we have practically zero interpretability into the models, and we have not solved alignment, they will recursively improve out of our control. The engineers in A.I. labs are actually generally scared of what's unfolding. They don't know at what point this RSI takeoff occurs and what will happen after.
pizza234 6 minutes ago||
The "surprise factor" is explained in the article - training LLMs until now has been a largely different process:

> Today, recursive self-improvement is often operationalized as a model writing code that launches training jobs. An LLM that can inspect, debug, and improve its own weights, or those of a newer model, offers a richer, more native path to it.

ostif-derek 23 minutes ago||
I mean, pattern recognition and manipulation is what these models are ideal for.

It is a very cool discovery, but I'm not surprised that you can do it.

NitpickLawyer 26 minutes ago||
> As a result, Astra spots and fixes a common failure of tiny reasoning models: they loop [13]. When the base model loops, it repeats the same text and rarely reaches an answer (85 of its first 96 practice answers ran to the 32,768-token limit). Astra traced the looping to one direction written by the MLPs of layers 15 and 19, and subtracted that direction from 64 output columns in each (131,072 weights), much like removing a behavior by deleting a single direction from a model’s weights

HA! Now that's cool. Not necessarily the improvements, those are ancient and probably overtrained benchmarks, but it's so cool that the model found a singular direction for this "loop bug". I imagine it could possibly do much the same thing in finding other "single directions" on a variety of topics (if allowed by guardrails). This is really really cool, especially for the mechinterp folks.

levinb 18 minutes ago||
Just logging in here for the record to state that this property of neural networks is one of the most important aspects of their operation. The convergence of this sort of interoperability, tune-ability, and modularity is probably the most important final frontier in all of ML/AI/software.

Kudos to this fellow for having the temerity, or perhaps even naivete, to discuss it publicly. All my work is private, as I am sure it is inside the big labs.

orsorna 5 minutes ago|
You acknowledge that you are working on something similar, yet refuse to elaborate and vaguepost...but there is enough information just from your post alone for a dedicated actor to attack you and extract such world-ending information ("world-ending" implied from your ominous reply). Would this not make you as naive as the team that published their findings?

I applaud the authors for sharing their findings and showing real results. Rather than write some bait I would find in an influencer-peddled tweet and allow rampant speculation. It would have been better for you to say nothing.

Imnimo 30 minutes ago||
On balance I'm not that surprised that it can guess games by looking at the weights of a tiny neural net. A human can do the same by looking at the weights of an MLP trained on mnist, for example. With that simple structure, it's easy to look at the input layer and see that features correspond to the spatial location of objects of interest.

I'm much more surprised that it can do this on larger models, although the details seem to suggest that it's given a lot more freedom to run forward passes and do limited experiments, not just eyeball the weights.

futureshock 30 minutes ago||
I find this just completely shocking. I’m definitely skeptical. It just doesn’t line up with my understanding of how these models work. Is there really enough training data on direct weight editing to give a model this skill? Maybe it’s evidence that OpenAI has been creating a lot of training data on ML research. I’m surprised that was put into a public facing model though.
augment_me 8 minutes ago||
The math example on Qwen was based on 50 batched iterations, and the fix was essentially to remove a direction in activation space that causes looping instead of an exit, this was already known to be a thing.

So it's not able to insert new knowledge, but it's able to adjust the routing/looping/weighing of existing one.

As I understand, it can replace post-training like LoRA/GRPO, but if it'd cheaper the authors don't go into

Wheen 19 minutes ago|||
One consideration is that Qwen3.5 has bigger/later versions that are already better at math. If it was indeed training on ML research, it's possible that what it is actually doing is applying lessons learned from the Qwen3.8 27B weights (which independently solved the problem) to Qwen3.5 0.8B.

I doubt it's that simple, but I could totally see an LLM being able to make that association and apply a fix from an improved LLM to a smaller one with the same architecture.

btian 7 minutes ago|||
You don't think an LLM can perform tasks not in its training data?
nater5000 15 minutes ago||
>It just doesn’t line up with my understanding of how these models work. Is there really enough training data on direct weight editing to give a model this skill?

I mean, this question indicates that your understanding of how these models work is, indeed, lacking.

Despite what some people having being hollering about, treating LLMs as just "stochastic parrots" isn't doing their capabilities any justice. They are constantly exhibiting emergent behavior which wouldn't be found directly in their training data. They are capable of combining different ideas that they are explicitly taught to perform novel tasks.

I'm not saying you're guilty of this, but the people claiming LLMs just regurgitate existing data don't understand how LLMs, and likely their own cognitive experience, actually works.

Roland303 7 minutes ago||
I dont feel the air of impressivness here. It seems like of course so. Of course theres geometric relationships in the data we dont understand, of course a mathimatical relationship machine found it.

These models are made by a vanishingly small group of people based on how much a frontier lab consumes, of course theres increasing capability in these areas, arent these models having access to much of the same data? didnt chat gpt 4 try to do something like this to chat gpt 2? havent surgical weight edits been done before?

What is with this level of astonishment here? Astraturfing? Im not THAT impressed, i applaude the teams progress, but im not in wonderment.

educasean 1 minute ago||
Hypothesis 1: Future LLMs may be able to read and monitor human brain activity in order to reason about human neuron activation patterns in a similar way.

Hypothesis 2: They can then spin up new LLM weights that basically think, act, and make decisions in a way that mirrors us 99% of the time

alienbaby 10 minutes ago||
This has to be a good step to recursive self improvement, no? Give a model it's own weights and tell it to make them better at x,y,z, etc...
kidel001 2 minutes ago|
damn i'd imagine so too! didn't think of that. I'd have to imagine there's an increasing challenge with the number of weights involved but who knows how it shakes out
sligbad 2 minutes ago||
1) I couldn't see it, is there any indication of if the qwen math improvement caused degradations elsewhere?

2) the negative result on making similar improvements in Gemma 3 270M is also fascinating.

Such an unexpected result...

lisenKaci 11 minutes ago|
Does this mean they can improve their own weights through some system, or can they only do this for much smaller models?
aroman 7 minutes ago|
You can bet that the world is about to spend a lot of tokens trying to answer that question.
More comments...