A large scale study of Knowledge Distillation. NYU Computer Vision Project
This codebase only supports Python 3.6+.
Required Python packages:
torch torchvision tqdm numpy pandas seaborn
All packages can be installed using
pip3 install --user -r requirements.txt.
This project is also integerated with Pytorch Lightning. Use the lightning branch to see Pytorch Lightning compatible code.
The benchmarks can be run via
python3 evaluate_kd.py and providing the
respective command line parameters. For example:
python3 evaluate_kd.py --epochs 200 --teacher resnet18 --student resnet8 --dataset cifar10 --teacher-checkpoint pretrained/resnet18_cifar10_95260_parallel.pth --mode nokd kd
Runs basic student training and knowledge distillation for 200 epochs using a
pretrained teacher. There are checkpoints of multiple models in the pretrained folder.
Supported distillation modes
Plain training with no knowledge distillation.
Hinton loss to distill a student network.
Distill from a list of teacher models and pick the best performing one.
Distill using varying combinations of temperature and alpha and pick the best performing combination.
Knowledge Distillation with a triplet loss using the student as negative example.
Train a student under an ensemble of students that are picked from a list.
Run knowledge distillation in combination with unsupervised data augmentation.
Run distillation using Teacher-Assistant distillation.
Run feature distillation using Activation-Boundary distillation.
Run feature distillation using the Feature Overhaul distillation.
Run distillation using the Relational Knowledge distillation.
Run feature distillation using the Patient Knowledge distillation.
Runs a custom feature distillation distillation (Simple Feature Distillation) that just pools and flattens feature layers.
Subscribe to Python Awesome
Get the latest posts delivered right to your inbox