Weighted quantization-regularization in DNNs for weight memory minimization toward HW implementation

M Wess, SMP Dinakarrao…�- IEEE Transactions on�…, 2018 - ieeexplore.ieee.org
M Wess, SMP Dinakarrao, A Jantsch
IEEE Transactions on Computer-Aided Design of Integrated Circuits�…, 2018ieeexplore.ieee.org
Deployment of deep neural networks on hardware platforms is often constrained by limited
on-chip memory and computational power. The proposed weight quantization offers the
possibility of optimizing weight memory alongside transforming the weights to hardware
friendly data types. We apply dynamic fixed point (DFP) and power-of-two (Po2) quantization
in conjunction with layer-wise precision scaling to minimize the weight memory. To alleviate
accuracy degradation due to precision scaling, we employ quantization-aware fine-tuning�…
Deployment of deep neural networks on hardware platforms is often constrained by limited on-chip memory and computational power. The proposed weight quantization offers the possibility of optimizing weight memory alongside transforming the weights to hardware friendly data types. We apply dynamic fixed point (DFP) and power-of-two (Po2) quantization in conjunction with layer-wise precision scaling to minimize the weight memory. To alleviate accuracy degradation due to precision scaling, we employ quantization-aware fine-tuning. For fine-tuning, quantization-regularization (QR) and weighted QR are introduced to force the trained quantization by adding the distance of the weights to the desired quantization levels as a regularization term to the loss-function. While DFP quantization performs better when allowing different bit-widths for each layer, Po2 quantization in combination with retraining allows higher compression rates for equal bit-width quantization. The techniques are verified on an all-convolutional network. With accuracy degradation of 0.10% points, for DFP with layer-wise precision scaling we achieve compression ratios of 7.34 for CIFAR-10, 4.7 for CIFAR-100, and 9.33 for SVHN dataset.
ieeexplore.ieee.org
Showing the best result for this search. See all results