AWQ: Activation-aware Weight Quantization for On-Device LLM Compression and Acceleration

By Ji Lin, Jiaming Tang, Haotian Tang, Shang Yang, Xingyu Dang, Song Han. 2023

1,557 citations recorded by the source corpus.

Large language models (LLMs) have transformed numerous AI applications. On-device LLM is becoming increasingly important: running LLMs locally on edge devices can reduce cloud computing costs and protect users' privacy. However, the astronomical model size and the limited hardware resources pose significant deployment challenges. To solve these issues, we propose Activation-aware Weight Quantization (AWQ) and TinyChat, an algorithm-system full-stack solution for efficient on-device LLM deployment. AWQ is a novel quantization method that identifies and protects salient weights based on activation distribution, significantly reducing model size while preserving performance. TinyChat, an optimized inference framework, translates AWQ's theoretical memory savings into practical speedups through techniques such as on-the-fly dequantization, SIMD-aware weight packing, and kernel fusion. Together, they enable 4x model size reduction and 3-4x acceleration across various edge platforms, from high-end desktop GPUs to resource-constrained IoT devices. This solution democratizes on-device LLM deployment, offering privacy-preserving, low-latency AI capabilities across a wide range of applications.

Subfields: Computer Vision and Pattern Recognition, Electrical and Electronic Engineering, Large Language Models, Efficiency & Compression, Signal Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Vision and Pattern Recognition

  1. Kurt Keutzer
  2. Yu Qiao
  3. Michael W. Mahoney
  4. Luc Van Gool
  5. A. Gholami
  6. Dacheng Tao
  7. Xiaogang Wang
  8. Ping Luo

In Electrical and Electronic Engineering

  1. Kurt Keutzer
  2. A. Gholami
  3. Michael W. Mahoney
  4. Zhen Dong
  5. Yu Qiao
  6. Peng Gao
  7. Wenqi Shao
  8. Zhiqiang Li

In Large Language Models

  1. Furu Wei
  2. Shaohan Huang
  3. Shuming Ma
  4. Wenhui Wang
  5. A. Gholami
  6. Michael W. Mahoney
  7. Yu Qiao
  8. Peng Gao

In Efficiency & Compression

  1. Kurt Keutzer
  2. A. Gholami
  3. Michael W. Mahoney
  4. Sehoon Kim
  5. Zhen Dong
  6. Sheng Shen
  7. Xiuyu Li
  8. Coleman Hooper

In Signal Processing

  1. Yu Qiao
  2. Kaipeng Zhang
  3. A. Gholami
  4. Zhen Dong
  5. Michael W. Mahoney
  6. Mickael Rouvier
  7. Kurt Keutzer
  8. Linquan Liu

Citations in the corpus