来源:arXiv · cs.AI查看原文 ↗
原文版权归来源方所有,本站仅作收录、翻译或格式整理。
事实脉络
arXiv:2608.12385v1 Announce Type: new Abstract: As large language models serve more requests, cumulative inference cost is becoming increasingly important relative to one-time training cost. The two inference phases str
解读与影响
Dual-Flow Transformers:将预填充主路径与额外解码计算解耦
研究问题与方法
传统 Transformer 在推理时,预填充阶段一次性处理全部输入提示并生成 KV 缓存,随后解码阶段逐 token 生成输出。两个阶段共享同一套前向计算路径,这意味着解码阶段每一次迭代都要经过与预填充相同深度的完整网络。论文作者观察到,预填充阶段承担的是"理解输入"的密集计算任务,而解码阶段更多是"基于已有上下文续写"的增量计算,两者对模型容量的需求并不对称。基于这一观察,Dual-Flow Transformers 将主预填充路径与额外的解码计算解耦,使解码阶段可以走一条更轻量的计算通道。原文摘要在此处截断,具体实现细节(如是否引入独立的轻量解码模块、是否对注意力机制做稀疏化处理等)原文未提供。
关键发现与证据强度
从摘要透露的信息来看,论文的核心主张是推理成本的累积效应在服务规模扩大后不可忽视,而解耦两条计算流是应对这一问题的方向之一。不过,由于目前公开的摘要内容有限,具体的实验设置、基准测试结果、延迟与吞吐量数据均未在给定素材中呈现。该论文标注为 arXiv 新提交(v1),尚未经过同行评议,因此其结论应视为初步的技术提案而非经过验证的成熟方案。相关素材中同日发布的几篇论文分别涉及 LLM 安全对齐的多语言差异、自反思机制、AI 对齐方法以及多语言语音识别,与本篇在主题上无直接交叉,无法为 Dual-Flow Transformers 的效果提供旁证。
意义与局限
如果解耦思路成立,其实际意义在于:对推理服务提供商而言,解码阶段的计算量通常远大于预填充阶段(因为解码要逐 token 迭代),任何针对解码路径的轻量化改造都可能直接转化为可观的成本节约。但局限同样明显——摘要没有给出该方法是否会影响生成质量、长文本场景下 KV 缓存管理是否变得更加复杂、以及该架构与现有推理优化技术(如投机解码、KV 缓存压缩)之间是互补还是重叠。在这些细节公开之前,Dual-Flow Transformers 更宜被看作一个值得关注的方向性信号,而非可以立即评估的成熟方案。
参考来源
- arXiv · cs.AI ↗
- Don't Want Your LLM to Recommend Nuclear Strike? Try Asking It in Japanese ↗
- What Drives LLM Self-Reflection? A Controlled Ablation of Uncertainty Routing in Armed Conflict Forecasting ↗
- Position: We Need Practical AI Alignment Methods to Mirror Human Reasoning ↗
- Comparative Analysis of Multilingual Pre-trained Models for Nepali Automatic Speech Recognition ↗
- Learning to Adapt Cross-Domain Preferences via Meta-LoRA for LLM Personalization ↗
来源原文
arXiv:2608.12385v1 Announce Type: new Abstract: As large language models serve more requests, cumulative inference cost is becoming increasingly important relative to one-time training cost. The two inference phases stress hardware differently: prompt prefill is parallel and typically compute-bound, whereas autoregressive decode is sequential and often memory-bandwidth-bound. Conventional width or depth scaling increases both costs together because every added layer is evaluated in both phases. We ask whether additional learned computation can instead be allocated to continuation prediction while preserving the prompt-wide primary computation and a single persistent key-value (KV) cache. We introduce the Dual-Flow Transformer. Its primary flow is a complete causal language model that processes the prompt and writes the KV cache. The auxiliary flow is omitted during prompt processing and activated only from the final prompt position onward, adding continuation-prediction computation without writing persistent state or influencing the primary flow. The two flows share major attention, MLP, and output matrices, while using separate token embeddings and lightweight coupling. Sharing weights and the primary cache also creates opportunities to reuse loaded weights and cached keys and values during grouped execution. Across matched-token comparisons, Dual-Flow achieves lower validation loss across architectures and data configurations. In MoE models, the separation makes primary and auxiliary expert fan-outs independent controls over prompt cost, continuation cost, and predictive quality. We study two regimes: increasing decode computation at fixed prefill expert computation, and reallocating a fixed decode expert budget between the two flows. These experiments expose a prefill-decode-quality trade-off and demonstrate the potential of phase-specific expert allocation.