arXiv:2507.22349cs.LG2025-07ICCV被引 2

提出MSQ方法,让模型在低内存下高效训练,同时保持高精度。

MSQ: Memory-Efficient Bit Sparsification Quantization

  • 用可微分的舍入截断机制,直接优化权重低位比特。
  • 训练参数减少8倍,时间缩短86%,精度不降。
  • 适合移动端和边缘设备的轻量级模型训练。

随着深度神经网络在移动和边缘设备上的广泛应用,模型效率优化变得至关重要。混合精度量化因其在效率与精度间更优的平衡而被广泛采用,但确定每层最优精度仍具挑战。近期基于位级稀疏性的研究虽有潜力,却常带来显著的训练复杂度和高显存需求。本文提出一种新型方法——记忆高效的位级稀疏量化(MSQ),通过引入可微分的舍入截断量化器,实现对模型权重最低有效位(LSBs)的可导计算;并结合正则化使这些低位比特产生稀疏性,从而在无需显式位级参数拆分的情况下实现有效精度降低。此外,MSQ引入海森信息,支持同时剪枝多个低位比特,进一步提升训练效率。实验表明,与以往位级量化相比,MSQ可实现高达8.00倍的可训练参数减少和最高86%的训练时间节省,同时保持优异的精度与压缩率,为资源受限设备上的高效DNN训练提供了实用方案。

原文摘要 · Abstract (English)

As deep neural networks (DNNs) see increased deployment on mobile and edge devices, optimizing model efficiency has become crucial. Mixed-precision quantization is widely favored, as it offers a superior balance between efficiency and accuracy compared to uniform quantization. However, finding the optimal precision for each layer is challenging. Recent studies utilizing bit-level sparsity have shown promise, yet they often introduce substantial training complexity and high GPU memory requirements. In this paper, we propose Memory-Efficient Bit Sparsification Quantization (MSQ), a novel approach that addresses these limitations. MSQ applies a round-clamp quantizer to enable differentiable computation of the least significant bits (LSBs) from model weights. It further employs regularization to induce sparsity in these LSBs, enabling effective precision reduction without explicit bit-level parameter splitting. Additionally, MSQ incorporates Hessian information, allowing the simultaneous pruning of multiple LSBs to further enhance training efficiency. Experimental results show that MSQ achieves up to 8.00x reduction in trainable parameters and up to 86% reduction in training time compared to previous bit-level quantization, while maintaining competitive accuracy and compression rates. This makes it a practical solution for training efficient DNNs on resource-constrained devices.

量化稀疏化边缘计算

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