arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.12422cs.CVcs.AI
收藏

基于张量分解的DNN模型压缩及其优化框架

Towards Efficient Tensor Decomposition-Based DNN Model Compression with Optimization Framework

Rutgers University(罗格斯大学);Amazon(亚马逊)

Miao Yin, Yang Sui, Siyu Liao, Bo Yuan

摘要:

高级张量分解,例如张量训练(TT)和张量环(TR),已被广泛研究用于深度神经网络(DNN)模型压缩,特别是用于递归神经网络(RNN)。然而,使用TT/TR压缩卷积神经网络(CNN)总是遭受显著的精度损失。在本文中,我们提出了一个系统的框架,张量分解为基础的模型压缩使用交替方向乘法(ADMM)。通过将基于TT分解的模型压缩转化为一个具有张量秩约束的优化问题,利用ADMM技术以迭代方式系统地求解该优化问题.在此过程中,整个DNN模型以原始结构而不是TT格式进行训练,但逐渐享有所需的低张量秩特性。然后,我们将这个未压缩的模型分解为TT格式,并对其进行微调,最终获得高精度的TT格式DNN模型。我们的框架非常通用,适用于CNN和RNN,并且可以很容易地修改以适应其他张量分解方法。我们评估我们提出的框架上不同的DNN模型的图像分类和视频识别任务。实验结果表明,我们的ADMM为基础的TT格式的模型表现出非常高的压缩性能和高精度。值得注意的是,在具有2.3倍和2.4倍压缩比的CIFAR-100上,我们的模型分别比原始ResNet-20和ResNet-32高出1.96%和2.21%。对于在ImageNet上压缩ResNet-18,我们的模型实现了2.47倍的FLOPs减少,而没有准确性损失。

英文摘要:

Advanced tensor decomposition, such as Tensor train (TT) and Tensor ring (TR), has been widely studied for deep neural network (DNN) model compression, especially for recurrent neural networks (RNNs). However, compressing convolutional neural networks (CNNs) using TT/TR always suffers significant accuracy loss. In this paper, we propose a systematic framework for tensor decomposition-based model compression using Alternating Direction Method of Multipliers (ADMM). By formulating TT decomposition-based model compression to an optimization problem with constraints on tensor ranks, we leverage ADMM technique to systemically solve this optimization problem in an iterative way. During this procedure, the entire DNN model is trained in the original structure instead of TT format, but gradually enjoys the desired low tensor rank characteristics. We then decompose this uncompressed model to TT format and fine-tune it to finally obtain a high-accuracy TT-format DNN model. Our framework is very general, and it works for both CNNs and RNNs, and can be easily modified to fit other tensor decomposition approaches. We evaluate our proposed framework on different DNN models for image classification and video recognition tasks. Experimental results show that our ADMM-based TT-format models demonstrate very high compression performance with high accuracy. Notably, on CIFAR-100, with 2.3X and 2.4X compression ratios, our models have 1.96% and 2.21% higher top-1 accuracy than the original ResNet-20 and ResNet-32, respectively. For compressing ResNet-18 on ImageNet, our model achieves 2.47X FLOPs reduction without accuracy loss.

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

↑