/ Paper arXiv 2611.04814
Curator's notes
LoCo learns when to stop reasoning locally, cutting chain length while keeping accuracy.
Curator's notesWritten by AI from the paperEvery number here appears in the abstractReport a mistake
What's new
It trains models on state‑conditioned preference pairs to decide continuation versus conclusion, unlike prior methods that only shorten whole trajectories.
Key result
LoCo reduces completion length by 41.7%–45.4% on math and 23.4%–36.3% on code while maintaining or improving accuracy.
Why it matters
Builders can deploy cheaper, faster reasoning models without sacrificing performance.
Abstract
From the paperThe authors' own words.
Large reasoning models achieve strong performance on challenging tasks, but their long chain-of-thought often incurs substantial inference costs by continuing to deliberate after sufficient information has been obtained. Existing reasoning compression methods primarily operate on reasoning trajectories, paying less attention to the local decision of whether further reasoning is necessary given the current state. We propose LoCo, a local preference learning method for reasoning compression. LoCo detects candidate overthinking states from the model's own traces, freezes the corresponding reasoning prefixes, regenerates shorter verified continuations from the same states, and trains on the resulting state-conditioned preference pairs with DPO. This formulation teaches the model when to continue reasoning and when to conclude, rather than simply encouraging shorter trajectories overall. We construct training data entirely from self-generated mathematical traces and evaluate LoCo on five mathematical and three code generation benchmarks. LoCo reduces completion length by 41.7\%--45.4\% on mathematics and 23.4\%--36.3\% on code, while maintaining or improving accuracy and achieving the best AUC across all code benchmarks. LoCo consistently improves the accuracy-efficiency trade-off on mathematics and transfers effectively to code generation despite math-only training, demonstrating the cross-domain generality of local preference learning for reasoning compression. The code and model will be publicly released at https://github.com/MuyaoYuan/LoCo.
From the full paper
From the paperCopied from the paper itself by our code.
Reading the paper on arXiv…
What you could build
Written by AIIntermediateRequires fine‑tuning with DPO and generating self‑traces
- Implement a math‑solver API that stops early using LoCo’s decision model
- Create a code‑completion tool that trims generation time with LoCo
- Build a benchmark suite comparing inference cost versus accuracy with and without LoCo
Try it
Code, models and demos that use this paper.
- Code:
- ★ 12 on GitHub ↗
- Models:
- MuyaoYuan/LoCo-7B ↗1 model on Hugging Face
- Demos:
- fastgpu/vram-and-gpu-cost-calculator ↗6 demos on Hugging Face
Ask this paper
/ Ask a questionAnswers come from the full paper and name the section
/ Quiz yourself
What does LoCo train the model to predict instead of merely encouraging shorter trajectories?
It trains the model to predict whether to continue reasoning or to conclude given the current reasoning state.
/ Your notes
Select text in the abstract to highlight it or add a note. They stay in this browser.