arXiv:2507.09616cs.LGcs.CV2025-07被引 1

将低秩与量化结合,高效压缩Transformer模型

MLoRQ: Bridging Low-Rank and Quantization for Transformer Compression

  • 分两阶段优化每层的秩和比特位数,满足内存约束
  • 在图像分类等任务中性能提升最高达15%
  • 可适配现有量化方法,适合边缘设备部署

在资源受限的边缘设备上部署基于Transformer的神经网络面临巨大挑战。通常通过低秩近似和混合精度量化等技术应对。本文提出混合低秩与量化(MLoRQ)方法,整合两种技术。该方法采用两阶段优化:(i) 层内优化,从所有低秩与量化组合中筛选最优压缩方案;(ii) 层间优化,在满足预设内存约束的前提下,为各层分配比特位精度与秩。可选的最后一步采用改进的自适应舍入技术,缓解联合低秩近似与量化带来的误差。该方法兼容主流量化算法,且在视觉Transformer的图像分类、目标检测和实例分割任务中实现当前最佳表现,性能提升最高达15%。

原文摘要 · Abstract (English)

Deploying transformer-based neural networks on resource-constrained edge devices presents a significant challenge. This challenge is often addressed through various techniques, such as low-rank approximation and mixed-precision quantization. In this work, we introduce Mixed Low-Rank and Quantization (MLoRQ), a novel method that integrates both techniques. MLoRQ employs a two-stage optimization process to determine optimal bit-width and rank assignments for each layer, adhering to predefined memory constraints. This process includes: (i) an intra-layer optimization that identifies potentially optimal compression solutions out of all low-rank and quantization combinations; (ii) an inter-layer optimization that assigns bit-width precision and rank to each layer while ensuring the memory constraint is met. An optional final step applies a sequential optimization process using a modified adaptive rounding technique to mitigate compression-induced errors in joint low-rank approximation and quantization. The method is compatible and can be seamlessly integrated with most existing quantization algorithms. MLoRQ shows state-of-the-art results with up to 15\% performance improvement, evaluated on Vision Transformers for image classification, object detection, and instance segmentation tasks.

模型压缩低秩分解量化Transformer

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