diffusion model

Diffusion models may soon design AI weights, not just images

What if you could ask an AI to design a neural network the same way you ask it to generate an image?

4 min readMachine Learning
Diffusion models may soon design AI weights, not just images
What if diffusion models generates Neural Network weights? Gen AI generate Gen AI [R]

Diffusion models learned to generate images by mapping noise into coherent pixel grids. Now someone is asking whether the same process can generate the weights of a neural network itself. A developer working under the handle Specific-Tax-6700 has published a proof-of-concept called nngen that attempts exactly this: train a diffusion model to output the parameter values of a small network, then prompt it with a description of what that network should do. The early results are crude and the training is only half-finished on a laptop CPU, but the direction is worth paying attention to. This is not a polished product. It is a provocative experiment that connects directly to work we have covered recently, such as When AI deciphers a neural network's strategy from raw weights, where models read a network's weights as plain text and infer its behavior. That work showed weights contain a readable signature of function. This experiment asks whether that signature can be generated on demand.

Our take is straightforward: this idea is more important than its current implementation suggests. The developer frames it as a way to say "create me a neural network that will guide a pen to draw a man" and get a small, working model. That is a radical simplification of what it currently takes to design and train a neural network. Today, building even a toy model requires choosing an architecture, setting hyperparameters, collecting data, running a training loop, and debugging convergence. A diffusion model that outputs weights directly would bypass most of that pipeline. It would treat the neural network itself as a generated artifact, not a trained one. The implications extend beyond parlor tricks. Consider the work of Physicl's $10M bet bridges digital commerce and physical robot training, which tackles the sim-to-real gap by building pipelines that transfer digital behaviors into physical robots. A weight-generating diffusion model could one day produce controller networks for specific robot tasks on the fly, skipping the simulation step entirely.

The practical barrier right now is scale. nngen operates on tiny networks with a handful of parameters, and the results are inconsistent even within that narrow scope. The developer openly shows the best outputs, which suggests the average quality is lower. But the pattern is familiar. Diffusion models for images started with blurry 32x32 thumbnails. The underlying method, learn the distribution of valid outputs, then sample from it, scaled with compute and data. The same scaling laws should apply to weight generation, provided the community invests in datasets of trained networks and finds a stable representation for parameter tensors that the diffusion process can handle. That is the open question the developer is asking the community to help solve. No papers exist on this exact approach yet, which means there is room for genuine discovery.

What makes this experiment different from typical hobbyist projects is the specificity of the ask. The developer is not claiming to have solved neural architecture search or meta-learning. They are pointing to a concrete possibility: that the same generative model that produces images can produce the internal machinery that produces images. If that holds, it collapses two levels of abstraction into one. A user could describe a behavior and receive a network that implements it, without ever touching a loss function. That is not a distant future scenario. It is a testable hypothesis sitting in a GitHub repository right now, trained to only 3000 of 4000 steps on a laptop CPU. The next thousand steps will tell us more than any prediction.

From Machine Learning

I'have been dreaming about using a Diffusion Model that will generates Weights instead of pixels. Diffusion models actually are vast used in order to generate from a natural language text description into images, and what is an image? a three dimensional matrix of value representig luminosity of red , green and blue. What if instead of matrix of colour intensity we train a diffusion model in order to create a neural network, and maybe a drawing NN too. If it works you can ask something like: "create me a neural network that will guide a pen in order to draw…

Read the original at Machine Learning