Some AI systems write out long chains of thought, meaning step-by-step reasoning, even after they have enough information to answer. This wastes time and computing power, and extra steps do not always improve answers.
LoCo finds points in a model’s own worked solutions where it may already have enough information. From each point, it compares a shorter, checked answer with a longer continuation, then trains the model to choose when to stop and when to keep reasoning.
The method shortened generated text by roughly two-fifths on math tasks and about a third on code tasks, while maintaining or improving accuracy. Code improvements came even though training used only math examples, suggesting the learned stopping behavior can transfer across tasks.
