Papers
arxiv:2610.19121

Unlocking Structured Pruning for Reasoning Language Models via Attention-Preserving MLP Trimming

Published on Oct 13
· Submitted by
Vincent-Daniel Yun
on Oct 15
Authors:
,
,

Abstract

LayerTrim is a training-free structured pruning approach that removes redundant decoder layers and trims MLP features to preserve reasoning ability, reducing inference costs and generation length.

Structured pruning reduces LLM inference costs by removing model components to produce smaller dense models. However, existing methods can cause reasoning performance to collapse on long-form reasoning tasks such as AIME. We propose LayerTrim, a training-free structured pruning method that identifies redundant decoder layers and selectively trims their MLP features while preserving attention and decoder depth. LayerTrim is the first training-free structured pruning method designed to preserve long-form reasoning without gradients or recovery training. Across three reasoning language models and five benchmarks, LayerTrim outperforms existing structured pruning methods in most evaluated settings, improving average accuracy by up to 21.4 percentage points over RESP. On Qwen3-14B, LayerTrim achieves 1.24times faster prefill and 19.7\% lower peak GPU memory than the dense model, while better preserving reasoning ability. It also achieves the shortest average generation length among evaluated pruned models, while retaining the highest average accuracy. These results demonstrate that LayerTrim improves reasoning efficiency through both structural compression and reduced token generation.

Community

Paper author Paper submitter

We demonstrate that attention-preserving structured pruning can retain long-form reasoning capabilities than conventional approaches that remove attention modules or entire decoder layers. Building on this insight, we introduce LayerTrim, a training-free method that preserves attention and model depth while selectively trimming redundant MLP features using reasoning-trace calibration. Across three reasoning LLMs, LayerTrim outperforms existing structured pruning methods by up to 21.4 percentage points in average accuracy while reducing inference costs, without gradients or recovery training.

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2610.19121
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2610.19121 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2610.19121 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2610.19121 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.