Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Relative Kinetic Utility for Reasoning-Aware Structural Pruning in Large Language Models

About

Chain-of-Thought (CoT) prompting symbolized a huge improvement of reasoning capabilities of Large Language Models (LLMs). However, scaling up test-time computation yields extensive CoT sequences, introducing severe inference latency and key-value (KV) cache memory bottlenecks. While structural pruning offers a fundamental, hardware-aware solution to alleviate static parameter burdens, existing magnitude-based methods may cut off the neurons of CoT: by over-indexing on discrete cross-entropy objectives, these heuristics fall into a \textit{magnitude trap}: they prioritize high-frequency, low-information syntactic tokens and trigger a disappointing reasoning collapse at high sparsities (e.g., 40\%). To overcome this topological phase transition, we propose \textsc{Relative Kinetic Utility} (RKU), a novel theoretical framework that elevates discrete pruning to a continuous kinetic integral over the depth manifold of the model based on Alternating Gradient Flow(AGF). By modifying it with Fisher trace normalization, RKU acts as a lightweight curvature-aware normalization to isolate \textit{kinetic spikes} -- the fundamental structural pathways responsible for high-curvature logical routing. Extensive experiments on Qwen-2.5-7B and LLaMA-3-8B improves performance in the high-sparsity regime around 40\%. RKU attains 13.34\% accuracy on GSM8K at 40\% sparsity, outperforming the strongest baseline, and appears to better preserve reasoning-relevant representations under out-of-distribution evaluation.

Tianhao Qian• 2026

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText2
Perplexity8.1496
3785
Language ModelingC4
Perplexity26.2717
1688
Language ModelingC4
Perplexity14.0493
1565
Commonsense ReasoningWinoGrande--
1442
Question AnsweringPIQA
Accuracy72.03
505
Question AnsweringARC Easy
Accuracy66.54
210
ReasoningGSM8K--
111
Mathematical ReasoningMathQA OOD (test)
Accuracy63
24
Mathematical ReasoningAQuA (test)
Accuracy28.35
18
Complex ReasoningAQUA
Accuracy28.35
12
Showing 10 of 14 rows

Other info

Follow for update