← back to courses

VQ-CoT: Discretising Latent Chain-of-Thought

Understanding what latent thoughts actually learn.

Maria Pilligua, Chengheng Li Chen, Nil Biescas, Pau Amargant

EPFL CS-552 Modern NLP · Spring 2026 · Final grade 6/6

Chain-of-Thought vs Chain of Continuous Thought (Coconut)
Classic chain-of-thought (left) vs latent chain-of-thought / Coconut (right).

Instead of spelling out every reasoning step as text tokens (left), latent chain-of-thought models like Coconut and CODI feed the LLM's own last hidden state back in as the next input embedding (right). The intermediate reasoning stays in a continuous space and is never emitted as text. The claim is that this gives richer, more efficient reasoning per step. Because the latents are opaque, that claim is rarely tested. We looked into it.

VQ-CoT: latent chain-of-thought quantised via a codebook
Our probe: a VQ-VAE codebook inserted between the LLM's latent thoughts.

We push each latent thought through a VQ-VAE codebook before feeding it back to the LLM. If the model still answers correctly, the codebook entries tell us what content the latents actually carry. In parallel we probe the latents directly to see what they encode. The story is consistent: many of the latents are structured but content-free, and swapping or removing them barely changes the model's answer. The latent thoughts turn out to be a much thinner channel than the framing suggests.

The cause is the training objective. Both Coconut and CODI are supervised to reconstruct the original chain-of-thought text at the output, so the model has no choice but to route the answer through latents that preserve the surface form of natural-language reasoning, dataset delimiters included. The structure the latents carry is not a more efficient encoding of the problem; it is a compressed version of the written answer, artefacts and all. Latent chain-of-thought, as currently trained, is not learning to reason more efficiently, it is learning to compress natural language.