发现只有从零训练的模型才真正用上连续思维的叠加能力
The Illusion of Superposition? A Principled Analysis of Latent Thinking in Language Models
- 通过三种不同训练方式对比,检验连续思维中的叠加现象
- 仅从零训练的模型表现出叠加迹象,其他方式均失效
- 揭示预训练偏差与模型容量如何决定叠加能否成立
连续思维(Latent CoT)作为离散思维的替代方案,理论上可实现超叠加:在单一表征中并行维持多个候选解。我们通过三种范式研究这一现象:无训练的凸组合构造、微调基模型生成连续思维,以及完全从零训练模型以连续思维解题。利用Logit Lens和实体级探针分析内部表示,发现仅从零训练的模型显示出叠加迹象。在无训练和微调范式中,叠加要么崩溃,要么未被使用,模型反而找到捷径解法。这源于两个互补机制:一是在最后层预训练数据导致模型提前锁定一个词元;二是模型容量显著影响其偏好解。结果统一解释了连续思维中叠加出现与崩溃的条件。
原文摘要 · Abstract (English)
Latent reasoning via continuous chain-of-thoughts (Latent CoT) has emerged as a promising alternative to discrete CoT reasoning. Operating in continuous space increases expressivity and has been hypothesized to enable superposition: the ability to maintain multiple candidate solutions simultaneously within a single representation. Despite theoretical arguments, it remains unclear whether language models actually leverage superposition when reasoning using latent CoTs. We investigate this question across three regimes: a training-free regime that constructs latent thoughts as convex combinations of token embeddings, a fine-tuned regime where a base model is adapted to produce latent thoughts, and a from-scratch regime where a model is trained entirely with latent thoughts to solve a given task. Using Logit Lens and entity-level probing to analyze internal representations, we find that only models trained from scratch exhibit signs of using superposition. In the training-free and fine-tuned regimes, we find that the superposition either collapses or is not used at all, with models discovering shortcut solutions instead. We argue that this is due to two complementary phenomena: i) pretraining on natural language data biases models to commit to a token in the last layers ii) capacity has a huge effect on which solutions a model favors. Together, our results offer a unified explanation for when and why superposition arises in continuous chain-of-thought reasoning, and identify the conditions under which it collapses.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。