arXiv:2604.07808cs.CLcs.LG2026-04ACL被引 1

GRASS通过动态调整层采样策略,实现高效低内存大模型微调。

GRASS: Gradient-based Adaptive Layer-wise Importance Sampling for Memory-efficient Large Language Model Fine-tuning

  • 基于梯度均值动态评估各层重要性,随任务和训练阶段自适应调整。
  • 在多个基准上平均性能提升4.38点,内存占用减少最多19.97%。
  • 适合资源受限环境下需高精度微调的大模型应用。

大语言模型全参数微调受制于巨大的GPU内存需求。低秩适配方法虽通过仅更新部分参数缓解该问题,但常限制模型表达能力并导致性能下降。分层微调方法作为替代方案,通过静态层重要性采样实现内存高效训练,但忽略了层重要性随任务和训练阶段的变化,导致下游任务表现不佳。为此,我们提出GRASS——一种基于梯度的自适应分层重要性采样框架。GRASS利用梯度均值作为任务感知与训练阶段感知的层重要性指标,并通过自适应训练策略动态调整层采样概率。同时引入分层优化器状态卸载机制,通过重叠计算与通信进一步降低内存开销,同时保持相近的训练吞吐量。在多个模型与基准上的大量实验表明,GRASS持续优于现有先进方法,平均准确率提升最高达4.38点,内存使用减少最多19.97%。

原文摘要 · Abstract (English)

Full-parameter fine-tuning of large language models is constrained by substantial GPU memory requirements. Low-rank adaptation methods mitigate this challenge by updating only a subset of parameters. However, these approaches often limit model expressiveness and yield lower performance than full-parameter fine-tuning. Layer-wise fine-tuning methods have emerged as an alternative, enabling memory-efficient training through static layer importance sampling strategies. However, these methods overlook variations in layer importance across tasks and training stages, resulting in suboptimal performance on downstream tasks. To address these limitations, we propose GRASS, a gradient-based adaptive layer-wise importance sampling framework. GRASS utilizes mean gradient norms as a task-aware and training-stage-aware metric for estimating layer importance. Furthermore, GRASS adaptively adjusts layer sampling probabilities through an adaptive training strategy. We also introduce a layer-wise optimizer state offloading mechanism that overlaps computation and communication to further reduce memory usage while maintaining comparable training throughput. Extensive experiments across multiple models and benchmarks demonstrate that GRASS consistently outperforms state-of-the-art methods, achieving an average accuracy improvement of up to 4.38 points and reducing memory usage by up to 19.97\%.

大模型微调内存优化自适应采样

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