arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.05529cs.LG

Recurrence of Optimum for Training Weight and Activation Quantized Networks

University of California, Irvine(加州大学欧文分校);University at Albany, State University of New York(纽约州立大学奥尔巴尼分校)

Ziang Long, Penghang Yin, Jack Xin

英文摘要:

Deep neural networks (DNNs) are quantized for efficient inference on resource-constrained platforms. However, training deep learning models with low-precision weights and activations involves a demanding optimization task, which calls for minimizing a stage-wise loss function subject to a discrete set-constraint. While numerous training methods have been proposed, existing studies for full quantization of DNNs are mostly empirical. From a theoretical point of view, we study practical techniques for overcoming the combinatorial nature of network quantization. Specifically, we investigate a simple yet powerful projected gradient-like algorithm for quantizing two-linear-layer networks, which proceeds by repeatedly moving one step at float weights in the negation of a heuristic \emph{fake} gradient of the loss function (so-called coarse gradient) evaluated at quantized weights. For the first time, we prove that under mild conditions, the sequence of quantized weights recurrently visits the global optimum of the discrete minimization problem for training fully quantized network. We also show numerical evidence of the recurrence phenomenon of weight evolution in training quantized deep networks.

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

↑