knowledge base Β· entry 07
The winning subnetwork was in there all along
A read-through of the paper that found a small network already hiding inside a big one β and the one counterintuitive step that decides whether it works.
every figure zooms β click it, scroll to magnify, drag to pan
part one
The idea, before any jargon
figure 1 Β· the whole idea
A smaller network was in there all along
Train a big neural network and most of its weights turn out to be nearly useless β you can delete 80β90% of them and barely dent its accuracy. That much was already known; it's called pruning. The puzzle was that the pruned small network, if you tried to train it from scratch on its own, learned badly. So the smallness seemed to be a property you could only reach by training big first.
This paper found the loophole. Inside the big network there is a small subnetwork β a winning ticket β that trains just fine on its own, reaching the full network's accuracy in about the same number of steps. The catch is that you have to keep its original starting weights. Same connections, same initial values, trained alone.
click any figure to zoom Β· scroll to magnify Β· drag to pan
The distinction that makes it surprising: it isn't that a small network can be good β we knew that. It's that the small network was already present, already well-positioned, at initialization, before a single step of training. Training didn't build it. Training revealed which one it was.
figure 2 Β· the metaphor
Why it's called a lottery
Here is the mental model the name comes from. A big network doesn't contain one subnetwork β it contains a combinatorially huge number of them, overlapping and sharing weights. Think of each as a ticket. Every ticket starts with whatever random initial values it happened to be dealt.
Training is the draw. A few tickets began with an initialization that happens to sit in a good spot β close enough to a solution that gradient descent can get there. Those are the winners. Most tickets are duds.
This reframes something everyone had noticed but couldn't explain: bigger networks are easier to train. Not because the extra capacity is used β most of it isn't β but because buying more tickets makes it likelier that at least one of them is a winner. Over-parameterization is buying lottery tickets in bulk.
part two Β· the method
How you find one
You can't inspect a fresh network and point to the winning ticket. You have to train the whole thing once to see which connections mattered β then do something slightly strange.
figure 3 Β· iterative magnitude pruning
Train, prune, rewind, repeat
The recipe is a loop. Train the full network. Delete the weights that ended up smallest β closest to zero, so presumably least important. Then, for the weights that survived, throw away their trained values and put back the exact numbers they started with. Retrain from there, and prune a little more. Round after round, the network gets sparser and the winning ticket emerges.
Two details do all the work. Doing it iteratively β a small fraction per round over many rounds β finds much smaller tickets than one aggressive cut. And the rewind step in the corner is the whole discovery: you reset survivors to their original initialization, not their trained values, and not fresh random ones. The next figure shows why that one choice is everything.
part three Β· the one bit of arithmetic
What "prune the smallest" actually means
Skippable, and genuinely simple β pruning is just sorting a list of numbers and cutting the bottom of it.
figure 4 Β· the toy example
Magnitude pruning, by hand
Take one layer's weights. Magnitude pruning ranks them by absolute value β how far from zero they are β and removes the smallest slice. A weight near zero barely affects the output, so it's the first to go. That's the entire rule; there is no cleverness to it.
Eight weights, cut the four nearest zero. The survivors β the tall bars β keep their positions and become the ticket's connections; everything below the line is set to zero permanently. Real networks prune gently, around 20% per round over many rounds, but each round is exactly this.
worth noticing Nothing here looks at what a weight does, only how big it is. That magnitude is a decent proxy for importance is itself a little surprising β and it's why the method is cheap enough to run many times over.
part four Β· the surprise
The result that names the paper
figure 5 Β· rewind vs. random
The ticket is a structure and an initialization
Here is the experiment that proves the point. Take a winning ticket's sparse structure and train it two ways. First, with its original initial weights β the rewind. Second, with the same connections but fresh random values. If the structure were what mattered, both would learn the same. They don't.
The rewound ticket climbs fast and lands at the full network's accuracy. Re-roll the initial weights and the identical structure learns slower and plateaus lower. So a winning ticket isn't a lucky shape β it's a lucky shape paired with the specific starting numbers it was dealt. Separate them and the magic is gone.
The paper found these tickets at 10β20% of the original size, matching or beating the full network and often training faster. The headline result β a small network that trains to full accuracy β had been thought impossible precisely because everyone had been re-initializing at random.
the math Β· in the paper's notation
The hypothesis as the paper states it
The paper's section 1 states the claim in mask notation, and the masks are the only formal machinery in it. Decoded here, plus the one line of arithmetic behind the pruning schedule.
the equations Β· worked
Masks, the ticket condition, and the schedule
Β§1 Β· the lottery ticket hypothesis, formally
dense: f(x; ΞΈ), ΞΈ ~ πΞΈ subnetwork: f(x; m β ΞΈ), m β {0,1}|ΞΈ|
winning ticket: β m : jβ² β€ j, aβ² β₯ a, βmβ0 βͺ |ΞΈ|
read aloud: there exists a 0/1 mask such that the masked network, trained alone from the original initialization ΞΈβ, learns in no more iterations, to no less accuracy, with far fewer weights. All three inequalities at once β that is what "winning" means, and the rewind to ΞΈβ is baked into the definition.
- mthe mask: 1 keeps a weight, 0 deletes it forever
- βelement-wise multiply β applying the mask
- ΞΈ, πΞΈthe weights and the random distribution they were initialized from
- j, jβ²training iterations to converge: full network vs. ticket
- a, aβ²test accuracy reached: full network vs. ticket
- βmβ0the count of surviving weights β the "L0 norm" is just counting ones
Β§2 Β· iterative magnitude pruning, as arithmetic
keep rate after n rounds = (1 β p)n
read aloud: prune a fraction p of the survivors each round, and the surviving share compounds down geometrically. Each round: train to convergence, cut the smallest |w| β magnitude is the whole criterion β rewind the rest to ΞΈβ, repeat. The gentleness is the point: one aggressive cut to the same sparsity finds worse tickets.
- pthe per-round pruning rate β around 0.2 in the paper
- nrounds of train-prune-rewind
part five Β· what we take from it
Sparsity in a system you ship
applied Β· our reading
The small model, found rather than trained
For the work we do β specialist models that have to run inside someone's boundary β this is the other half of the compression story. Distillation makes a new small model that imitates a big one; the lottery ticket says a usable small model may already be hiding inside the one you trained, if you can find it.
the caveat we would flag Finding the ticket isn't free β you must train the full network first, then prune and retrain over many rounds, so this buys a cheaper model to serve, not a cheaper one to build. And magnitude pruning leaves sparsity scattered unpredictably through the weights; without kernels or hardware built for it, a 90%-sparse model can run no faster than the dense one. The result is a profound statement about over-parameterization first, and a deployment recipe second.
next in the arc
Where this goes
Distillation copied a model down; the lottery ticket finds the small model already inside. The third entry in making models smaller takes yet another angle: LoRA doesn't shrink the model at all β it freezes it and trains a thin adapter beside it, on the theory that the change a fine-tune needs is far lower-rank than the model. Three papers, one question: how little is actually necessary?
source
Jonathan Frankle, Michael Carbin β MIT CSAIL (2019). The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks. arXiv:1803.03635. The MNIST-scale claims are the paper's; the toy weights, the network diagrams, and the learning curves are our own illustration.
β back to the knowledge base Β· β the illustrated decoder Β· βΆ run the forward pass
attention.sh β a division of Jinacode Systems. Transformers,
fine-tuning, and sovereign LLMs.
we're hiring the curious β hello@attention.sh