[edit]
Stochastic Layer-Wise Precision in Deep Neural Networks
Proceedings of the 34th Conference on Uncertainty in Artificial Intelligence, PMLR R16:662-671, 2018.
Abstract
Low precision weights, activations, and gradi- ents have been proposed as a way to improve the computational efficiency and memory foot- print of deep neural networks. Recently, low precision networks have even shown to be more robust to adversarial attacks. How- ever, typical implementations of low precision DNNs use uniform precision across all lay- ers of the network. In this work, we explore whether a heterogeneous allocation of preci- sion across a network leads to improved per- formance, and introduce a learning scheme where a DNN stochastically explores multi- ple precision configurations through learning. This permits a network to learn an optimal pre- cision configuration. We show on convolu- tional neural networks trained on MNIST and ILSVRC12 that even though these nets learn a uniform or near-uniform allocation strat- egy respectively, stochastic precision leads to a favourable regularization effect improving generalization.