$ attention.sh

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.

paper The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks
authors Jonathan Frankle, Michael Carbin β€” MIT CSAIL (2019)
source arXiv:1803.03635
why we read it sparsity is the other route to a small model you can actually deploy

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
plug in β†’ p = 0.2: after 10 rounds, 0.810 β‰ˆ 10.7% of weights remain; after 15 rounds, 0.815 β‰ˆ 3.5%. the paper's headline tickets at 10–20% of the original size are ~7–10 rounds of this compounding β€” nothing more exotic than repeated multiplication by 0.8.

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