UniQL: Unified Quantization and Low-rank Compression for Adaptive Edge LLMs

TL;DR

UniQL is a comprehensive framework that combines quantization and low-rank compression techniques to enable efficient deployment of large language models on mobile devices, reducing memory and increasing throughput while maintaining accuracy.

Contribution

UniQL introduces a unified post-training compression framework with on-device configurable pruning, integrating quantization and low-rank methods for diverse edge LLMs.

Findings

Achieves 4x-5.7x memory reduction in models.

Improves token throughput by 2.7x-3.4x.

Maintains accuracy within 5% at 15% pruning.

Abstract

Deploying large language models (LLMs) on mobile platforms faces significant challenges due to the limited memory and shared computational resources of the device. Resource availability may be an issue as it is directly impacted by the current device workload, adding to the uncertainty of model deployment. We introduce UniQL, a unified post-training quantization and low-rank compression framework with on-device configurable pruning rates for edge LLMs. UniQL is a general framework that integrates quantization and low-rank compression for Transformers, State Space Models (SSMs), and hybrid models to support diverse edge applications. In our proposed joint framework, we introduce an efficient structured weight-sorting method that speeds up computation by 20x, quantization-aware singular value decomposition (SVD) to minimize quantization errors, state-aware weight sorting for SSMs, and a fused rotary positional embedding (RoPE) kernel for pruned models. Our framework performs weight-sorting, fine-tuning, and quantization in the cloud in a single-pass workflow, while enabling on-device configurable pruning rates up to 35%. Our experiments show that quantized and pruned models achieve a memory reduction of 4x-5.7x and a token-throughput improvement of 2.7x-3.4x, maintaining accuracy within 5% of the original models at 15% pruning across Transformers (Llama3 and Qwen2.5), SSMs (Mamba2), and hybrid models (Nemotron-H and Bamba-v2). The code and quantized models are available at: https://github.com/enyac-group/UniQL.