arXiv:2607.18264cs.AIcs.CL2026-07被引 1

用连续向量压缩推理步骤,提升大模型思考效率

MUX: Continuous Reasoning via Multiplexed Tokens

论文配图:MUX: Continuous Reasoning via Multiplexed Tokens
图 1 · 摘自论文原文
  • 将离散推理词拆解为可叠加的连续向量,实现高带宽压缩
  • 在32个任务中超越主流潜空间推理方法,平均提升12.7%
  • 适合需要高效推理与可解释性的复杂任务研究者

语言模型通过自然语言表达中间推理步骤来解决复杂问题。然而,这一过程计算开销大:每个推理步骤仅传递一个子词,大量时间用于表达而非计算。我们提出MUX,一种基于蒸馏的连续多路推理方法,将离散推理内容映射到潜在空间中的连续多路令牌。每个潜在令牌被训练为一段离散推理子词的加权线性叠加(多路复用),该叠加在构造上无损,且原段落可完全恢复(解复用)。我们证明,简单的依赖位置的权重分配(如几何衰减)即可支持无损多路复用,从而避免由潜在空间坍塌引发的捷径行为。进一步实验表明,多路推理可在需搜索的问题中实现并行探索。在涵盖四种语言模型的32个评估设置中,MUX显著优于强基线。消融与探测分析显示,学习到的潜在令牌编码了忠实且可解释的推理过程。结果表明,以局部学习目标的无损叠加构成实现强大高效连续潜空间推理的充分条件。

原文摘要 · Abstract (English)

Language models solve complex problems by articulating intermediate reasoning steps in natural language. While effective, this process is computationally bottlenecked: each reasoning step conveys only a single subword, and many are spent expressing a thought instead of carrying out computation. We propose MUX, a simple method for high-bandwidth and compact reasoning based on distillation of discrete reasoning into continuous multiplexed tokens in a latent space. Here, each latent token is trained to represent a weighted linear superposition (multiplexing) of a span of discrete reasoning subwords, where this superposition is lossless by construction and the span can be fully recovered (demultiplexing). We prove that simple position-dependent weightings, such as suitable geometric decay, support lossless multiplexing, which in turn prevents shortcut behaviors caused by latent collapse. We further show that multiplexed reasoning can perform parallel exploration in problems that require search. Across 32 evaluation settings spanning four language models, MUX outperforms strong latent reasoning baselines. Ablation and probing analyses further show that the learned latent tokens encode faithful and interpretable reasoning. Our results suggest that lossless superposition as local learning targets constitutes a sufficient condition for achieving strong and efficient latent continuous reasoning.

连续推理多路复用语言模型潜空间

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。