English

Distributed Low Precision Training Without Mixed Precision

Computer Vision and Pattern Recognition 2019-12-30 v2 Distributed, Parallel, and Cluster Computing

Abstract

Low precision training is one of the most popular strategies for deploying the deep model on limited hardware resources. Fixed point implementation of DCNs has the potential to alleviate complexities and facilitate potential deployment on embedded hardware. However, most low precision training solution is based on a mixed precision strategy. In this paper, we have presented an ablation study on different low precision training strategy and propose a solution for IEEE FP-16 format throughout the training process. We tested the ResNet50 on 128 GPU cluster on ImageNet-full dataset. We have viewed that it is not essential to use FP32 format to train the deep models. We have viewed that communication cost reduction, model compression, and large-scale distributed training are three coupled problems.

Keywords

Cite

@article{arxiv.1911.07384,
  title  = {Distributed Low Precision Training Without Mixed Precision},
  author = {Zehua Cheng and Weiyang Wang and Yan Pan and Thomas Lukasiewicz},
  journal= {arXiv preprint arXiv:1911.07384},
  year   = {2019}
}