Skip to content
The Daily Triptych221 / 365
Weights kept, and whether training succeeds

Sizes reflect the range reported by Frankle and Carbin for feed-forward networks on MNIST and CIFAR10. The third bar is the same sparse shape given fresh random values, which trained poorly.

II · THE IDEA · ARTIFICIAL INTELLIGENCE

The Lottery Ticket Hypothesis

Frankle and Carbin, arXiv 1803.03635 · pruning, sparsity, initialisation · MNIST and CIFAR10 feed-forward networks · Malach and colleagues, arXiv 2002.00585

▶ 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.

Look closer

  1. 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.

  2. 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.

  3. 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?

Go deeper

Image: Original diagram, The Daily Triptych. Licence: Original work. Source.

← Back to day 221