arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2601.02569cs.CL
收藏

LoRA-Drop:用于高效LLM推理的时序LoRA解码

LoRA-Drop: Temporal LoRA Decoding for Efficient LLM Inference

University of Waterloo(滑铁卢大学);University of Toronto(多伦多大学);Queen’s University(皇后大学)

Hossein Rajabzadeh, Maryam Dialameh, Chul B. Park, Il-Min Kim, Hyock Ju Kwon

AI总结:

LoRA-Drop通过时序计算计划和低秩LoRA校正,实现高效LLM推理,提升解码速度2.6倍并减少45-55%的KV缓存占用。

摘要:

自回归大语言模型(LLMs)受限于顺序解码,每个新token通常需要执行所有transformer层。现有的动态深度和层跳过方法减少了这种成本,但通常依赖于辅助路由机制或在跳过层未补偿时导致精度下降。我们提出了LoRA-Drop,一种即插即用的推理框架,通过在固定子集的中间层上应用时序计算计划来加速解码:在大多数解码步骤中,选定的层重用前一token的隐藏状态并应用低秩LoRA校正,而周期性的刷新步骤则执行完整模型以防止漂移。LoRA-Drop不需要路由网络,与标准KV缓存兼容,并且可以在LoRA步骤中通过跳过可丢弃层的KV更新来减少KV缓存足迹。在LLaMA2-7B、LLaMA3-8B、Qwen2.5-7B和Qwen2.5-14B上,LoRA-Drop实现了最高2.6倍的解码速度提升和45-55%的KV缓存减少,同时保持在基线精度0.5个百分点以内。在推理(GSM8K、MATH、BBH)、代码生成(HumanEval、MBPP)和长上下文/多语言基准(LongBench、XNLI、XCOPA)上的评估发现了一致的调度配置安全区,该安全区在保持质量的同时提供了显著的效率提升,为LLM的自适应容量推理提供了一条简单的路径。代码可在https://github.com/hosseinbv/LoRA-Drop.git获取。

英文摘要:

Autoregressive large language models (LLMs) are bottlenecked by sequential decoding, where each new token typically requires executing all transformer layers. Existing dynamic-depth and layer-skipping methods reduce this cost, but often rely on auxiliary routing mechanisms or incur accuracy degradation when bypassed layers are left uncompensated. We present \textbf{LoRA-Drop}, a plug-and-play inference framework that accelerates decoding by applying a \emph{temporal compute schedule} to a fixed subset of intermediate layers: on most decoding steps, selected layers reuse the previous-token hidden state and apply a low-rank LoRA correction, while periodic \emph{refresh} steps execute the full model to prevent drift. LoRA-Drop requires no routing network, is compatible with standard KV caching, and can reduce KV-cache footprint by skipping KV updates in droppable layers during LoRA steps and refreshing periodically. Across \textbf{LLaMA2-7B}, \textbf{LLaMA3-8B}, \textbf{Qwen2.5-7B}, and \textbf{Qwen2.5-14B}, LoRA-Drop achieves up to \textbf{2.6$\times$ faster decoding} and \textbf{45--55\% KV-cache reduction} while staying within \textbf{0.5 percentage points (pp)} of baseline accuracy. Evaluations on reasoning (GSM8K, MATH, BBH), code generation (HumanEval, MBPP), and long-context/multilingual benchmarks (LongBench, XNLI, XCOPA) identify a consistent \emph{safe zone} of scheduling configurations that preserves quality while delivering substantial efficiency gains, providing a simple path toward adaptive-capacity inference in LLMs. Codes are available at https://github.com/hosseinbv/LoRA-Drop.git.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑