

















Abstract:Large language models (LLMs) have demonstrated remarkable capabilities, but their massive scale poses significant challenges for practical deployment. Structured pruning offers a promising solution by removing entire dimensions or layers, yet existing methods face critical trade-offs: task-agnostic approaches cannot adapt to task-specific requirements, while task-aware methods require costly training to learn task adaptability. We propose DIET (Dimension-wise global pruning of LLMs via merging Task-wise importance scores), a training-free structured pruning method that combines dimension-level granularity with task-aware selection. DIET profiles activation magnitudes across tasks using only 100 samples per task, then applies majority voting to construct a single global mask. DIET does not require large costs from pre-computation or training. Experiments on seven zero-shot benchmarks using Gemma-2 2B and 9B models demonstrate the effectiveness of DIET; for example, at 20% sparsity on Gemma-2 2B, DIET achieves near 10% average accuracy improvement, compared to previous state-of-the-art structured pruning methods. This advantage persists across various sparsity levels and model scales, positioning DIET as a practical and robust choice for structured LLM pruning.
| Comments: | 14 pages, 10 figures. Code available at this https URL |
| Subjects: | Machine Learning (cs.LG) |
| Cite as: | arXiv:2603.23985 [cs.LG] |
| (or arXiv:2603.23985v3 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2603.23985 arXiv-issued DOI via DataCite |
From: Jimyung Hong [view email]
[v1]
Wed, 25 Mar 2026 06:28:58 UTC (1,735 KB)
[v2]
Fri, 1 May 2026 22:34:22 UTC (1,554 KB)
[v3]
Tue, 26 May 2026 08:39:09 UTC (472 KB)
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。