Exploring the Limits of Pruning: Task-Specific Neurons, Model Collapse, and Recovery in Task-Specific Large Language Models
Not on the map. Why?
Abstract (arXiv)
Neuron pruning is widely used to reduce the computational cost and parameter footprint of large language models, yet it remains unclear whether neurons in task-specific models contribute uniformly to task performance. In this work, we provide empirical evidence for the existence and importance of task-specific neurons through a systematic pruning study on language models specialized for mathematical reasoning and code generation. We introduce an activation-based selectivity metric to identify neurons with low contribution to the target task and prune them while preserving target-task accuracy, and compare selective pruning with random pruning. Selective pruning consistently outperforms random pruning, indicating that activation-based selectivity provides a systematic advantage over random pruning. Reverse pruning experiments further show that removing a small subset of highly task-specific neurons (~10%) causes complete performance collapse, suggesting that there exist task specific neurons and critical task information is concentrated in a small portion of the network. In contrast, selective pruning of less critical neurons (~30% - ~35%) reduces accuracy but still preserves significant performance. We also observed consistent reductions in parameters and runtime VRAM usage, along with improved inference throughput as pruning increases. Experiments on both 1.5B and 7B models reveal a robustness threshold around 15-20% pruning, beyond which accuracy loss and generation failures increase sharply. Fine-tuning substantially recovers performance across pruning levels, particularly for aggressively pruned models. These findings provide empirical evidence of neuron specialization in task-specific language models and offer insights into pruning robustness, model redundancy, and post-pruning recoverability.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 22 references that are in the map. First posted on arXiv 29 Apr 2026.
Builds on
17 of 22 references.
References · 17
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,049
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
- 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 643
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2024SliceGPT: Compress Large Language Models by Deleting Rows and Columnscited by 378
5 more
- 2024Dissecting Language Models: Machine Unlearning via Selective Pruningcited by 38
- 2024Sharing Matters: Analysing Neurons Across Languages and Tasks in LLMscited by 33
- 2023One Wide Feedforward is All You Needcited by 15
- 2025From Macro to Micro: Probing Dataset Diversity in Language Model Fine-Tuningcited by 3
- 2025Scaling Laws are Redundancy Lawscited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Improved Methods for Model Pruning and Knowledge Distillation
- 2026Batch-wise Adaptive Pruning: Periodic Neuron Activation-Aware Weight Pruning for Language Reasoning Model
- 2026On the Limits of Layer Pruning for Generative Reasoning in Large Language Models
- 2024Mitigating Copy Bias in In-Context Learning through Neuron Pruning
- 2026Gradually Compacting Large Language Models for Reasoning Like a Boiling Frog
- 2025From Local to Global: Revisiting Structured Pruning Paradigms for Large Language Models
- 2026SHarP: Saliency-based Pruning of Agent Harnesses
- 2026Neural Activation Patterns Across Language Model Architectures: A Comprehensive Analysis of Cognitive Task Performance