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

有损与无损(L$^2$)训练后模型尺寸压缩

Lossy and Lossless (L$^2$) Post-training Model Size Compression

Beihang University(北京航空航天大学);SenseTime Research(商汤科技研究院)

Yumeng Shi, Shihao Bai, Xiuying Wei, Ruihao Gong, Jianlei Yang

AI总结:

本文提出一种将有损与无损压缩统一结合的训练后模型尺寸压缩方法,通过统一参数化权重变换和可微分计数器优化,实现稳定10倍无损精度压缩及20倍轻微精度损失压缩。

摘要:

深度神经网络已展现出卓越的性能,并被广泛应用于各类视觉任务中。然而,其庞大的模型尺寸给传输和存储带来了显著不便。许多先前的研究已经探索了模型尺寸压缩,但这些研究往往孤立地处理各种有损和无损压缩方法,导致难以高效地实现高压缩比。本文提出了一种训练后模型尺寸压缩方法,将有损压缩和无损压缩以统一的方式结合起来。我们首先提出了一种统一的参数化权重变换,确保不同的有损压缩方法能够在训练后以联合方式进行。然后,引入一个专用的可微分计数器来指导有损压缩的优化,使其到达一个更适合后续无损压缩的位置。此外,我们的方法可以轻松控制期望的全局压缩比,并为不同层分配自适应压缩比。最终,我们的方法能够在短时间内实现稳定的$10\ imes$压缩比而不牺牲精度,以及$20\ imes$压缩比且仅有轻微精度损失。我们的代码可在 https://github.com/ModelTC/L2_Compression 获取。

英文摘要:

Deep neural networks have delivered remarkable performance and have been widely used in various visual tasks. However, their huge size causes significant inconvenience for transmission and storage. Many previous studies have explored model size compression. However, these studies often approach various lossy and lossless compression methods in isolation, leading to challenges in achieving high compression ratios efficiently. This work proposes a post-training model size compression method that combines lossy and lossless compression in a unified way. We first propose a unified parametric weight transformation, which ensures different lossy compression methods can be performed jointly in a post-training manner. Then, a dedicated differentiable counter is introduced to guide the optimization of lossy compression to arrive at a more suitable point for later lossless compression. Additionally, our method can easily control a desired global compression ratio and allocate adaptive ratios for different layers. Finally, our method can achieve a stable $10\times$ compression ratio without sacrificing accuracy and a $20\times$ compression ratio with minor accuracy loss in a short time. Our code is available at https://github.com/ModelTC/L2_Compression .

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

↑