QLPO: Quadrant-weighted Sampling for Length-aware Policy Optimization

Explainable & Ethical AI
Published: arXiv: 2607.21793v1
Authors

Siwei Chen Siqi Chen Xupeng Miao Bin Cui

Abstract

Recent large reasoning models often develop long chain-of-thought responses during reinforcement learning (RL), resulting in high inference latency and deployment cost. Existing methods for response length control typically rely on explicit length penalties or additional control modules, which require careful tuning and may compromise reasoning quality. We propose Quadrant-weighted Sampling for Length-aware Policy Optimization (QLPO), a simple resampling-based variant of GRPO that introduces implicit length control without modifying the reward function. QLPO first over-generates candidate responses and then resamples the training group by preserving the empirical correct/incorrect ratio while favoring short correct responses and long incorrect responses. This reshapes the training distribution and implicitly encourages shorter model outputs. Across models ranging from 1.5B to 32B parameters, including both base models and strong reasoning models, QLPO consistently improves the accuracy-length trade-off. It reduces response length by 30% to 70% while preserving reasoning performance. These results suggest that structured resampling provides an effective and robust approach to efficient reasoning.

Paper Summary

Problem
Large reasoning models, like those used in AI systems, often develop long and complex reasoning paths during training. This "length explosion" phenomenon leads to high inference costs, deployment latency, and a critical bottleneck for real-world applications. To address this issue, researchers have been exploring ways to control the length of model responses without compromising reasoning quality.
Key Innovation
The research proposes a new approach called Quadrant-weighted sampling for Length-aware Policy Optimization (QLPO). QLPO is a simple and robust reinforcement learning algorithm that introduces implicit length control without modifying the reward function. It works by over-generating candidate responses and then resampling the training group to favor short correct responses and long incorrect responses.
Practical Impact
QLPO has the potential to significantly improve the accuracy-efficiency trade-off of reasoning models. By reducing response length by 30% to 70% while preserving reasoning performance, QLPO can help mitigate the length explosion phenomenon and make large reasoning models more practical for real-world applications. This can enable the development of more efficient and effective AI systems, such as chatbots, virtual assistants, and decision-making tools.
Analogy / Intuitive Explanation
Imagine you're trying to solve a complex math problem, and your AI model is generating multiple steps to reach the solution. QLPO is like a coach that looks at the steps and says, "Hey, let's focus on the shortest path to the solution, but also make sure we're not missing any important details." By favoring short correct responses and long incorrect responses, QLPO helps the model find the most efficient and accurate solution.
Paper Information
Categories:
cs.AI
Published Date:

arXiv ID:

2607.21793v1

Quick Actions