arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1902.08153cs.LGstat.ML
收藏

学习步长Quantization

Learned Step Size Quantization

IBM Research(IBM研究院)

Steven K. Esser, Jeffrey L. McKinstry, Deepika Bablani, Rathinakumar Appuswamy, Dharmendra S. Modha

摘要:

深度网络在推理时使用低精度运算,与高精度替代方案相比具有功率和空间优势,但需要克服随着精度降低而保持高精度的挑战。在这里,我们提出了一种训练这种网络的方法,学习步长量化,当使用来自各种架构的模型时,在ImageNet数据集上实现了迄今为止最高的准确度,权重和激活量化到2位,3位或4位精度,并且可以训练达到全精度基线精度的3位模型。我们的方法建立在现有的方法,通过改进量化器本身的配置来学习量化网络中的权重。具体来说,我们引入了一种新的方法来估计和缩放每个权重和激活层量化器步长的任务损失梯度,以便它可以与其他网络参数一起学习。这种方法根据给定系统的需要使用不同的精度水平,并且只需要对现有的训练代码进行简单的修改。

英文摘要:

Deep networks run with low precision operations at inference time offer power and space advantages over high precision alternatives, but need to overcome the challenge of maintaining high accuracy as precision decreases. Here, we present a method for training such networks, Learned Step Size Quantization, that achieves the highest accuracy to date on the ImageNet dataset when using models, from a variety of architectures, with weights and activations quantized to 2-, 3- or 4-bits of precision, and that can train 3-bit models that reach full precision baseline accuracy. Our approach builds upon existing methods for learning weights in quantized networks by improving how the quantizer itself is configured. Specifically, we introduce a novel means to estimate and scale the task loss gradient at each weight and activation layer's quantizer step size, such that it can be learned in conjunction with other network parameters. This approach works using different levels of precision as needed for a given system and requires only a simple modification of existing training code.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑