让大模型用连续值思维链并行推理,提升效率与表现
Continuous Chain of Thought Enables Parallel Exploration and Reasoning
- 用连续值令牌构建思维链,支持多路径并行搜索
- 嵌入维度决定最佳并行度,实验验证可提升推理效率
- 适合需要复杂逻辑推理或高效搜索的场景
现代语言模型通过从离散词表中自回归采样生成思维链。本文提出连续值思维链(CoT2),以更丰富的表达方式实现并行推理。理论上,我们证明了CoT2能同时追踪多个离散推理路径,并量化其可实现的并行程度及其对推理效率的增益;进一步设计了一种基于单层Transformer的结构,在足够高的嵌入维度下可解决组合“子集和问题”。核心在于一种新监督策略:将模型输出匹配目标路径的实证令牌分布。同时引入采样策略,使政策优化适用于CoT2。主要策略在每步解码时采样并组合K个离散令牌,以控制并行程度。实验表明:(i) 最优并行度由嵌入维度决定;(ii) 连续监督优于其他方法;(iii) 结合政策优化后,模型性能超越初始离散或连续监督。
原文摘要 · Abstract (English)
Modern language models generate chain-of-thought traces by autoregressively sampling tokens from a finite vocabulary. While this discrete sampling has achieved remarkable success, conducting chain-of-thought with continuously-valued tokens (CoT2) offers a richer and more expressive alternative. Our work provides new theoretical guarantees and algorithms for CoT2, motivated by logical reasoning tasks that inherently require search capabilities. Theoretically, we establish how CoT2 facilitates the model to track multiple discrete traces in parallel; and quantify the level of achievable parallelism and its benefits for inference efficiency. We also provide a CoT2-based one-layer transformer construction that solves the combinatorial "subset sum problem" given a sufficient embedding dimension. These insights arise from a novel and effective supervision strategy where we match the language model outputs to the empirical token distributions of a set of target traces. Complementing this, we introduce sampling strategies that unlock policy optimization methods for CoT2. Our primary strategy samples and composes $K$ discrete tokens at each decoding step to control the level of parallelism. Experiments confirm that (i) the optimal level of parallelism is governed by the embedding dimension, (ii) our continuous supervision strategy can outperform alternative methods, and (iii) policy optimization with CoT2 indeed improves the performance of the model beyond its initial discrete or continuous supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。