arXiv:2512.14237cs.CLcs.LG2025-12

用轻量侧网络实现低内存微调,7B模型单卡可跑2000词上下文。

Ladder Up, Memory Down: Low-Cost Fine-Tuning With Side Nets

  • 添加轻量侧网络替代全模型反向传播,减少内存占用。
  • 在多个任务上达到QLoRA精度,峰值内存降低50%。
  • 适合显存受限的消费级显卡用户,尤其适合长文本推理。

大语言模型微调常受消费级显卡内存限制。参数高效微调(PEFT)方法如QLoRA虽减少可训练参数,但仍因全模型反向传播导致高内存开销。我们重新研究一种少被关注的PEFT技术——梯子侧网络微调(LST),发现其计算扩展斜率与QLoRA相当,同时将峰值内存降低50%。在涵盖自然语言理解、数学推理和LLM评判等下游任务中,LST平均性能与QLoRA相当,且显著更省内存。该效率使得7B参数模型可在单张12 GB GPU上以2000令牌上下文进行微调,无需梯度检查点,而QLoRA在此条件下会内存溢出。此外,我们建立缩放定律,证明LST与QLoRA具有相似扩展特性。通过引入xLadder——一种通过跨连接扩展有效深度的变体,在固定参数量下增强推理深度并缩短思维链(CoT)。当内存是瓶颈时,LST表现优异;而xLadder进一步在不增加内存开销的前提下实现更深推理。

原文摘要 · Abstract (English)

Fine-tuning large language models (LLMs) is often limited by the memory available on commodity GPUs. Parameter-efficient fine-tuning (PEFT) methods such as QLoRA reduce the number of trainable parameters, yet still incur high memory usage induced by the backward pass in the full model. We revisit Ladder Side Tuning (LST), a rarely explored PEFT technique that adds a lightweight side network, and show that it matches QLoRA's compute scaling slope while cutting peak memory by 50\%. Across different downstream benchmarks spanning natural language understanding, mathematical and LLM-critic tasks, LST has competitive performance with QLoRA's accuracy on average while being much more memory-efficient. This efficiency enables fine-tuning of 7B-parameter models on a single 12 GB consumer GPU with 2k-token contexts, requiring no gradient checkpointing\textemdash conditions under which QLoRA exhausts memory. Beyond memory efficiency, we also establish scaling laws showing that LST scales similarly to QLoRA. We exploit Ladder's architectural flexibility by introducing xLadder, a depth-extended variant that increases effective depth via cross-connections and shortens chain-of-thought (CoT) at fixed parameter count. Ladder is strong when memory is the bottleneck; xLadder builds on this by enabling deeper reasoning without additional memory overhead.

微调低内存侧网络7B模型

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