arrow
Return

Universal Deep Neural Network Compression

delete2020-05-01
delete46
delete
OA
AI
Y
Yoojin Choi *
M
Mostafa El‐Khamy
J
Jungwon Lee
DOI:10.1109/JSTSP.2020.2975903delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
We consider compression of deep neural networks (DNNs) by weight quantization and lossless source coding for memory-efficient deployment. Whereas the previous work addressed non-universal scalar quantization and entropy source coding, we for the first time introduce universal DNN compression by universal vector quantization and universal source coding. In particular, the proposed scheme utilizes universal lattice quantization, which randomizes the source by uniform random dithering before lattice quantization and can perform near-optimally on any source without relying on knowledge of the source distribution. Moreover, we present a method of fine-tuning vector quantized DNNs to recover any accuracy loss due to quantization. From our experiments, we show that the proposed scheme compresses the MobileNet and ShuffleNet models trained on ImageNet with the state-of-the-art compression ratios of 10.7 and 8.8, respectively.
Keywords:
Entropy
Vector quantization
Lattices
Source coding
Image coding
Neural networks
Deep neural networks
lossy compression
universal compression
entropy coded vector quantization
universal quantization
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Journal of Selected Topics in Signal Processing cover
IEEE Journal of Selected Topics in Signal Processing
IF:
13.7
Papers:
1.9K
Citations:
1.1W

Organization

S
Samsung Electronics
Scholars:
3.0K
Papers: 2.0K
Citations: 21