Reproducing Lottery Ticket Hypothesis : Finding Sparse, Trainable Neural Networks (ICLR 2019)
This artifact reproduces the core findings of "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks" (ICLR 2019) by Jonathan Frankle and Michael Carbin. The experiment uses a LeNet architecture trained on the MNIST dataset, applies iterative magnitude-based pruning, resets surviving weights to their original initialization, and retrains with early stopping. Each pruning iteration removes 20% of the lowest-magnitude weights and the process is repeated for 5 iterations over 5 separate trials. The results confirm that sparse subnetworks ("winning tickets") can train to comparable accuracy as the original dense model.
Launching this artifact will open it within Chameleon’s shared Jupyter experiment environment, which is accessible to all Chameleon users with an active allocation.
Download ArchiveDownload an archive containing the files of this artifact.
Download with git
Clone the git repository for this artifact, and checkout the version's commit
git clone https://github.com/codenameyizzz/yizreel_repro_the_lottery_ticket_hypothesis.git
# cd into the created directory
git checkout 6f9b91fd015b9e08d98eee2c4511d8452f6322e4
Submit feedback through GitHub issues