arXiv:2511.16193cs.DCcs.AI2025-11被引 5

用快速推测法加速大模型微调推理,速度提升2.4倍

Fast LLM Post-training via Decoupled and Fastest-of-N Speculation

  • 分离式推测机制解决大批量训练时的计算效率问题
  • 动态选择最优推测方法,实现最高2.7倍加速
  • 适用于密集模型和MoE模型,适合大规模训练场景

大语言模型微调中,生成阶段的推理占用了大部分训练时间。本文提出SpecActor,通过推测解码加速不可并行的生成过程,利用快速草稿路径生成候选输出,并通过原模型并行验证确保正确性。针对推测效率的两大瓶颈,提出:(1) 分离式推测方法,克服大批次配置下推测解码的计算低效问题;(2) 最快者选优的推测策略,根据生成进度动态选择并组合不同草稿方法,即使事先未知最优方法也能逼近最佳性能。在生产级数据追踪中,SpecActor使平均推理速度提升2.0–2.4倍(最高达2.7倍),在稠密和MoE模型上均表现稳定,且相比基线推测方案快1.1–2.6倍。端到端训练时间因此提速1.4–2.3倍。

原文摘要 · Abstract (English)

Rollout dominates the training time in large language model (LLM) post-training, where the trained model is used to generate tokens given a batch of prompts. This work, SpecActor, achieves fast rollout with speculative decoding that deploys a fast draft path to accelerate the unparallelizable generation, while the correctness is guaranteed by fast parallel verification of the outputs with the original model. SpecActor addresses two foundational challenges that hinder speculation efficiency: (1) a Decoupled speculation method that overcomes the computation inefficiency issue when executing speculative decoding with relative large per-worker batch size -- a common configuration in training but unfriendly to speculation, and (2) a Fastest-of-N speculation method that selects and combines different draft methods according to the rollout progress to approximate the optimal draft method even when the best one is unknown a priori. Extensive evaluations on production traces show that SpecActor accelerates mean rollout speed by 2.0--2.4x, with up to 2.7x speedup, over common post-training baselines. The results are consistent across both dense and MoE models and across different RL algorithms. Notably, SpecActor is 1.1--2.6x faster compared to vanilla speculative rollout in different traces. The accelerated rollout achieves 1.4--2.3x faster end-to-end training time.

大模型微调推测解码加速训练MoE

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