atlas

Research map / Architectures and efficiency / Quantization

ShiftAddLLM: Accelerating Pretrained LLMs via Post-Training Multiplication-Less Reparameterization

NeurIPS 2024 · Architectures and efficiency › Quantization · approach: Post-training quantization of LLMs · 6 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.

Post-training conversion of pretrained LLMs into multiplication-free models by quantizing weights into binary-coding (BCQ) matrices with power-of-two scaling factors, plus multi-objective error optimization and automated mixed bit allocation.

Problem

Deploying LLMs on resource-constrained hardware is limited by parameter size and dense multiplications; existing shift-and-add reparameterizations need training from scratch or full fine-tuning, which is too costly for LLMs.

Approach

Reparameterize each pretrained weight matrix as a sum of binary matrices with group-wise scaling factors (BCQ), quantize scaling factors to powers of two so multiplications become shifts, and replace remaining multiplications with LUT queries and adds; optimize a combined weight-plus-activation error objective and allocate bits per layer via an integer program driven by a sensitivity criterion.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply 3-bit ShiftAddLLM instead of OPTQOPT-66B · WikiText-2perplexity: 14.16 -> 9.43Helped
Apply 3-bit ShiftAddLLM instead of AWQLLaMA-2-70B · WikiText-2perplexity: 3.74 -> 3.64Helped
Apply ShiftAddLLM instead of LUT-GEMMOPT-66B · zero-shot downstream tasksmean accuracy: 50.70 -> 63.89Helped
Use automated mixed bit allocation (2.2-bit average) instead of fixed 2-bit allocation (ablation)Ours (Lat.) · WikiText-2, OPT-13Bperplexity: 14.76 -> 13.79Helped
Add multi-objective optimization (ablation)vanilla ShiftAddLLM · WikiText-2, OPT-13B, 2-bitperplexity: 1.5e4 -> 14.76Helped
Add automated mixed bit allocation (ablation)ShiftAddLLM with multi-objective optimization · WikiText-2, OPT-13B, 2-bitperplexity: 14.76 -> 13.79Helped
Use multi-objective column-wise scaling instead of a weight-only objective (ablation)ShiftAddLLM · WikiText-2, OPT-30Bperplexity: 222.6 -> 9.6Helped
Use multi-objective column-wise scaling instead of an activation-only objective (ablation)ShiftAddLLM · WikiText-2, OPT-30Bperplexity: 10.5 -> 9.6Helped
Use multi-objective column-wise scaling instead of the vanilla weight-plus-activation objective (ablation)ShiftAddLLM · WikiText-2, OPT-30Bperplexity: 16.3 -> 9.6Helped

Surprising findings

Limitations

Experimental setup

Datasets: WikiText-2

Benchmarks: WikiText-2 ARC BoolQ COPA PIQA RTE StoryCloze MMLU

Metrics: perplexity zero-shot accuracy latency energy GPU memory

Baselines: OPTQ LUT-GEMM QuIP AWQ FlexRound OmniQuant MSFP

Builds on: binary-coding quantization OPTQ DenseShift Optimal Brain Compression AWQ

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