II · THE IDEA · ARTIFICIAL INTELLIGENCE
The Lottery Ticket Hypothesis
▶ Listen · narrated
A network stripped to a tenth of its connections normally refuses to learn. Give those surviving connections back their original random starting values, and it learns as well as the whole thing did.
At a glance
- The hypothesis
- Dense random networks contain subnetworks that train to comparable accuracy
- Ticket size
- Less than 10 to 20 per cent of the original weights, in the cases reported
- Crucial step
- Surviving weights are reset to the random values they held before training
- Tested on
- Fully-connected and convolutional feed-forward architectures
- Theory
- Existence proved for sufficiently over-parameterised random networks
- Practical catch
- Identifying a ticket requires training the dense network first
Suppose a large orchestra where every player has been handed a random instrument and a random part. They play the whole piece through, and you note which players actually contributed to the sound. Then you dismiss 9 out of 10 of them and rehearse again with only the players who mattered — and, crucially, you hand each one back the exact instrument and part they started with. The small group plays the piece as well as the full orchestra did.
Now try the same thing with a twist: keep the same 1 in 10 players, but shuffle their instruments and parts. It goes badly. That comparison is the whole experiment. The small group did not succeed because small groups are enough. It succeeded because those particular players, with their particular original assignments, happened to fit together.
The catch is built in. You cannot tell in advance which players will matter — the full orchestra has to play the piece through first, and only then can you see who counted. So this does not yet let you hire a smaller orchestra. It tells you a smaller one would have been enough, if only you had known which one.
The procedure Frankle and Carbin report is iterative magnitude pruning with weight rewinding. Initialise a dense network at random and keep a copy of that initialisation. Train to convergence. Prune a fraction of the weights, which yields a binary mask — a record of which connections survive. Apply that mask to the stored initialisation rather than to the trained weights, so every surviving parameter returns to its original random value. Train the masked network for at most as many steps as the dense run took, and compare test accuracy.
The hypothesis, stated formally: a randomly initialised dense network contains a subnetwork that, initialised with those same values and trained in isolation, matches the test accuracy of the original within the same number of iterations. Three quantifiers carry the claim. The subnetwork is asserted to exist, not constructed. The initialisation is fixed, not free. The training budget is bounded, not open-ended. Drop any one and you are testing a different claim.
Reported results cover fully-connected and convolutional feed-forward architectures on MNIST and CIFAR10. Winning tickets were consistently found at under 10 to 20 per cent of the original parameter count; above that threshold they learned faster than the dense network and reached higher test accuracy. The motivating tension, as the paper frames it, is that pruning can remove over 90 per cent of a trained network's parameters without compromising accuracy, yet the resulting sparse architectures had proved hard to train from scratch — so the gains landed on inference, never on training.
The binding limitation is search cost. The mask is a product of dense training, so the method identifies the expensive run rather than avoiding it. Training sparse from the outset would require finding a good mask without that run, and this paper does not supply a way.
Malach and colleagues address existence theoretically. For sufficiently over-parameterised random networks, they show that a subnetwork with untrained random weights can approximate a given target network — pruning alone suffices. This is a statement about what exists under an over-parameterisation assumption, not about tractable search. Treat it as licence to view pruning as a form of function selection, not as an algorithm.
Look closer
The rewind is the whole trick
The procedure has one unusual step in the middle. Start a dense network at random and record every starting weight. Train it. Remove a fraction of the weights, which leaves behind a mask, meaning a pattern of which connections still exist. Then rewind: put each surviving connection back to the exact random value it held before any training happened, and train again using only those connections. The network that results is sparse from the first step of training, not sparse only at the end. That is what makes it interesting, and it is the detail that separates this experiment from ordinary pruning.
Sparse architectures are normally hard to start from
The paper opens with a tension already familiar to practitioners. Pruning a trained network can cut its parameter count by over ninety per cent without compromising accuracy, which makes the finished model cheaper to store and faster to run. But the sparse architectures that pruning produces have been difficult to train from scratch. If they were trainable from the start, training itself would get cheaper, not just the running of the finished model. The lottery ticket experiments are a probe into why that gap exists.
What the numbers actually say
Frankle and Carbin report consistently finding winning tickets smaller than ten to twenty per cent of several fully-connected and convolutional feed-forward architectures, tested on MNIST and CIFAR10. Above that size threshold, the tickets they found did more than match the dense network: they learned faster and reached higher test accuracy. Below it, the claim weakens. That boundary is worth holding on to, because it is easy to remember the headline and forget that the result is bounded on both sides.
The story
A neural network stores everything it learns in its weights — the numbers attached to each connection between its units. Before training, those numbers are set at random. Training nudges them, step by step, until the network does its job. Afterwards, it turns out that most of the weights were barely contributing. Pruning is the practice of deleting them: train a large network, remove the connections that mattered least, and keep the rest. Done carefully, this can cut more than 90 per cent of the weights without hurting accuracy, which is why pruning is the standard way to make a finished model smaller and faster to run.
That raises an obvious question. If a tenth of the network was enough in the end, why not build the small network in the first place and skip the expensive part, which is training the big one? Practitioners had tried. Take the sparse shape that pruning leaves behind — the same connections, nothing more — give it fresh random starting values, and train it from scratch. It trains poorly and ends up less accurate. The small shape works after pruning but not before, and nobody could say why.
Frankle and Carbin changed exactly one thing. Before training their dense network, they wrote down every one of its random starting values. Then they trained it, pruned it, and kept the record of which connections survived. Now the unusual step: instead of giving those survivors fresh random values, they reset each one to the exact random number it had held before training began. The deleted connections stayed deleted. Then they trained this sparse network, from that particular starting point, for no more steps than the original had used.
It worked. In their experiments — fully-connected and convolutional feed-forward networks, tested on the MNIST and CIFAR10 image datasets — the sparse network matched the accuracy of the dense one it came from. The subnetworks they found held fewer than 10 to 20 per cent of the original weights, and above that size they did better than match: they learned faster and reached higher test accuracy than the full network. So the sparse shape was never the problem. The problem was the starting values. A pruned network trains well only when its surviving connections begin from the same random numbers they held in the original network.
The authors call such a subnetwork a winning ticket, and the name explains their reading of the result. Setting up a large network at random is like buying a great many lottery tickets at once: each possible combination of connections and starting values is one ticket. Training the whole network, then pruning it, reveals which combination happened to win. Their hypothesis, stated carefully, says a randomly initialised dense network contains some subnetwork that, kept at its original starting values and trained on its own, matches the full network's accuracy within the same number of training steps. Every clause is load-bearing. Not any subnetwork — one that exists in there somewhere. Not any starting values — the ones it was born with. Not unlimited training — the same budget.
The method is honest about its own limit. You only learn which connections mattered by training the dense network and pruning it. So the procedure saves no training cost. It proves that a small trainable network was in there; it does not tell you how to find one without doing the expensive run first. Most of the follow-up work lives in that gap.
Two years later, Malach and colleagues proved something stronger on paper. For a random network with enough spare capacity, they showed, you can find a subnetwork that approximates a target network with no training at all — you delete connections and never adjust the survivors. Their title puts it plainly: pruning is all you need. Read that way, a large random network is not a blank surface waiting for training to write on it. It already contains, by chance, an approximation of the function you want, and training is one way of uncovering it.
Why it mattered then
In 2018, pruning was filed under deployment. You trained a model, pruned it, and shipped the smaller version so it would fit on a phone or answer faster in a data centre. All the benefit came at the end of the pipeline. Everyone also knew that the sparse shapes pruning produced trained badly from scratch, but that observation had settled into folklore — a thing you worked around rather than a question you asked. Frankle and Carbin showed the folklore was imprecise. Those sparse shapes were not untrainable. They were untrainable from fresh random values, which is a narrower and more revealing claim. By restoring the original values and watching training succeed, they isolated the starting values as the ingredient that mattered. That turned a routine piece of engineering into a question about why hugely over-sized networks train well at all — a question that was live in 2018 and is still only partly answered.
Why it matters now
The economics have only sharpened. Training is the expensive half of building a modern model, and any credible route to training a small network instead of a large one would be worth a great deal. The lottery ticket hypothesis names that prize exactly, and is equally exact about not claiming it: you still have to train the dense network to discover which subnetwork was the winner. It also changes what a model's size means. If part of a dense network's success is that it contains many candidate subnetworks, then scale is not just more room to store knowledge. It is also more draws at a lucky starting configuration. And the later proof — that a large enough random network holds a subnetwork approximating a target with no training of the survivors — suggests the line between choosing a structure and learning weights is blurrier than the standard account allows. One caution. The original experiments cover fully-connected and convolutional feed-forward networks on MNIST and CIFAR10, which are small image datasets. Whether any of this holds for the largest models in use today is beyond what these two sources establish.
The surprising detail
The theoretical follow-up goes further than the experiments. Malach and colleagues prove that a random network with enough spare capacity contains a subnetwork, still holding its random weights, that approximates a target network — and reaching it requires no training of the survivors at all. Deleting connections is enough. On that view, a large random network is not an empty vessel that gradient descent fills. The function you want is already in there by chance, and the work is deletion rather than adjustment. It is a proof about what exists in principle, under a spare-capacity condition; it supplies no practical way to find such a subnetwork.
What is disputed
The two sources here do different kinds of work and should not be merged. Frankle and Carbin report an empirical finding on fully-connected and convolutional feed-forward networks trained on MNIST and CIFAR10; they consistently found winning tickets, but that is evidence from those settings, not a general law. Malach and colleagues prove an existence result under an over-parameterisation condition, which says nothing about whether such a subnetwork can be found efficiently in practice. Neither source establishes how the result behaves at the scale of today's largest models, and nothing above should be read as claiming it does.
Remember this
A dense random network already holds a small subnetwork that trains just as well — but only if it keeps its original random starting values, and the only known way to find it is to train the big network first.
Test yourself
Suppose you prune a trained network down to ten per cent of its weights, then give those surviving connections fresh random values before retraining. Why does the lottery ticket result predict something different from resetting them to their original values?
Because the hypothesis makes a claim about a pairing, not about a shape. A winning ticket is a particular set of connections together with the specific random values those connections were assigned at the outset. The pruning mask tells you which connections mattered during that one training run, and it mattered partly because of where those weights started. Hand the same mask new random numbers and you have kept the architecture but discarded half the claim, which is why sparse architectures had been observed to train poorly from scratch. This is also why the method cannot yet save you any training: the only way to learn which connections and which starting values formed a winner is to run the dense training that reveals them. The later theoretical work complicates the picture usefully, showing that for sufficiently over-parameterised random networks a good subnetwork can be found by pruning alone, with no training of the survivors at all.
Go deeper
- [1803.03635] The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks · arxiv.org
- [2002.00585] Proving the Lottery Ticket Hypothesis: Pruning is All You Need · arxiv.org
Image: Original diagram, The Daily Triptych. Licence: Original work. Source.