用预训练权重分解初始化,让大模型微调更高效准确
DoTA: Weight-Decomposed Tensor Adaptation for Large Language Models
- 用MPO分解预训练权重做初始化,替代随机初始化
- 参数少于1%时仍保持比随机初始化更好的推理性能
- 适合资源受限场景,尤其对4比特量化有优化支持
低秩适配(LoRA)通过低秩矩阵近似降低大语言模型微调的计算与内存开销。然而,二维空间中的低秩近似难以捕捉目标矩阵中的高维结构。近期张量分解方法被用于大模型微调,因其能提取结构性信息。但这些方法多依赖随机初始化,且初始化对张量适配的影响尚未深入研究。本文发现,随机初始化与全量微调的验证损失存在显著差异。为此,我们提出权重分解张量适配(DoTA),利用预训练权重的矩阵乘积算子(MPO)分解进行有效初始化。此外,我们提出QDoTA,一种针对4比特量化的版本。在常识与算术推理任务上的实验表明,DoTA在参数更少的情况下优于随机初始化方法;QDoTA进一步降低内存消耗,在常识推理任务上表现与DoTA相当。代码将公开以支持后续研究。
原文摘要 · Abstract (English)
Low-rank adaptation (LoRA) reduces the computational and memory demands of fine-tuning large language models (LLMs) by approximating updates with low-rank matrices. However, low-rank approximation in two-dimensional space fails to capture high-dimensional structures within the target matrix. Recently, tensor decomposition methods have been explored for fine-tuning LLMs, leveraging their ability to extract structured information. Yet, these approaches primarily rely on random initialization, and the impact of initialization on tensor adaptation remains underexplored. In this paper, we reveal that random initialization significantly diverges from the validation loss achieved by full fine-tuning. To address this, we propose Weight-Decomposed Tensor Adaptation (DoTA), which leverages the Matrix Product Operator (MPO) decomposition of pre-trained weights for effective initialization in fine-tuning LLMs. Additionally, we introduce QDoTA, a quantized version of DoTA designed for 4-bit quantization. Experiments on commonsense and arithmetic reasoning tasks show that DoTA outperforms random initialization methods with fewer parameters. QDoTA further reduces memory consumption and achieves comparable performance to DoTA on commonsense reasoning tasks. We will release our code to support future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。