提出轻量级低秩微调方法,提升大模型泛化能力且节省显存。
LORENZA: Enhancing Generalization in Low-Rank Gradient LLM Training via Efficient Zeroth-Order Adaptive SAM
- 用单次梯度估算实现高效自适应优化,避免传统方法的高计算开销。
- 在多个数据集上相比基线提升1.8%~4.2%准确率,显著增强泛化性能。
- 适合资源受限场景,尤其适用于移动端或边缘设备的大模型微调。
我们研究了鲁棒的参数高效微调(PEFT)技术,旨在严格计算与内存约束下提升大语言模型(LLMs)的准确率与泛化能力。现有方法常缺乏鲁棒性,在多样任务间泛化效果差,导致真实场景表现不佳。为此,我们提出一种新型高效计算框架AdaZo-SAM,结合Adam与尖锐感知最小化(SAM),每轮仅需一次梯度计算即可完成优化,通过随机零阶估计获得SAM的上升扰动。我们提供了AdaZo-SAM的收敛性保证,并证明其能有效提升先进PEFT方法的泛化能力。进一步设计了名为LORENZA的低秩梯度优化方法,为AdaZo-SAM的内存高效版本。LORENZA采用随机SVD方案高效计算子空间投影矩阵,并在选定子空间上执行优化步骤。该方法实现了全参数微调的同时保持与低秩梯度投影方法相当的内存消耗。我们给出了LORENZA的收敛分析,并验证其在预训练与微调阶段的优势。
原文摘要 · Abstract (English)
We study robust parameter-efficient fine-tuning (PEFT) techniques designed to improve accuracy and generalization while operating within strict computational and memory hardware constraints, specifically focusing on large-language models (LLMs). Existing PEFT methods often lack robustness and fail to generalize effectively across diverse tasks, leading to suboptimal performance in real-world scenarios. To address this, we present a new highly computationally efficient framework called AdaZo-SAM, combining Adam and Sharpness-Aware Minimization (SAM) while requiring only a single-gradient computation in every iteration. This is achieved using a stochastic zeroth-order estimation to find SAM's ascent perturbation. We provide a convergence guarantee for AdaZo-SAM and show that it improves the generalization ability of state-of-the-art PEFT methods. Additionally, we design a low-rank gradient optimization method named LORENZA, which is a memory-efficient version of AdaZo-SAM. LORENZA utilizes a randomized SVD scheme to efficiently compute the subspace projection matrix and apply optimization steps onto the selected subspace. This technique enables full-parameter fine-tuning with adaptive low-rank gradient updates, achieving the same reduced memory consumption as gradient-low-rank-projection methods. We provide a convergence analysis of LORENZA and demonstrate its merits for pre-training and fine-tuning LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。