Think Wider: Mitigating Latent Rank Collapse in Implicit Chain-of-Thought Reasoning

Yuwen Hao, Menglin Yang


Abstract
Chain-of-thought (CoT) reasoning improves the reasoning ability of large language models by introducing intermediate computation, but explicit rationales increase decoding length, latency, and context cost. Implicit CoT offers a more efficient alternative by moving intermediate reasoning into continuous latent states. However, latent reasoning can be unstable: successive latent states may become overly similar and collapse toward a shared dominant direction, reducing the diversity of the reasoning trajectory. In this work, we identify latent rank collapse and propose WIDER, a lightweight spectral regularizer for implicit CoT. During training, WIDER estimates the shared direction of each latent trajectory and penalizes projections onto this direction, encouraging latent states to span a broader representational subspace. The method is plug-and-play and leaves the backbone model, latent schedule, and inference-time decoding procedure unchanged. We further formulate this collapse as a geometric bottleneck in implicit reasoning, casting its mitigation as a training-time regularization problem rather than an inference-time decoding change. Extensive experiments show that WIDER improves matched implicit CoT baselines, while mechanistic analyses reveal higher effective rank, lower dominant-direction energy, and reduced redundancy among latent steps. These results highlight latent subspace utilization as an important factor for efficient continuous reasoning, providing a geometric perspective for analyzing and improving implicit CoT. Code is available at https://github.com/whitesweater/WIDER.
Anthology ID:
2026.findings-emnlp.1773
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
37936–37951
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1773/
DOI:
10.18653/2026.findings-emnlp.1773
Bibkey:
Cite (ACL):
Yuwen Hao and Menglin Yang. 2026. Think Wider: Mitigating Latent Rank Collapse in Implicit Chain-of-Thought Reasoning. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 37936–37951, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Think Wider: Mitigating Latent Rank Collapse in Implicit Chain-of-Thought Reasoning (Hao & Yang, Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1773.pdf