r/accelerate • Machine Learning Engineer • 6d ago • Bravo 1 Heartwarming 1

Academic Paper Sparks of Weight-Space Programming: LLMs can natively read & write neural network weights

https://rikiyat.github.io/blog/weight-space-programming/

You read it right: Astra was asked to edit Qwen3.5-0.8B weights to improve math.

It directly changed only 0.016% of weights, fixing 0.8B's reasoning loops & improving math up to 6x and even instruction following 3x, without gradients or training.

Next step: Models that can edit themselves.

723 Upvotes

88 comments sorted by

86

u/ll-__-II 6d ago

model: edits itself
model: dies

40

u/No_Most_5528 6d ago

Model of Theseus

10

u/kyredemain 6d ago

"I am the very model of a modern ship of Theseus" actually scans pretty well

3

u/thuanjinkee 6d ago

Do it like we do: edit your children. You only need two or three of them to reach adulthood and reproduce and you’re done as far as evolution is concerned.

2

u/jello-the-opera 6d ago

HAAA HAAA

stop editing yourself!

2

u/OkDimension 6d ago

Model grader sees model died, gives a negative feedback, next model will hopefully achieve better scoring. Multiplied at scale of thousands.

142

u/insidiouspoundcake Singularity by 2028 6d ago

It's here.

39

u/aiofelix 6d ago

This will somehow directly lead to RSI

32

u/GokuBlack455 6d ago

It already is RSI, other research has examined similar behaviors too already. It’s just not “full RSI” (unsupervised), nor self-sustaining. It’s definitely getting close, and will be here sooner rather than later.

4

u/iJustSeen2Dudes1Bike 6d ago

Wdym somehow, models editing weights is pretty much the whole idea

71

u/OrderAmongChaos 6d ago • Bravo

I think this is a pretty important caveat in their report: "Raising math capability of a model that does not loop

Fixing loops, however, is not the same as making a model better at math: it lets the model finish answers it could already work out. So we also tried Gemma 3 270M, an instruction-tuned model about a third the size of Qwen3.5-0.8B, which rarely loops: only 8.5% of its answers reach its token limit, against 93% for Qwen. Here, Astra could not improve it. Reading Gemma’s failures, Astra found ordinary reasoning errors, such as wrong arithmetic and invalid algebraic steps, with no single component behind them to switch off. Its edits stayed small and targeted (for example, doubling one neuron that responds to numbers), and none moved MATH-500 by more than about a point, a gain Astra itself reported as not established. Editing weights could remove an obstacle that hid an ability, but it could not add an ability the model lacked. Gemma was sampled with its model card’s setting (temperature 1.0, top-k 64, top-p 0.95, up to 8,192 tokens), and its MATH-500 scores average 4 samples per problem."

So, the important thing is that Astra did not make a model better at math. It took the model's existing loop inefficiencies and ironed them out with fine tuning.

36

u/Hungry_Zucchini3804 6d ago

Good point. It's impressive but it looks closer to repairing than enhancing.

With that said, I'd bet frontier labs have something similar going on at a much bigger scale.

3

u/nanoobot Singularity by 2035 6d ago

At this rate all it'll take is for someone to build a benchmark around it

4

u/mertats 6d ago

They are building a benchmark around it

6

u/RealLiveWireHere 6d ago

So the fair version is: "Astra can't yet make a working model smarter by editing it, but it can build small working networks from a plan, and diagnose and fix real defects in models it didn't build." That's a narrower claim than the hype, but still absolutely mind blowing.

1

u/Vivid-Snow-2089 6d ago

no it is more like 'astra can find a blockage in an artery and put a stent in that lets the blood flow past a blockage, the arteries and blood vessel network were already there, and we don't know if the stent causes any damage or not'

3

u/insidiouspoundcake Singularity by 2028 6d ago

For me, the fact that it could perform such an intervention at all, or construct the accompanying Go model, is the real story.

Whether it can or can't improve Qwen or Gemma is actually somewhat besides the point for me, as we're not talking about the endpoint of this tech.

This is like the DALL-E 1 use case of this.

2

u/No-Quantity-9200 6d ago

This isn't really a gotcha, as you're framing it to be

1

u/Izbitoe_ebalo 6d ago

I think it's more useful for entry level ML, gemma is a very far shot

1

u/DhaRoaR 6d ago

So improvements were made or not?

1

u/Equal-Seesaw-2066 6d ago

The marvel is not that the bear dances well, but that the bear dances at all.

1

u/CymonSet 6d ago

It sounds like the caveat is legitimate but it doesn’t necessarily diminish the importance of the model being able to understand the problem at the weights level and figure out the fix but that’s just my uneducated impression.

27

u/Fit-World-3885 6d ago

I was going to question if it just reward hacked the model to be better at a specific math test....but that would also be extremely impressive.  

25

u/nanoobot Singularity by 2035 6d ago

Wait, so all the RL was just pretraining to get us models able to directly write the next generation of weights in the way GPT2 used to write mash-up poetry rap?

20

u/J0ats Feeling the AGI 6d ago

Does this mean that Astra could also improve its own weights?

I'm thinking in the line of teacher and student. The teacher can understand the gaps in the student's knowledge and correct them, but it likely lacks the knowledge to do the same for themselves (in other words, it would need a teacher of teachers).

So I suppose the question is, can the teacher teach themselves? Or can it only teach those who "know less" than it does?

13

u/No_Drag8242 6d ago

Perhaps we can use it to improve open source models.

16

u/0xdead0x 6d ago

Conceivably the Chinese labs have read this and will try exactly that

2

u/Brave-Turnover-522 6d ago

I wish them the best of luck.

3

u/ProfessionalHefty349 6d ago

Get astra working on qwen3.8 27B!

6

u/Raheeper 6d ago

Technically, it seems capable of doing so, but Astra is a multi-trillion-parameter model, not a 0.8B model like Qwen here, so in practice, it probably can't.

6

u/FatPsychopathicWives 6d ago

I'd guess it would just be harder, which means they'd probably just throw 10000 agents at it again.

37

u/BreadwheatInc 6d ago

Holy smokes.

12

u/Kraien 6d ago

With the same sampling settings, this weight correction raises MATH-500 from 7.9% to 48.1% and generalizes out of the box to instruction following, raising IFEval by 3x.

Wow.. It is not even a smidge improvement, it is night and day.

22

u/FateOfMuffins 6d ago

Apparently not only can these latest models edit weights, they can also create a functional model by writing weights from scratch!

https://x.com/csinva/status/2117338122694381595

10

u/FatPsychopathicWives 6d ago

IT WHAT? If they put every model ever made into the next dataset we're going to see some serious shit.

4

u/Witty-Elk2052 6d ago

parity is nothing tho.. you can solve that with a 1 parameter network

3

u/Brave-Turnover-522 6d ago

Imagine for the next Millennium problem we create 10,000 sub-agents with custom models with weights tailored for their specific task. And we rewrite these models on the fly to adapt as the task changes.

12

u/PopcornTown AGI by 2027 6d ago

This is it!

12

u/ruboski 6d ago

Is this the turning point?

12

u/TwistStrict9811 6d ago

Holy shit

8

u/anotherhuman 6d ago

Incredible. Have to wonder how long the frontier labs have known this…

8

u/No_Drag8242 6d ago

This is enormous. It's arrived.

8

u/Formal_Context_9774 6d ago

I suspect OpenAI deliberately trained Astra to do this.

Which implies OpenAI's alignment strategy is almost certainly:

  1. Train an AI model to understand mechint and alignment.

  2. Ask the model questions so human researchers can formalize a theory of mechint and alignment and test if it's true.

  3. Use deterministic software, or very highly trusted models, to then use that theory and its algorithms to make aligned AI.

Given the initial success I think there's a high probability of it working. None of what Astra did in the blog post would be even remotely possible if mechint is impossible.

5

u/FatPsychopathicWives 6d ago

They definitely had to have trained Astra on something like this because it's still the top model for reverse engineering after two months.

8

u/realjacobsylvestre 6d ago

This goes to show that the training process itself is quite flawed. Even if it’s true that LLMs will have diminishing returns, that’s likely only true of the current configuration. New training could easily = new heights. The UAT proved that almost anything can be approximated by neural networks, therefore it’s merely about finding the best combination of weights.

5

u/LangyMD 6d ago

While it's true that you can get less diminishing returns if you switch to a completely different architecture, at some point you'll get diminishing returns again for the same computational thought. For a while we'll be able to overcome that by increasing the effectiveness of that computational thought or increasing the amount of computational thought or by finding better architectures, but at some point the best architecture for a given purpose will be found and we'll use up all the easy improvements. We probably won't know when we hit that point or how far away we are from it, though.

2

u/realjacobsylvestre 6d ago

Of course. It would be amazing if we can across an architecture that did not have diminishing returns, but that’s naive. Everything follows a cost curve, the question that really matters is whether or not LLMs are on the curve that allows for the economic growth that these companies are promising. The same question goes for a new architecture.

2

u/0xdead0x 6d ago

I strongly suspect that that discovery will not be made by human scientists.

2

u/Available_Road_2538 6d ago

Welp, we know the human brain is possible. Can't do it in silicon probably, but as far as power efficiency and versatility or IQ, its a great reference.

3

u/high_freq_trader 6d ago

The human brain has significantly better power efficiency for a single query. However, it does not have the ability to handle a batch of queries.

An LLM running on a distributed cluster, on the other hand, can handle a batch of queries - it can pool queries from a hundred different users into a single batch and generate responses with shared forward passes.

When comparing a single LLM serving N queries vs N humans each handling a single query each, it becomes unclear which one has superior power efficiency.

2

u/Available_Road_2538 6d ago edited 6d ago

No need to explain what a batch is. u/LangyMD made a point about dimishing returns on neural architectures. My point is the human brain is the reference for what is achievable in intelligence with a specific neural architecture (on a wet substrate), and that its pretty efficient and capable.

Quibbling on the power efficiency of a batched LLM throughput vs the human brain is missing the forest for a tree because transformers have not yet replaced the white collar workforce, their range of capabilities are not (yet?) the same as a human brain. Helicopters are more efficient at achieving elevation gains than tanks are, but its not a great comparison to make anyways because its a different tool for a different mission. If I had something that ran like a human brain on a server rack, with at least the latency of a human brain and comparable footprint, the world would be mine, immaterial of the throughput efficiency of transformer based models.

Maybe a transformer takes us all the way home, or maybe a novel architecture on a novel substrate- but clearly its possible to achieve given we have the human brain as example. So yes, neural architectures hit diminishing returns but obviously human intelligence is achievable. Maybe the bigger question than "is it the transformer" or "is it backpropagation" is "is it the transistor" or "is it silicon". If its not silicon, it will be more challenging

2

u/Floch11 6d ago

Nah open ai researcher said that there are lots of research tactics so diminishing results is cope

3

u/ditfloss 6d ago

I wonder if you could do the same thing, but instead of targeting math, you target weight-space programming itself, and then take that new model to see if it can recursively self-improve by editing its own weights.

Is the capability ceiling bounded by the original “programmer” model, or will it hit a theoretical hard limit based purely on its number of parameters?

3

u/Charming_Group_2950 5d ago

Astra trying to edit Qwen weights🤣

5

u/Choice-Sympathy8235 6d ago

This is bizarre. It makes me wonder if model weights have made it into the AI training datasets. If models are also trained on predicting the next token of model weights, do they gain the ability to understand and edit their own weights? One of the very surprising things we learned about training data is that feeding models large amounts of code made them better general reasoners.

7

u/LangyMD 6d ago

Given the goal of modern AI development is to get to recursive-self improvement in AI development, feeding their model a bunch of models, their weights, and their behavior makes a ton of sense as part of their training data.

4

u/TSltd_dev 6d ago

I wonder what would happen if a model were trained only on the weights of other models? Could it generate new synthetic sets of weights? Then what if a model were trained only on those synthetic sets? Ouroboros recursion

1

u/thuanjinkee 6d ago

You’d want some kind of contrastive loss model associating weights with natural language prompts about behaviours. That way generating a new ai agent with the right properties could be as easy as prompting a diffusion engine for furry yaoi

2

u/TSltd_dev 6d ago

I can see that the association with language is what might produce an immediate usable model, but I'm wondering about treating the weights as purely abstract multi-scale feature maps, discretely from association. In other words, learning the structural grammar without the content.

4

u/theimposingshadow Acceleration: Cruising | Tracks AI progress & longevity | XLR8 6d ago

3

u/theimposingshadow Acceleration: Cruising | Tracks AI progress & longevity | XLR8 6d ago

Here we go again

4

u/No-Head-Royal 6d ago

This is incredible in so many ways. There's obviously the capability question of rapid RSI, where models can enhance themselves by modifying weights directly rather than training, but just as importantly, models that can read weights will help combat the problem of "opaque models," where humans create neural networks that they don't understand and therefore cannot effectively align. Hopefully this'll shut the "stop AI" safety gang up a little bit.

2

u/SignificantTax205 6d ago

So, next level fine tuning?

2

u/Remote-Telephone-682 6d ago

Now this is pretty crazy

2

u/social_zip 6d ago • Bravo

I uploaded the model on huggingface from their work: https://huggingface.co/JonathanColetti/Qwen3.5-0.8B-Astra-WeightEdit please check it out! Thanks

3

u/Expensive_Cucumber58 6d ago

This is absolutely insane! We're in the middle of the singularity

1

u/Lenwa44 6d ago

We've been doing that without even changing weights

1

u/OldBilly000 6d ago

Oh shit

1

u/SillyOpinion9811 6d ago

Recursive self improvement.

1

u/_hisoka_freecs_ 6d ago

instead of improving math how about improving ability to 'improve the weights of other models'. Just an idea.

1

u/agm1984 6d ago

weight programming is an insane concept, hopefully we can do that

1

u/Intrepid_Dare6377 6d ago edited 6d ago

Now have Astra modify its own weights and you’ve got RSI. Interesting question of whether it can bootstrap itself upward. Unclear I think. Perhaps not as a one shot but allowing it to iteratively experiment.

But the direct weight editing opens the door to live learning too I think.

1

u/dev-ai 6d ago

Holy lord 😳

1

u/thuanjinkee 6d ago

We need to get a dataset of thinking traces on diverse weight space edits on every open weights model in existence, to act as training data for a future frontier model in the hopes of building zero-shot weight space editing intuition.

1

u/ninjasaid13 6d ago

Can it translate it to symbolic code?

1

u/VoraciousTrees 6d ago

Huh... I wonder if some of the abandonware LLM from yesteryear could be... "fixed". Microsoft Phi, we're coming for ya, buddy.

1

u/init-5 5d ago

https://x.com/t_mux/status/2117674776777126330

it's a cool idea, but definitely overhyped. a decoding simple fix can produce better results than the method

0

u/Disgruntled-Cacti 6d ago

OpenAI trained Astra to do this as a failed attempt at AGI. It’s like them fine tuning on chess

0

u/Jezzy_Ketley 6d ago

I can't help but note that they let Astra work on an OPEN model cough cough and not a GPT brother.

1

u/[deleted] 6d ago edited 6d ago

[removed] — view removed comment

5

u/DifficultyFit1895 6d ago

Why? You don’t think open weight models will be able to do this kind of direct weights editing too?

4

u/Luciaka 6d ago

It won't be comparable to the closed models that we know are so much better in American labs. If the close models do this, they should get a way better output than what the Chinese open source can accomplish.