arrow
返回

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
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

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.
Keyword:
Convolutional neural networks
Post-training quantization

期刊

Machine Learning 封面图
Machine Learning
IF:
2.9
论文数:
2.7K
被引数:
3.4W

机构

T
Technion Israel Institute of Technology
学者数:
1.6W
论文数: 1.5W
被引数: 2.0W
引用论文

引用论文

err
IF0
err
err0
PREAI
err
err分享
err收藏
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
err分享
err收藏
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
err分享
err收藏
err分享
err收藏
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
err分享
err收藏
学者 查看更多内容