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

用于深度网络压缩的簇正则化Quantization

Cluster Regularized Quantization for Deep Networks Compression

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);University of Chinese Academy of Sciences(中国科学院大学)

Yiming Hu, Jianquan Li, Xianlei Long, Shenhua Hu, Jiagang Zhu, Xingang Wang, Qingyi Gu

摘要:

深度神经网络(DNN)在计算机视觉领域取得了巨大的成功,但由于其高存储和计算成本,在移动设备上的应用受到限制。已经投入了很多努力来压缩DNN。在本文中,我们提出了一种简单而有效的深度网络压缩方法,称为聚类正则化量化(CRQ),它可以将全精度模型的表示精度降低到三进制值,而不会显着降低精度。特别是,所提出的方法旨在通过引入聚类正则化项来减少量化误差,该聚类正则化项施加在全精度权重上,使它们自然地集中在目标值周围。通过在重训练阶段显式正则化权值,全精度模型可以实现向低比特模型的平滑过渡。在基准数据集上的综合实验证明了该方法的有效性。

英文摘要:

Deep neural networks (DNNs) have achieved great success in a wide range of computer vision areas, but the applications to mobile devices is limited due to their high storage and computational cost. Much efforts have been devoted to compress DNNs. In this paper, we propose a simple yet effective method for deep networks compression, named Cluster Regularized Quantization (CRQ), which can reduce the presentation precision of a full-precision model to ternary values without significant accuracy drop. In particular, the proposed method aims at reducing the quantization error by introducing a cluster regularization term, which is imposed on the full-precision weights to enable them naturally concentrate around the target values. Through explicitly regularizing the weights during the re-training stage, the full-precision model can achieve the smooth transition to the low-bit one. Comprehensive experiments on benchmark datasets demonstrate the effectiveness of the proposed method.

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

↑