The Potential of CoT for Reasoning: A Closer Look at Trace Dynamics
Gregor Bachmann, Yichen Jiang, Seyed Mohsen Moosavi Dezfooli, Moin Nabi
TL;DR
The paper investigates chain-of-thought (CoT) reasoning by introducing the potential, a metric that quantifies how conditioning on partial CoT tokens affects the probability of a correct answer $y^{*}$, i.e., $\operatorname{pot}(\bm{c}_{<t};\bm{x})$. Using sampling with $N=128$ trajectories, it analyzes CoT traces from competition-level mathematics (notably AIME-2024) to reveal that potential curves are often non-monotonic and feature sharp reasoning insights, tangents, and occasional late guessing; it also defines an estimator and proves a monotonicity property in expectation for correct runs. The study then demonstrates that CoT can be optimized to produce more monotonic progress and that a weaker model can be substantially aided by partial CoT from a stronger model, with as little as $20\%$ partial CoT delivering meaningful improvements across model families. Together, these findings advance understanding of CoT dynamics, offer a principled way to assess and improve reasoning traces, and suggest practical implications for transfer learning and reinforcement learning reward shaping in large language models.$
Abstract
Chain-of-thought (CoT) prompting is a de-facto standard technique to elicit reasoning-like responses from large language models (LLMs), allowing them to spell out individual steps before giving a final answer. While the resemblance to human-like reasoning is undeniable, the driving forces underpinning the success of CoT reasoning still remain largely unclear. In this work, we perform an in-depth analysis of CoT traces originating from competition-level mathematics questions, with the aim of better understanding how, and which parts of CoT actually contribute to the final answer. To this end, we introduce the notion of a potential, quantifying how much a given part of CoT increases the likelihood of a correct completion. Upon examination of reasoning traces through the lens of the potential, we identify surprising patterns including (1) its often strong non-monotonicity (due to reasoning tangents), (2) very sharp but sometimes tough to interpret spikes (reasoning insights and jumps) as well as (3) at times lucky guesses, where the model arrives at the correct answer without providing any relevant justifications before. While some of the behaviours of the potential are readily interpretable and align with human intuition (such as insights and tangents), others remain difficult to understand from a human perspective. To further quantify the reliance of LLMs on reasoning insights, we investigate the notion of CoT transferability, where we measure the potential of a weaker model under the partial CoT from another, stronger model. Indeed aligning with our previous results, we find that as little as 20% of partial CoT can ``unlock'' the performance of the weaker model on problems that were previously unsolvable for it, highlighting that a large part of the mechanics underpinning CoT are transferable.
