atlas

Research map / Large language models / Language, safety and interpretability

Fluctuation-Based Adaptive Structured Pruning for Large Language Models

AAAI 2024 · Large language models › Language, safety and interpretability · approach: Pruning and circuits in LLMs · 43 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 FLAP, a retraining-free structured pruning framework for LLaMA models that scores input channels by a fluctuation metric, globally standardizes scores for adaptive structure search, and adds baseline-mean bias compensation.

Problem

Retraining-free pruning of LLMs has focused on unstructured sparsity, which needs special hardware; existing structured methods such as LLM-Pruner rely on LoRA fine-tuning, and metrics from unstructured pruning transfer poorly to rows/columns.

Approach

Score each input channel by the sample variance of its input features weighted by the squared norm of the corresponding weight column (fluctuation metric), standardize scores per layer and module for a global adaptive threshold search, and compensate removed channels with a bias equal to the weight times the baseline (mean) activation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use FLAP instead of Wanda-sp at 50% pruningLLaMA-7B · WikiText2 validationperplexity: 366.43 -> 31.80Helped
Use FLAP instead of LLM-Pruner with LoRALLaMA-7B at 50% pruning · WikiText2 validationperplexity: 38.12 -> 31.80Helped
Use FLAP instead of LLM-Pruner with LoRALLaMA-7B at 50% pruning · Zero-shot averageaccuracy: 48.98 -> 50.37Helped
Use FLAP instead of Wanda-spLLaMA-7B at 20% pruning · Zero-shot averageaccuracy: 63.35 -> 62.08Hurt
Use FLAP instead of LLM-Pruner with LoRALLaMA-13B at 20% pruning · Zero-shot averageaccuracy: 64.02 -> 64.52Helped
Use adaptive layers and modules (AL-AM) instead of uniform layers and modules (UL-UM) (ablation)WIFV structured pruning · WikiText2 validation, LLaMA-7B at 50% pruningperplexity: 57.57 -> 31.80Helped
Use WIFV instead of WIFN with AL-AM (ablation)FLAP structured pruning · WikiText2 validation, LLaMA-7B at 50% pruningperplexity: 34.09 -> 31.80Helped
Use WIFV instead of IFV with AL-AM (ablation)FLAP structured pruning · WikiText2 validation, LLaMA-7B at 50% pruningperplexity: 33.33 -> 31.80Helped
Add baseline bias compensation (ablation)FLAP · WikiText2 validation, LLaMA-7B at varying pruning ratiosperplexityHelped
Increase calibration sample count (ablation)FLAP on LLaMA-7B at 50% pruning · WikiText2 validationperplexityHelped
Use WikiText2 rather than C4 calibration data (ablation)FLAP · Zero-shot averageaccuracy: 50.37 -> 48.33Hurt
Use C4 rather than WikiText2 calibration data (ablation)FLAP · PIQA accuracyaccuracy: 63.82 -> 67.52Helped

Surprising findings

Limitations

Experimental setup

Datasets: WikiText2 C4

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

Metrics: perplexity zero-shot accuracy parameter count memory tokens/s

Baselines: Wanda-sp LLM-Pruner LLM-Pruner (w/ LoRA)

Builds on: Wanda SparseGPT LLM-Pruner Optimal Brain Damage

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