arXiv:2604.09406cs.LG2026-04

通过在线学习激活子空间,实现大模型训练内存减半且不降性能。

OASIS: Online Activation Subspace Learning for Memory-Efficient Training

  • 训练中动态更新低维激活子空间,实时压缩中间结果。
  • 峰值内存降低2倍,微调性能媲美全参数训练。
  • 适合资源受限下高效训练大语言模型的研究者。

大语言模型训练受内存限制,其中激活值占用大量存储。现有方法通过低秩参数化或低秩梯度子空间减少内存,而激活内存则依赖架构修改或周期性投影压缩。本文提出OASIS,一种在线激活子空间学习算法,在训练过程中持续追踪并更新低维激活子空间。中间激活值被投影到该动态子空间,实现内存节省且无需改变前向计算。演化后的子空间生成低秩梯度表示,使梯度与优化器状态可直接在子空间中维护,并通过投影感知优化器确保状态在子空间更新时稳定传递。在多种微调与预训练任务中,OASIS达到全参数微调2倍的内存降低,性能相当且优于先前低秩方法。

原文摘要 · Abstract (English)

Training large language models (LLMs) is constrained by memory requirements, with activations accounting for a substantial fraction of the total footprint. Existing approaches reduce memory using low-rank weight parameterizations or low-rank gradient subspaces for optimizer states, while activation memory is addressed through architectural modifications or compression schemes based on periodically updated projections. We propose OASIS, an online activation subspace learning algorithm for memory-efficient training that tracks and continuously updates a low-dimensional activation subspace during training. Intermediate activations are projected onto this evolving subspace, reducing memory without modifying forward-pass computations. The evolving activation subspace induces low-rank gradient representations, enabling both gradients and optimizer states to be maintained directly in this subspace, while a projection-aware optimizer consistently transports optimizer states across subspace updates for stable training. Across various finetuning and pretraining tasks, OASIS achieves up to $2\times$ lower peak memory than full fine-tuning while matching its performance and outperforming prior low-rank methods.

大模型训练内存优化子空间学习

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