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 389bc333fcf6849d3f326423abb57d2c0d1abe5f
Submit feedback through GitHub issues