arXiv:2509.05915cs.CL2025-09被引 1

通过协同设计算法与架构,让大模型推理更快更省资源。

Accelerating Large Language Model Inference via Early-Exiting Algorithms

  • 用并行解码减少早期退出的计算开销
  • 深度参数共享解决动态推理的同步瓶颈
  • 预训练轻量路由,自动决定每词推理深度

大语言模型虽能力强大,但部署成本高昂。自适应计算方法如早期退出本可节省算力,却常因逐令牌动态性导致批处理吞吐下降,形成悖论。本文通过协同设计自适应算法与模型架构,实现动态性与效率的平衡。首先提出高效并行解码机制,降低传统早期退出的开销;其次证明深度参数共享不仅使模型紧凑、参数高效,还能天然缓解动态推理中的同步问题;最后构建统一框架,预训练轻量路由,为每个令牌动态分配最优递归深度。该方法在单一模型中同时优化自适应计算与参数效率,确立了效率与性能的新帕累托前沿。

原文摘要 · Abstract (English)

Large language models have achieved remarkable capabilities, but their practical deployment is hindered by significant computational costs. While adaptive computation methods like early-exiting promise to reduce these costs, they introduce a fundamental conflict: the per-token dynamism intended to save computation often creates system-level bottlenecks that can paradoxically reduce throughput in batched inference. This dissertation resolves this conflict by co-designing adaptive algorithms and model architectures to strike an optimal balance between dynamism and efficiency. To this end, our work first addresses critical sources of overhead in conventional early-exiting by proposing an efficient parallel decoding mechanism. We then show that deep parameter sharing provides an architectural foundation that not only yields compact, parameter-efficient models but also inherently mitigates the critical synchronization issues affecting dynamic inference. Finally, this work presents a unified framework where lightweight routers are pretrained to dynamically assign an optimal recursion depth for each token. This approach establishes a new Pareto frontier between efficiency and performance by effectively optimizing for both adaptive computation and parameter efficiency within a single model.

大模型推理早期退出自适应计算参数共享

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