用无损压缩技术让大模型体积缩小超50%,还能提速
ZipNN: Lossless Compression for AI Models
- 针对神经网络设计专用无损压缩算法,可还原原始模型
- 在Llama 3等模型上实现最高50%的存储节省,平均超17%
- 适合需要高效部署大模型的云服务与边缘设备
随着模型规模扩大和部署范围增长,其庞大的体积对网络和存储造成巨大压力。尽管已有大量研究通过删减模型权重来加速推理,本文探索更传统的无损压缩方法——将模型以紧凑形式存储,并通过解压算法完整恢复。我们提出针对神经网络的无损压缩方案ZipNN。令人意外的是,该方法在主流模型上实现了显著的网络与存储节省,通常可减少33%的模型体积,某些情况下超过50%。我们分析了模型可压缩性的来源,并引入针对性压缩变体以提升效果。在典型模型(如Llama 3)上,ZipNN相比通用压缩方案节省空间超17%,同时压缩与解压速度提升62%。估算表明,此类方法每月可为像Hugging Face这样的大型模型仓库节省超过1艾字节的下载流量。
原文摘要 · Abstract (English)
With the growth of model sizes and the scale of their deployment, their sheer size burdens the infrastructure requiring more network and more storage to accommodate these. While there is a vast model compression literature deleting parts of the model weights for faster inference, we investigate a more traditional type of compression - one that represents the model in a compact form and is coupled with a decompression algorithm that returns it to its original form and size - namely lossless compression. We present ZipNN a lossless compression tailored to neural networks. Somewhat surprisingly, we show that specific lossless compression can gain significant network and storage reduction on popular models, often saving 33% and at times reducing over 50% of the model size. We investigate the source of model compressibility and introduce specialized compression variants tailored for models that further increase the effectiveness of compression. On popular models (e.g. Llama 3) ZipNN shows space savings that are over 17% better than vanilla compression while also improving compression and decompression speeds by 62%. We estimate that these methods could save over an ExaByte per month of network traffic downloaded from a large model hub like Hugging Face.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。