atlas

Research map / Architectures and efficiency / Quantization

Agile-Quant: Activation-Guided Quantization for Faster Inference of LLMs on the Edge

AAAI 2024 · Architectures and efficiency › Quantization · approach: Post-training quantization of LLMs · 35 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Agile-Quant quantizes both weights (4-bit via GPTQ) and activations (8/4-bit) of LLaMA, OPT and BLOOM, adds activation-aware token pruning and a TRIP per-channel refinement, and implements INT4/INT8 kernels on edge CPUs for up to 2.55x on-device speedup over FP16.

Problem

Activation quantization of LLMs is hard because of outliers, and commodity edge CPUs/SIMD units do not efficiently execute sub-8-bit activation x weight matrix multiplications, so inference acceleration on edge devices is limited.

Approach

Combine GPTQ-style 4-bit weight quantization with 8/4-bit activation quantization guided by attention-based token pruning (using attentivity to the start token) and a TRIP channel-wise power-of-two refinement factor, plus a SIMD-based 4-bit multiplier implemented in ARM kernels for edge devices.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add token pruning to AgileQ (ablation)AgileQ with 8-bit activations · Wikitext-2perplexity: 6.16 -> 6.09Helped
Add token pruning to AgileQ (ablation)AgileQ with 8-bit activations · Wikitext-2perplexity: 15.90 -> 14.90Helped
Add token pruning to AgileQ (ablation)AgileQ with 8-bit activations · C4perplexity: 16.72 -> 15.95Helped
Add token pruning to AgileQ (ablation)AgileQ with 8-bit activations · Wikitext-2perplexity: 11.73 -> 11.36Helped
Add token pruning to AgileQ (ablation)AgileQ with 8-bit activations · C4perplexity: 14.70 -> 14.03Helped
Use token pruning at large token sparsity or short input sequence length (ablation)AgileQ · Wikitext-2perplexityHurt

Surprising findings

Limitations

Experimental setup

Datasets: WikiText-2 C4 PTB

Benchmarks: WikiText-2 C4 PTB

Metrics: perplexity latency (s) model size (GB)

Baselines: RTN GPTQ AWQ SpQR SqueezeLLM MoFQ8 ZeroQuant-FP ZeroQuant-V2

Builds on: GPTQ ZeroQuant-V2 GEMMLOWP ArmComputeLibrary

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy