Get the app

Beyond Tokens: Why LLMs Are Trading Words for Continuous Latent Thoughts

By replacing discrete word tokens with raw hidden-state embeddings, latent reasoning architectures unlock breadth-first planning while slashing inference overhead.

Language was invented for human communication, not computational reasoning. Yet for the last three years, the dominant paradigm for complex problem-solving in large language models has been Chain-of-Thought (CoT)—forcing models to verbalize internal calculations into discrete dictionary tokens. This approach introduces an unavoidable bottleneck: models allocate identical computational budgets to high-entropy syntax ("Let us consider step 2") as they do to critical mathematical or logical leaps, while greedily committing to single-path trajectories that make backtracking nearly impossible.

A foundational paradigm shift called Coconut (Chain of Continuous Thought), developed by researchers at Meta FAIR and UC San Diego, circumvents this token tax entirely. Instead of projecting internal representations onto a discrete vocabulary at every reasoning step, the architecture feeds its raw hidden states directly back into the transformer input stream. The result is an unconstrained continuous latent reasoning loop that fundamentally transforms how models explore search spaces.

Standard CoT:   Input -> [h_1] -> Token_1 -> [h_2] -> Token_2 -> Output
Coconut Latent: Input -> [h_1] ----------> [h_2] ----------> Output
                         (Continuous Latent Space)

The Architecture: Bypassing the Unembedding Bottleneck

In standard autoregressive language modeling, predicting token $x_{t+1}$ given preceding context $x_{\leq t}$ requires projecting the final hidden layer representation $h_t$ through an unembedding matrix $W$ into vocabulary space:

$$\mathcal{M}{\text{CoT}}(x{t+1} \mid x_{\leq t}) = \text{softmax}(W h_t)$$

This projection is an information bottleneck. By forcing high-dimensional vector representations into a discrete set of ~32,000 to 128,000 tokens, the model collapses rich probabilistic distributions into a single token selection.

Coconut introduces a dual-mode mechanism:

  • Language Mode: Operates as a conventional autoregressive LLM for user prompts and final response generation.
  • Latent Mode: Activated between special boundary tokens <bot> (begin-of-thought) and <eot> (end-of-thought). In this mode, the model disables the softmax projection over vocabulary words. Instead, the final hidden state vector $h_t \in \mathbb{R}^d$ replaces the token embedding $e(x_{t+1})$ directly for the subsequent forward pass.

Under Latent Mode, the transformer input embedding sequence becomes:

$$E_t = [e(x_1), \dots, e(x_i), h_i, h_{i+1}, \dots, h_{t-1}]$$

Because the internal reasoning steps are never discretized into tokens, the entire intermediate "thinking" process remains continuous and fully differentiable. Backpropagation can flow directly end-to-end across multiple reasoning cycles during training.


Multi-Stage Curriculum: Learning to Think in Vectors

Training a model to reason in continuous latent space presents a severe optimization challenge: without ground-truth labels for high-dimensional hidden states, how does the network learn coherent reasoning paths?

The solution is a multi-stage curriculum that gradually "internalizes" explicit text thoughts into latent states:

  1. Stage 0 (Cold Start): The model is fine-tuned on standard textual Chain-of-Thought demonstrations using negative log-likelihood loss.
  2. Stage $k$ (Gradual Latent Substitution): The first $k$ verbal reasoning steps are stripped out and replaced with $L = k \times c$ continuous thoughts (where $c$ is a hyperparameter determining the continuous-to-discrete compression ratio).
  3. Final Stage: The model receives only the initial prompt and must generate continuous thought vectors directly leading to the final target answer.

Crucially, loss is computed only on the final answer and downstream text tokens, with loss masked over both the question and the latent vectors. This prevents the model from simply trying to autoencode text into vectors; instead, it forces the latent representations to optimize strictly for downstream problem-solving utility.


Superposition and Emergent Breadth-First Search

The most striking discovery of continuous latent reasoning lies in how models handle branching logic. In standard Chain-of-Thought, when an LLM reaches a decision fork, it must sample a single discrete token, committing itself down a single path. If that branch hits a dead end, recovery requires explicit backtracking prompts or external Monte Carlo Tree Search (MCTS) wrappers.

By probing Coconut's latent thought vectors with linear diagnostic heads, researchers discovered that the model naturally maintains vocabulary-space superposition:

  • Parallel Hypothesis Exploration: Rather than selecting Branch A or Branch B, early continuous thought vectors maintain high activation probabilities for multiple contradictory intermediate variables simultaneously.
  • Emergent BFS Tree Search: As the latent thinking chain progresses through consecutive hidden states, the model progressively sharpens its internal probabilities, pruning invalid branches without having to write them out.
  • Drastic Reduction in Hallucinated Chains: On planning-heavy graph navigation tasks (such as ProsQA), discrete CoT frequently hallucinates invalid edges because it prematurely locks into wrong paths. Coconut avoids premature commitment, dynamically exploring multiple candidate paths in latent space before producing the final token stream.

Benchmark Performance: Smarter Reasoning with 70% Fewer Tokens

When evaluated against standard Chain-of-Thought, No-CoT, and pause-token baselines, continuous latent reasoning demonstrates massive efficiency and accuracy gains across rigorous planning benchmarks:

  • ProsQA (Logical Planning): Coconut achieved 97.0% accuracy compared to 77.5% for standard CoT, while significantly reducing token generation and sequence length during inference.
  • ProntoQA: Outperformed standard Chain-of-Thought reasoning with lower latency.
  • Ablation vs. Pause Tokens: Simply inserting empty <pause> tokens between prompt and answer without latent feedback failed completely, proving that computational capacity alone is insufficient—the dynamic feedback loop of $h_t \to h_{t+1}$ is what enables actual multi-step reasoning.

The Engineering Trade-offs

While continuous latent reasoning represents a dramatic leap in computational efficiency, it redefines the frontier of AI interpretability and serving infrastructure:

  • The Loss of Human-Readable Scratchpads: When an LLM reasons in English, developers can inspect every line of logic to identify why an agent made a bad tool call or incorrect deduction. Latent thoughts are opaque high-dimensional manifolds requiring dedicated probing classifiers to decode.
  • KV Cache Dynamics: Continuous thoughts still occupy slots in the key-value (KV) cache. However, because $c$ latent steps replace dozens of word tokens, memory pressure during long-horizon reasoning tasks drops significantly.
  • Test-Time Compute Allocation: Rather than spending compute by generating thousands of verbose output tokens (as seen in models like o1 or Claude 3.7 Sonnet), architectures can dynamically scale depth by looping through latent layers, separating test-time thinking compute from output bandwidth.

The Road Ahead

The shift toward continuous latent spaces signals the beginning of the post-token era for internal model cognition. As frontier labs look for ways to scale test-time compute without ballooning token generation costs and memory footprints, letting neural networks "think in vectors" before "speaking in tokens" may prove to be the architectural bridge between next-token prediction and true autonomous reasoning systems.

Sources

The daily AI brief, on your phone.

Feed, daily deep-dive and bytes — readable offline, with push alerts for the topics you follow.

Get it on Google Play