将低秩与量化结合,高效压缩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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。