arXiv Computation and Language By Mahdi Salmani, Adel Javanmard, Devansh Gupta, Meisam Razaviyayn

Direct Pass@$k$ Optimization for Thought Pruning: Balancing Quality and Diversity in Test-Time Reasoning

Read the original on arXiv Computation and Language →

The paper introduces a pruning method for test-time reasoning that directly optimizes pass@$k$ by selecting a pool of partial solutions that maximizes the chance of at least one correct final answer. This approach balances quality and diversity automatically, avoiding separate design choices for each. Experiments on eight open-weight reasoning models show that the method outperforms existing quality-based and diversity-based pruning strategies, especially when a moderate number of partial solutions are retained.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computation and Language.

arXiv AI
1d ago

Improving the Chain-of-Thought Compressibility of Pruned Reasoning Models via Long--Short Contrastive Pruning

The paper introduces Contrastive Set-Difference (CSD), a pruning method that targets neurons involved in redundant reasoning by contrasting activations on long versus short correct solutions. CSD is combined with an extended preference-based chain-of-thought compression that also considers problems a pruned model always or never solves. Applied to a 7B model, the approach reduces thinking tokens by 8–22% on math benchmarks while maintaining accuracy within 0.5 points compared to traditional pruning and compression pipelines.

By Tao Jing, Chen Kang, Feifei Zhai, Pengyuan Liu, Dong Yu
arXiv AI
Aug 25

Revisiting the Effectiveness of LLM Pruning for Test-Time Scaling

The paper revisits the impact of pruning on large language models (LLMs) during test-time scaling (TTS). While prior work found that structured pruning degrades reasoning performance, this study shows that unstructured pruning—removing only specific redundant weights—can actually improve TTS performance on reasoning benchmarks for models s1.1-7B and Qwen3-8B, sometimes surpassing the full-weight models. The authors also examine how different layer-wise sparsity allocation strategies affect these outcomes.

By Ocean Monjur, Shahriar Kabir Nahin, Anshuman Chhabra
arXiv AI
Sep 17

OBC-Prune: Outcome-Based Calibration for Large Reasoning Model Pruning

OBC‑Prune introduces an outcome‑based calibration approach for pruning large reasoning models, focusing on the causal importance of each reasoning sentence rather than uniform activation salience. By pairing correct and incorrect rollouts and using intervention‑based analysis, it assigns per‑token weights that guide one‑shot pruning methods such as SparseGPT, Wanda, and ALPS. Experiments on DeepSeek‑R1‑Distill‑Qwen models show consistent accuracy gains and shorter reasoning traces across multiple benchmarks at 40‑50% sparsity.

By Ha Lan Nguyen, Huy Hoang Tran, Trac-Duy Tran, Dung D. Le
arXiv Machine Learning
Oct 8

Continuous-Utility Direct Preference Optimization

Continuous-Utility Direct Preference Optimization (CU‑DPO) replaces binary preference labels with continuous scores to better capture fine‑grained reasoning quality. The framework aligns language models to multiple prompt‑based cognitive strategies, proving a θ(K log K) sample‑complexity improvement and convergence to an entropy‑regularized utility‑maximizing policy. A two‑stage pipeline—strategy selection via best‑vs‑all comparisons and execution refinement using margin‑stratified pairs—yields significant gains on mathematical reasoning, code generation, and causal reasoning benchmarks.

By Muhammad Ahmed Mohsin, Muhammad Umer, Emily Fox