Skip to content

Lottery ticket hypothesis

Why a sparse neural subnetwork and its original initial weights form a lottery ticket: reset mechanics, empirical comparisons, rewinding and compute limits.

The lottery ticket hypothesis proposes that a dense, randomly initialised neural network contains sparse subnetworks whose original initial weights let them reach comparable test accuracy within at most the same number of training iterations. For quant research, this makes compression a question about trainability, as well as the number of surviving weights.

A ticket includes its starting weights

In Frankle and Carbin's original experiment, the search has three essential steps:
  1. Save the dense network's random initial weights.
  2. Train it, then use neural network pruning to select connections from the trained network, removing low-magnitude weights.
  3. Reset the surviving connections to their saved initial values and train this subnetwork in isolation, keeping removed connections absent.
Repeated train–prune–reset rounds form iterative magnitude pruning. A candidate becomes a winning ticket only after its retraining performance meets the stated accuracy and iteration comparison. The name describes a successful empirical comparison, not every sparse mask.

What exactly gets reset?

Take four illustrative, dimensionless weights. Their initial values and a stipulated later trained state are

θ0=(0.1,−0.2,0.3,−0.4),θT=(1,−2,3,−4).\theta_0=(0.1,-0.2,0.3,-0.4),\qquad \theta_T=(1,-2,3,-4).
Now stipulate the keep mask m=(1,0,1,0)m=(1,0,1,0). A one means retain that connection; a zero means remove it. The original reset operation uses the initial values:
m⊙θ0=(0.1,0,0.3,0),m\odot\theta_0=(0.1,0,0.3,0),

where ⊙\odot means elementwise multiplication. Keeping the trained values instead gives

m⊙θT=(1,0,3,0).m\odot\theta_T=(1,0,3,0).
These are different starting points for subsequent training, despite having exactly the same surviving connections. Two of the four scalar weights remain: this ledger has 50% sparsity, counting the two zeroed positions among all four positions.

This mask is hand-chosen to isolate reset mechanics. Magnitude pruning that kept the two largest absolute trained weights would retain coordinates three and four, giving mask (0,0,1,1)(0,0,1,1). It would not produce our stipulated mask. No dataset, network outputs or training runs are supplied here, so the ledger establishes neither predictive accuracy nor a winning ticket.

Illustrative initial weights 0.1, −0.2, 0.3, −0.4 and trained weights 1, −2, 3, −4 under stipulated mask 1, 0, 1, 0. Resetting gives 0.1, 0, 0.3, 0; keeping trained values gives 1, 0, 3, 0. Two of four connections remain, with no accuracy measurement.
Illustrative initial weights 0.1, −0.2, 0.3, −0.4 and trained weights 1, −2, 3, −4 under stipulated mask 1, 0, 1, 0. Resetting gives 0.1, 0, 0.3, 0; keeping trained values gives 1, 0, 3, 0. Two of four connections remain, with no accuracy measurement.
An original reset ledger with fixed, dimensionless values. The mask is stipulated; it neither follows from magnitude pruning here nor demonstrates a winning ticket.

Separate the experiments

Experiment on the retained connectionsStarting valuesQuestion
Fine-tune a pruned modelTrained weightsCan the sparse model recover or preserve its fit?
Original lottery-ticket testSaved initial weightsCould this subnetwork have learned in isolation from its original start?
Random-reinitialisation controlFresh random weights, same maskIs the selected structure sufficient without the saved initial values?

The original paper's reinitialisation experiments found that replacing the saved initial values could substantially worsen performance. This is evidence about those settings, not a universal guarantee that fresh initialisation always fails.

Rewinding changes the starting point. Frankle et al. (2020), §4, extend the procedure by restoring surviving weights to an early training checkpoint θk\theta_k, rather than θ0\theta_0. Their larger vision-network experiments show that this can recover matching subnetworks where resetting to initialisation fails. A checkpoint with k>0k>0 already contains learning, so this result is distinct from discovering a trainable subnetwork at random initialisation. The authors call the rewound subnetworks matching, reserving winning ticket for the initialisation case.

What a quant experiment must keep separate

A sparse return-prediction network might match its dense baseline on one split and lose that advantage in a later market period. The lottery-ticket idea supplies an experimental question; it supplies no evidence of a financial signal or its persistence.

For a meaningful comparison, keep the data construction, optimiser, batch size, learning-rate schedule and stopping rule explicit. Include the dense baseline and the same-mask random-reinitialisation control. Discover masks from training data, select settings on chronological validation data and leave a later test period untouched. Searching masks against the final test result introduces overfitting just like searching model hyperparameters.
Also report discovery cost separately from retraining cost. Finding a mask can require training the dense network and repeating pruning rounds. A successful sparse retraining run does not erase that expenditure. An iteration count is not elapsed time, either: an irregular sparse mask may still execute through dense operations. Measure latency and memory on the intended implementation; 50% sparsity alone establishes neither a twofold speedup nor a halving of total research compute.
For background on starting-weight scale, SelfMadeQuant's initialisation lesson includes a section on Glorot and He initialisation. The preview is public; the full lesson requires membership.

Sources

Last updated
  • neural networks
  • pruning
  • initialisation
  • sparse networks
← Back to the glossaryBack to top ↑