arXiv:2412.06875cs.LGcs.AI2024-12

用统一码本压缩多个神经网络,提升效率并减少内存占用

VQ4ALL: Efficient Neural Network Representation via a Universal Codebook

  • 自底向上构建通用码本,共享所有网络层的编码表
  • 压缩率超16倍,多架构下仍保持高精度
  • 适合部署在资源受限设备的轻量化模型设计

大型神经网络的快速发展对轻量级表示方法提出了新要求。传统基于模型压缩的方法虽已取得显著成果,尤其以向量量化(VQ)技术实现高比率压缩,但因每层需独立构建码表,导致压缩率受限且频繁访问内存。本文提出一种自底向上的方法,使多个神经网络共享一个通用码本,不仅减少码表数量,还通过将静态码表存入片上只读存储器(ROM),降低内存访问频率和芯片面积。我们提出VQ4ALL,一种基于向量量化的通用表示方法,利用核密度估计提取通用码本,并通过可微分配逐步构建不同低比特网络。实验表明,VQ4ALL在多种网络架构下实现超过16倍的压缩率,同时保持高精度,展现出卓越的有效性与通用性。

原文摘要 · Abstract (English)

The rapid growth of the big neural network models puts forward new requirements for lightweight network representation methods. The traditional methods based on model compression have achieved great success, especially VQ technology which realizes the high compression ratio of models by sharing code words. However, because each layer of the network needs to build a code table, the traditional top-down compression technology lacks attention to the underlying commonalities, resulting in limited compression rate and frequent memory access. In this paper, we propose a bottom-up method to share the universal codebook among multiple neural networks, which not only effectively reduces the number of codebooks but also further reduces the memory access and chip area by storing static code tables in the built-in ROM. Specifically, we introduce VQ4ALL, a VQ-based method that utilizes codewords to enable the construction of various neural networks and achieve efficient representations. The core idea of our method is to adopt a kernel density estimation approach to extract a universal codebook and then progressively construct different low-bit networks by updating differentiable assignments. Experimental results demonstrate that VQ4ALL achieves compression rates exceeding 16 $\times$ while preserving high accuracy across multiple network architectures, highlighting its effectiveness and versatility.

模型压缩向量量化通用码本轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。