arrow
Return

Loss aware post-training quantization

delete2021-10-01
delete54
delete
OA
AI
Y
Yury Nahshan
B
Brian Chmiel
C
Chaim Baskin
E
Evgenii Zheltonozhskii *
R
Ron Banner
A
Alex Bronstein
A
Avi Mendelson
DOI:10.1007/s10994-021-06053-zdelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Neural network quantization enables the deployment of large models on resource-constrained devices. Current post-training quantization methods fall short in terms of accuracy for INT4 (or lower) but provide reasonable accuracy for INT8 (or above). In this work, we study the effect of quantization on the structure of the loss landscape. We show that the structure is flat and separable for mild quantization, enabling straightforward post-training quantization methods to achieve good results. We show that with more aggressive quantization, the loss landscape becomes highly non-separable with steep curvature, making the selection of quantization parameters more challenging. Armed with this understanding, we design a method that quantizes the layer parameters jointly, enabling significant accuracy improvement over current post-training quantization methods. Reference implementation is available at lattps://github.conilynahshaninn-quantization-pytorch/tree/masterilapq.
Keywords:
Convolutional neural networks
Post-training quantization

Journal

Machine Learning cover
Machine Learning
IF:
2.9
Papers:
2.7K
Citations:
3.4W

Organization

T
Technion Israel Institute of Technology
Scholars:
1.6W
Papers: 1.5W
Citations: 2.0W
Cited Papers

Cited Papers

err
IF0
err
err0
PREAI
err
errShare
errSave
Towards energy-autonomous wake-up receiver using Visible Light Communication
err2016-01-01
err0
errOAAI
errJoyce Sariol Ramos; Ilker Demirkol; Josep Paradells; Daniel Vossing; Karim M. Gad; Martin Kasemann
errShare
errSave
Numerical Analysis of Thermal Stress for a Stack of Planar Solid Oxide Fuel Cells
err2022-01-04
err0
errOAAI
errJianmin Zheng; Liusheng Xiao; Mingtao Wu; Shaocheng Lang; Zhonggang Zhang; Ming Chen; Jinliang Yuan
errShare
errSave
errShare
errSave
The comprehension of ambiguous idioms in aphasic patients
err2006-01-01
err0
PREAI
errC CACCIARI; F REATI; M COLOMBO; R PADOVANI; S RIZZO; C PAPAGNO
errShare
errSave
researcher View more