Search papers, labs, and topics across Lattice.
This paper introduces Typed Latent Reasoning (Tyler), a novel framework that enhances the reasoning capabilities of large language models (LLMs) by dynamically deciding when to switch between emitting text tokens and performing latent computations during autoregressive decoding. By learning a policy that allocates computational resources based on the reasoning context, Tyler significantly reduces redundancy and inference overhead associated with traditional chain-of-thought prompting. Experimental results demonstrate that Tyler achieves up to 14.49 points improvement in accuracy over standard CoT methods and outperforms existing baselines across various reasoning tasks while minimizing forgetting.
Tyler's adaptive reasoning framework boosts LLM accuracy by up to 14.49 points, redefining how models manage computational resources in real-time.
Chain-of-thought (CoT) prompting improves reasoning in large language models (LLMs) by externalizing intermediate computation as discrete text tokens, but this textual interface also introduces redundancy and inference overhead. Latent reasoning offers a promising alternative by carrying part of the computation in continuous representations. However, existing methods typically predefine when latent computation is invoked and how it is allocated during decoding, leaving a key problem unresolved: when to invoke latent computation, what type of computation to perform, and how much budget to allocate. We propose \textbf{Ty}ped \textbf{L}at\textbf{e}nt \textbf{R}easoning (Tyler), a typed and budget-aware framework for latent reasoning during autoregressive decoding. Tyler learns a policy that, at each decoding step, chooses between emitting a text token and switching to a latent computation module specialized for a particular reasoning function. Once invoked, an operator maps the current reasoning state into latent tokens that support global planning, local state updates, or reusable procedural abstraction. Across extensive experiments on three backbone LLMs, Tyler improves accuracy by up to 14.49 points over CoT and by up to 4.30 points over the strongest competing baseline. It further generalizes across diverse reasoning domains and achieves the best final-stage performance with the lowest forgetting.
