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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText2 | Perplexity8.1496 | 3785 | |
| Language Modeling | C4 | Perplexity26.2717 | 1688 | |
| Language Modeling | C4 | Perplexity14.0493 | 1565 | |
| Commonsense Reasoning | WinoGrande | -- | 1442 | |
| Question Answering | PIQA | Accuracy72.03 | 505 | |
| Question Answering | ARC Easy | Accuracy66.54 | 210 | |
| Reasoning | GSM8K | -- | 111 | |
| Mathematical Reasoning | MathQA OOD (test) | Accuracy63 | 24 | |
| Mathematical Reasoning | AQuA (test) | Accuracy28.35 | 18 | |
| Complex Reasoning | AQUA | Accuracy28.35 | 12 |