仅用100样本和一次梯度步,无须微调即可高效适配大模型。
Compress to Impress: Efficient LLM Adaptation Using a Single Gradient Step on 100 Samples
- 通过梯度分析筛选关键层,避免全层扫描。
- 在100样本上仅一步梯度更新,准确率提升最高达24.6个百分点。
- 适合快速部署、资源有限场景下的模型轻量适配。
近期,Sharma等人提出的层选择性秩压缩(LASER)方法表明,对大语言模型权重矩阵中高阶成分进行剪枝可提升下游任务精度,且无需基于梯度的微调。然而,该方法需逐层遍历并执行全数据集前向传播,难以快速部署。本文发现:(i)只需检查少量精心挑选的矩阵即可消除逐层搜索;(ii)各矩阵奇异值的梯度可有效指示哪些矩阵值得压缩;(iii)允许矩阵行在多个子空间中聚类,并分别分解,进一步降低过拟合,准确率最高提升24.6个百分点;(iv)仅使用100个样本计算梯度与评估性能即足够,因下游适配主要受提示风格影响而非数据集规模。综合上述发现,我们提出一种无需微调的快速鲁棒适配算法,仅需一次梯度步与对候选层及分解方式的快速扫描,即可完成模型适配。
原文摘要 · Abstract (English)
Recently, Sharma et al. suggested a method called Layer-SElective-Rank reduction (LASER) which demonstrated that pruning high-order components of carefully chosen LLM's weight matrices can boost downstream accuracy -- without any gradient-based fine-tuning. Yet LASER's exhaustive, per-matrix search (each requiring full-dataset forward passes) makes it impractical for rapid deployment. We demonstrate that this overhead can be removed and find that: (i) Only a small, carefully chosen subset of matrices needs to be inspected -- eliminating the layer-by-layer sweep, (ii) The gradient of each matrix's singular values pinpoints which matrices merit reduction, (iii) Increasing the factorization search space by allowing matrices rows to cluster around multiple subspaces and then decomposing each cluster separately further reduces overfitting on the original training data and further lifts accuracy by up to 24.6 percentage points, and finally, (iv) we discover that evaluating on just 100 samples rather than the full training data -- both for computing the indicative gradients and for measuring the final accuracy -- suffices to further reduce the search time; we explain that as adaptation to downstream tasks is dominated by prompting style, not dataset size. As a result, we show that combining these findings yields a fast and robust adaptation algorithm for downstream tasks. Overall, with a single gradient step on 100 examples and a quick scan of the top candidate layers and factorization techniques, we can adapt LLMs to new datasets -- entirely without fine-tuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。