Lune

ICML2026Top-tier venue

What Makes Effective Supervision in Latent Chain-of-Thought: An Information-Theoretic Analysis

Xinghao Chen, Chak Tou Leong, Wenjin Guo, Jian Wang, Wenjie Li, Xiaoyu Shen

2026Year
1Citations

Abstract

Latent Chain-of-Thought (CoT) internalizes reasoning within continuous hidden states, offering a promising alternative to verbose discrete reasoning traces. However, robust latent reasoning remains difficult because outcome supervision provides weak learning signals and leaves latent trajectories prone to semantic drift. In this work, we analyze Latent CoT from an information-theoretic perspective and identify this failure as a dual collapse: gradient attenuation along the optimization path and representational drift in the latent space. We further decompose process supervision into two complementary dimensions: Trajectory Supervision, which injects dense stepwise reasoning signals, and Space Supervision, which preserves the semantic structure of the latent manifold. Our analysis shows that rigid geometric compression can collapse the reasoning space, whereas generative reconstruction provides a more flexible semantic anchor that better preserves information capacity. To measure these effects, we introduce the Unified Latent Probe (ULP), which quantifies the mutual information between latent trajectories and explicit reasoning steps. Experiments reveal a clear Information-Performance Binding: reasoning accuracy depends on the information fidelity preserved in the latent chain. These findings provide a principled framework for latent reasoning supervision and suggest shifting from geometric imitation toward mutual information maximization. Our code is available at this repository. … cm c1 c2 … c m c 1 c 2 Reconstructed Sk Target Sk CE Loss Figure 1. Overview of our information-theoretic framework for Latent CoT. Process supervision is decomposed into trajectory supervision for sequential information injection and space supervision for semantic anchoring, while ULP probes frozen latent states to quantify recoverable reasoning information.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext d51ab5cc-d4db-4d11-8195-d1a105e6e5ec

Builds on19

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines