Awesome

Any-Precision Deep Neural Network

Official code and models in PyTorch for the paper Any-Precision Deep Neural Networks.

Run

Environment

• Python 3.7
• PyTorch 1.1.0
• torchvision 0.2.1
• tensorboardX
• gpustat

Train

Resnet-20 Models on CIFAR10

Run the script below and dataset will download automatically.

./train_cifar10.sh

SVHN Models on SVHN

Run the script below and dataset will download automatically.

./train_svhn.sh

Resnet-50 Models on ImageNet

Before running the script below, one needs to manually download ImageNet and save it properly according to data_paths in dataset/data.py.

./train_imagenet.sh

Test

To test a trained model, simply run the corresponding training script for one epoch with pretrained model loaded and without the training part.

Trained Models

Due to the following listed training hyperparameter changes, numbers below may be different from those in the paper.

• Init lr for any-precision models: 0.1 -> 0.5.
• We use ReLU for 32-bit model instead of Clamp (check here).
• We use tanh nonlinearity for 32-bit model for consistency with other precisions (check here).

Resnet-20 Models on CIFAR10

1: Softmax Cross Entropy Loss
2: Softmax Cross Entropy Loss with FP32 prediction as supervision
3: Softmax Cross Entropy Loss with higher-precision model as supervision for lower-precision model

SVHN Models on SVHN

Resnet-50 Models on ImageNet

4: Weight decay 1e-5
5: Weight decay 1e-4

Citation

If you find this repository helpful, please consider citing our paper:

@article{yu2019any,
  title={Any-Precision Deep Neural Networks},
  author={Yu, Haichao and Li, Haoxiang and Shi, Honghui and Huang, Thomas S and Hua, Gang},
  journal={arXiv preprint arXiv:1911.07346},
  year={2019}
}

Contact

Please feel free to contact Haichao Yu at haichao.yu@outlook.com for any issue.