arXiv:2506.03230cs.LGcs.AI2025-06中稿 · ICLR被引 2

只微调模型权重矩阵的对角块,就能达到全模型微调的效果。

DiaBlo: Diagonal Blocks Are Sufficient For Finetuning

  • 仅更新选定权重矩阵的对角块,无需低秩分解或特殊初始化。
  • 在常识推理、代码生成等任务上表现接近全模型微调,且内存效率与LoRA相当。
  • 理论证明其表达能力优于LoRA,适合追求高效微调的开发者使用。

微调是将大语言模型适配到特定下游任务的关键步骤。为降低全模型微调带来的巨大计算和内存开销,参数高效微调(PEFT)方法被提出,仅更新少量参数。然而,现有PEFT方法与全模型微调之间仍存在性能差距。本文提出DiaBlo,一种简单有效的PEFT方法,仅更新选定模型权重矩阵的对角块。不同于低秩适配(LoRA)及其变体,DiaBlo无需低秩矩阵乘积,避免了对辅助初始化或定制优化策略的依赖,从而实现稳定且鲁棒的收敛,同时保持与LoRA相当的内存效率和训练速度。我们还提供了理论保证:在温和的低秩条件下,DiaBlo在线性问题中表达能力优于LoRA,且在一般非线性全微调中收敛至驻点。通过跨多项任务的大量实验(包括常识推理、算术推理、代码生成和安全对齐),我们表明仅微调对角块已足够实现强而一致的性能。DiaBlo不仅取得具有竞争力的准确率,还保持高内存效率和快速微调速度。代码已公开于https://github.com/ziyangjoy/DiaBlo。

原文摘要 · Abstract (English)

Fine-tuning is a critical step for adapting large language models (LLMs) to domain-specific downstream tasks. To mitigate the substantial computational and memory costs of full-model fine-tuning, Parameter-Efficient Fine-Tuning (PEFT) methods have been proposed to update only a small subset of model parameters. However, performance gaps between PEFT approaches and full-model fine-tuning still exist. In this work, we present DiaBlo, a simple yet effective PEFT approach that updates only the diagonal blocks of selected model weight matrices. Unlike Low-Rank Adaptation (LoRA) and its variants, DiaBlo eliminates the need for low-rank matrix products, thereby avoiding the reliance on auxiliary initialization schemes or customized optimization strategies to improve convergence. This design leads to stable and robust convergence while maintaining comparable memory efficiency and training speed to LoRA. Moreover, we provide theoretical guarantees showing that, under mild low-rank conditions, DiaBlo is more expressive than LoRA in the linear problem and converges to a stationary point of the general nonlinear full fine-tuning. Through extensive experiments across a range of tasks, including commonsense reasoning, arithmetic reasoning, code generation, and safety alignment, we show that fine-tuning only diagonal blocks is sufficient for strong and consistent performance. DiaBlo not only achieves competitive accuracy but also preserves high memory efficiency and fast fine-tuning speed. Codes are available at https://github.com/ziyangjoy/DiaBlo.

参数高效微调大模型微调对角块低秩适配

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