通过计算预算分配,让模型在推理时更高效。
Budgeted LoRA: Distillation as Structured Compute Allocation for Efficient Inference

- 将压缩视为计算资源分配问题,动态调整密集与低秩路径比例。
- 在4.05倍加速下仅轻微降低困惑度,保留函数式上下文学习能力。
- 适合追求推理效率的部署场景,尤其关注低延迟应用。
我们研究在显式计算约束下的大语言模型蒸馏,目标是生成不仅训练成本低,且推理时结构高效的学生模型。现有参数高效蒸馏方法如LoRA虽降低适配成本,但未改变密集主干,无法带来显著推理收益。我们提出预算型LoRA(Budgeted LoRA),将模型压缩建模为结构化计算分配问题。不固定学生架构,而是引入全局计算预算,决定最终保留的密集计算比例。在此约束下,模型通过模块级密集保留系数、自适应低秩分配及后训练压缩(选择性移除、近似或保留密集组件)重新分配容量。该框架生成一组由单一预算调节的学生模型。实验表明,在中等预算下,其困惑度与标准LoRA相当,压缩模块速度提升1.74倍;在激进预算下实现4.05倍加速,困惑度略有下降,且在函数式上下文学习探测任务中保持更高准确率。结果表明,在计算受限蒸馏中,保持行为的关键不在于匹配困惑度或减少参数量,而在于控制密集计算向低秩路径的转移方式。
原文摘要 · Abstract (English)
We study distillation for large language models under explicit compute constraints, with the goal of producing student models that are not only cheaper to train, but structurally efficient at inference time. While prior approaches to parameter-efficient distillation, such as LoRA, reduce adaptation cost, they leave the dense backbone unchanged and therefore fail to deliver meaningful inference savings. We propose Budgeted LoRA, a distillation framework that treats model compression as a structured compute allocation problem. Instead of using a fixed student architecture, we introduce a global compute budget that sets the final target fraction of dense computation retained. Under this constraint, the model redistributes capacity across dense and low-rank pathways via (i) module-level dense retention coefficients, (ii) adaptive low-rank allocation, and (iii) post-training compression that selectively removes, approximates, or preserves dense components. This formulation yields a family of students controlled by a single budget dial. Empirically, Budgeted LoRA matches standard LoRA perplexity at a moderate budget with a 1.74x compressed-module speedup; at an aggressive budget it achieves a 4.05x speedup with moderate perplexity degradation, and it preserves higher accuracy on function-style in-context learning probes. These results suggest that, under compute-constrained distillation, retaining behavior is less about matching perplexity or removing more parameters than it is about controlling how dense computation is transferred to low-rank pathways.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。