Capability-Guided Compression: Toward Interpretability-Aware Budget Allocation for Large Language Models
On the map: Mechanistic interpretability → SAE features and their applications
Abstract (arXiv)
Large language model compression has made substantial progress through pruning, quantization, and low-rank decomposition, yet a fundamental limitation persists across all existing methods: compression budgets are allocated without any representation of what individual model components functionally encode. We term this the capability-blind compression problem and argue it is a root cause of two well-documented failures -- the insensitivity of perplexity-based evaluation to reasoning capability loss, and the abrupt phase transitions in model performance recently characterized by Ma et al. (2026). We propose Capability-Guided Compression (CGC), a framework that addresses this by using Sparse Autoencoder (SAE)-derived capability density maps to allocate differential compression budgets across transformer components. Capability density is a formally defined scalar measure combining the feature breadth, activation entropy, and cross-input consistency of a component's SAE feature activation distribution. We prove theoretically that components with higher capability density exhibit lower structural redundancy and reach their individual phase transition points at lower compression ratios, providing the first pre-compression mechanism for component-level phase transition prediction. Experiments on GPT-2 Medium confirm that capability density is statistically independent of Wanda importance scores (Spearman rho = -0.054, n = 384 heads), establishing it as a genuinely novel compression signal orthogonal to all existing importance metrics. We report a negative result on PPL-based compression comparison and provide a principled diagnosis identifying GPT-2 Medium as an insufficient test bed for the full CGC hypothesis. The theoretical framework, density formalism, and orthogonality finding constitute a foundation for capability-aware compression research.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in SAE features and their applications, one of 13 directions in Mechanistic interpretability. Builds on 16 of its 22 references that are in the map. First posted on arXiv 17 Mar 2026.
Builds on
16 of 22 references.
Closest ancestors · 1
Same direction, newest first
Other references · 15
- 2017Attention Is All You Needcited by 60,361
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,386
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,121
- 2022In-context Learning and Induction Headscited by 964
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 800
- 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 650
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2023A Survey on Model Compression for Large Language Modelscited by 269
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Compressed Sensing for Capability Localization in Large Language Models
- 2025Compression is Routing: Reconstruction Error as an Intrinsic Signal for Modular Language Models
- 2023The Cost of Compression: Investigating the Impact of Compression on Parametric Knowledge in Language Models
- 2026UniComp: A Unified Evaluation of Large Language Model Compression via Pruning, Quantization, and Distillation
- 2025TOGGLE: Temporal Logic-Guided Large Language Model Compression for Edge
- 2026From Layers to Submodules: Rethinking Granularity in Replacement-Based LLM Compression
- 2026Data Distribution Matters: A Data-Centric Perspective on Context Compression for Large Language Model
- 2024Model Compression and Efficient Inference for Large Language Models: A Survey