Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

Compressibility and Generalization in Large-Scale Deep Learning

This repository contains everything necessary to reproduce the experiments in our paper: Compressibility and Generalization in Large-Scale Deep Learning (Wenda Zhou, Victor Veitch, Morgane Austern, Ryan P. Adams and Peter Orbanz).

LeNet-5 pruning

The directory compression_lenet contais the scripts which implement the training, pruning, quantization and evaluation of the LeNet-5 network.

MobileNet Pruning

The directory compression-mobilenet contains the scripts which implement the pruning, quantization, and evaluation of noise stability of our network. The typical compression pipeline would be to take an already trained network (such as those available here) and prune then quantize the network. Please see the readme for more details on compressing MobileNet.

CIFAR-10 randomization tests

The directory randomization-cifar contains the scripts which implement the training and pruning of a ResNet-56 on the CIFAR-10 dataset with a portion of the labels randomized.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages