arXiv:2602.09395cs.LG2026-02被引 1

通过稀疏化选择计算层,让微调更高效且不损失性能。

Sparse Layer Sharpness-Aware Minimization for Efficient Fine-Tuning

  • 用多臂赌博机动态选层,只对部分层做扰动和更新。
  • 在视觉、中大型语言模型上分别仅激活47%、22%、21%参数。
  • 保持顶尖性能的同时大幅降低计算开销,适合资源受限场景。

尖锐感知最小化(SAM)通过寻找平坦的损失曲面来提升模型泛化能力,但在微调中因额外的参数扰动步骤使计算成本翻倍,成为实际应用瓶颈。本文提出SL-SAM方法,引入稀疏技术于网络层,将扰动与更新阶段的层选择建模为多臂赌博机问题。每轮迭代开始时,根据梯度范数采样部分层参与反向传播,显著降低计算复杂度。我们提供了收敛性分析。在多个任务的微调实验中,SL-SAM表现媲美最先进基线,甚至在大语言模型微调中取得第一;同时相比原始SAM,其反向传播活跃参数比例分别降至47%(视觉)、22%(中型语言模型)和21%(大型语言模型),验证了算法效率。

原文摘要 · Abstract (English)

Sharpness-aware minimization (SAM) seeks the minima with a flat loss landscape to improve the generalization performance in machine learning tasks, including fine-tuning. However, its extra parameter perturbation step doubles the computation cost, which becomes the bottleneck of SAM in the practical implementation. In this work, we propose an approach SL-SAM to break this bottleneck by introducing the sparse technique to layers. Our key innovation is to frame the dynamic selection of layers for both the gradient ascent (perturbation) and descent (update) steps as a multi-armed bandit problem. At the beginning of each iteration, SL-SAM samples a part of the layers of the model according to the gradient norm to participate in the backpropagation of the following parameter perturbation and update steps, thereby reducing the computation complexity. We then provide the analysis to guarantee the convergence of SL-SAM. In the experiments of fine-tuning models in several tasks, SL-SAM achieves the performances comparable to the state-of-the-art baselines, including a \#1 rank on LLM fine-tuning. Meanwhile, SL-SAM significantly reduces the ratio of active parameters in backpropagation compared to vanilla SAM (SL-SAM activates 47\%, 22\% and 21\% parameters on the vision, moderate and large language model respectively while vanilla SAM always activates 100\%), verifying the efficiency of our proposed algorithm.

微调优化稀疏计算SAM高效训练

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