atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

LoRAP: Transformer Sub-Layers Deserve Differentiated Structured Compression for Large Language Models

ICML 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Layer-wise pruning of LLMs · 0 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.

Proposes LoRAP, a mixed structured compression method for LLMs that applies activation-weighted SVD with per-matrix rank allocation to attention (MHA) sub-layers and gradient-free group channel pruning (keeping the least important 1%) to FFN sub-layers, followed by LoRA recovery.

Problem

LLMs require massive memory and compute; existing structured compression methods treat all Transformer sub-layers identically and often rely on gradient-based importance estimation, which is expensive.

Approach

MHA sub-layers show stronger low-rank structure than FFN sub-layers, so compress MHA with activation-weighted SVD (AWSVD) and allocate parameters across Wq/Wk vs Wv/Wo by their low-rank degree, while compressing FFN with gradient-free channel pruning that retains the least important 1% of weights.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use LoRAP instead of LLM-Pruner at 20% compression without tuningLLaMA-7B · 20% compression, zero-shot common-sense reasoningAverage accuracy: 56.69 -> 60.53Helped
use LoRAP instead of LLM-Pruner at 50% compression without tuningLLaMA-7B · 50% compression, zero-shot common-sense reasoningAverage accuracy: 42.13 -> 47.25Helped
retain the least-important 1% of FFN weights instead of retaining none (ablation)LoRAP · 20% compressionWikiText2 perplexity: 16.91 -> 15.69Helped
retain the least-important 1% of FFN weights instead of retaining none (ablation)LoRAP · 50% compressionAverage accuracy: 46.19 -> 47.25Helped
retain 2% rather than the least-important 1% of FFN weights (ablation)LoRAP · 50% compressionAverage accuracy: 47.25 -> 46.78Hurt
allocate MHA parameters at ratio 1:3 rather than 1:1 for (Wq+Wk):(Wv+Wo) (ablation)LoRAP · LLaMA-7B, 50% compressionAverage accuracy: 40.70 -> 47.25Helped
use L2 rather than L-infinity aggregation for channel importance (ablation)LoRAP · LLaMA-7B, 50% compressionAverage accuracy: 44.09 -> 47.25Helped
increase the quantity of calibration data (ablation)LoRAP · WikiText2 and PTBperplexityHelped
increase calibration sample length (ablation)LoRAP · WikiText2 and PTBperplexityMixed
apply AWSVD rather than AFM to the MHA sub-layer (ablation)LLaMA-7B at 50% compression · MHA sub-layer, WikiText2perplexity: 33.90 -> 19.49Helped
use channel pruning rather than AWSVD on the FFN sub-layer (ablation)LLaMA-7B at 50% compression · FFN sub-layer, WikiText2perplexity: 225.08 -> 32.64Helped

Surprising findings

Limitations

Experimental setup

Datasets: BookCorpus Alpaca (cleaned) C4 WikiText2 PTB

Benchmarks: WikiText2 PTB BoolQ PIQA HellaSwag WinoGrande ARC-easy ARC-challenge OpenbookQA

Metrics: perplexity accuracy MACs latency

Baselines: LLM-Pruner LoRAPrune LoRAShear AFM SVD Head Prune

Builds on: Wanda LLM-Pruner LoSparse FWSVD

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