(EMNLP-2023)预训练语言模型的稀疏低秩自适应

预训练语言模型的稀疏低秩自适应

paper是清华大学发表在EMNLP 2023的工作

paper title：Sparse Low-rank Adaptation of Pre-trained Language Models

Code:https://github.com/tsinghuac3i/sora

Abstract

以参数高效的方式对预训练的大型语言模型进行微调因其有效性和效率而受到广泛研究。 流行的低秩自适应 (LoRA) 方法提供了一种值得注意的方法，假设自适应过程本质上是低维的。尽管 LoRA 表现出了令人称赞的性能，但它是以固定且不可改变的内在秩来实现的，这可能并不总是理想的选择。认识到需要更灵活的自适应，我们将 LoRA 方法扩展到一种创新方法，我们称之为稀疏低秩自适应 (SoRA)，该方法可以在自适应过程中动态调整内在秩。我们通过在训练阶段加入使用近端梯度法优化的门单元来实现这一点，在门的稀疏性下控制秩的基数。在随后的推理阶段，我们消除与清零秩相对应的参数块，以将每个 SoRA 模块简化为简洁但秩最优的 LoRA。我们的方法通过以更高的秩初始化 LoRA 来增强其表示能力，同时通过稀疏方式更新有效地控制暂时增加的参数数量。我们进一步为 SoRA 引入了稀疏化调度程序，旨在检查非零参数数量对模型记忆和泛化的影响。我们的实验结果表明，即使保留 70% 的参数和 70% 的训练时间，SoRA 也能胜过其他基线。

1 Introduction

以参数高效 (He et al, 2022; Ding et al, 2023; Hu et al, 2023) 的方式调整大规模预训练语言模型 (Devlin et al, 2019; Brown et al, 2020; He et al, 2020; Bommasani et al, 2021; Han et al, 2021; Touvron et al, 2023) 在研究界越来越受到关注。该范式的方法通常保持底层模型的大部分参数不变，要么在模型中插入额外的可训练参数（Houlsby 等人，2019；Li 和 Liang，2021），要么指定少量参数（Zaken 等人，2021；Liu 等人，2021；Su 等人，2023）可训练，要么将适应过程重新参数化为更有效的形式（Hu 等人，2021；Qin 等人，2021）。它们已被证实在各种模型和任务中都是有效的，通常产生与全参数微调相当甚至更好的结果。

参数高效微调的发展潜力在经过广泛的性能验证后变得显而易见。这些方法提供了调整基础模型以适应任何数据的机会，从而允许针对特定任务和个性化用户特征定制语言模型。由于优化参数的轻量级特性，它们可以无缝插入模型中，从而实现有针对性的增强。在这些方法中，低秩自适应（LORA（Hu et al，2021））被认为是目前最有效的方法之一。它假设模型参数在自适应后的变化是“内在低维的”，并通过优化从低秩分解获得的矩阵来执行自适应。LoRA 避免了插入额外神经模块导致的前向传播延迟，同时表现出稳定的性能。 虽然有效，但内在秩（通常作为超参数）的设置仍不清楚。直观地说，更大的秩带来更大的优化空间，并创造了处理更具挑战性的任务的能力。然而，在实践中，最佳内在排名会根据主干模型和任务等多种因素而变化。

考虑到在大规模模型（例如具有 1750 亿个参数的 GPT-3（Brown 等人，2020 年）和具有 7 亿至 650 亿个参数的 LLaMA（Touvron 等人，2023 年））上搜索超参数的巨大计算成本，开发一种基于自适应秩的方法是一种自然的方法。现有的一些工作试图探索这个方向（Valipour 等人，2022 年；Zhang 等人，2023 年），但它们大多是启发式的或引入了额外的成本。在本文中，我们提出了 SoRA，一种简单、有效且自动化的自适应参数高效微调方法。我们引入了一个在 L1 正则化下具有近端梯度下降更新的门控模块来控制更新矩阵的稀疏性。训练后，门控向量的零条目记录下投影矩阵的列和上投影矩阵的行，可以简单地将其删除并以更参数高效的方式存储。与其他自适应方法相比，近端梯度法具有明确的数学含义，并且不必涉及其他计算和启发式方法。例如，AdaLoRA（Zhang 等人，2023 年）引入了一个额外的正则化器，以确保下部和上部投影矩阵严格遵守奇异值分解 (SVD) 的定义，每个矩阵都是正交的。然而，由于梯度计算，这个正则化项会产生大量的计算开销。相反，我们消除了这一要求，而是通过控制中间对角矩阵来选择性地过滤低秩分量。我们在第 3 节中详细比较了 SoRA 和相关方法。

SoRA 的机制还允许我们暂时控制稀疏性，并研究非零可训练参数的数量与记忆和泛化能力之间的关系。我们提出了一个稀疏化调度程序，并发现模型自适应过程表现出强大的“压缩能力”，即使一小部分参数（低于 LoRA 秩 1）也可以保持相当好的性能。进行了大量的实验来证明我们方法的有效性。特别是，我们的模型可以在广泛的下游任务上以更少的参数和缩短 30% 的训练时间持续优于参数高效的基线。

2 A Closer Look to Adaptive Rank

相关工作。在介绍我们的方法之前，我们首先简要回顾一下参数高效调优和我们的骨干低秩自适应 (LoRA)。 参数高效调优是一组仅优化一小部分参数并保持主模型不受影响以进行自适应的方法。 一些参数高效方法会将额外的神经模块或参数插入主干模型，例如适配器 (Houlsby 等人，2019)、前缀和提示调优 (Li and Liang，2021；Lester 等人，2021)。另一类此类方法试图将特定参数指定为可训练或可修剪 (Guo 等人，2021；Zhao 等人，2020；Zaken 等人，2021)。研究人员推导出一系列参数高效方法的变体来提高有效性或效率 (Karimi Mahabadi et al, 2021; Hu et al, 2022; Sung et al, 2022; He et al, 2022)。最近，参数高效微调的应用扩展到多模态和指令调整场景 (Gao et al, 2023; Dettmers et al, 2023)。在本文中，我们更多地关注 LoRA (Hu et al, 2021)，它使用低秩矩阵来近似权重的变化。

在 LoRA 中，预训练权重（记为 ${\mathbf{W}}_0\in {\mathbb{R}}^{p\times q}$）是冻结的，而可训练的 LoRA 模块是每个权重矩阵变化 $\mathbf{\Delta }$ 的低秩分解矩阵 ${\mathbf{W}}_d\in {\mathbb{R}}^{r\times q}$ 和 ${\mathbf{W}}_u\in {\mathbb{R}}^{p\times r}$，其中 $\mathbf{\Delta }={\mathbf{W}}_u{\mathbf{W}}_d\in {\mathbb{R}}^{p\times q}$。这样，当前层的输出 $\mathbf{h}$ 可以表示为：

$$\mathbf{y}\leftarrow {\mathbf{W}}_0\mathbf{x}+{\mathbf{W}}_u{\mathbf{W}}_d\mathbf{x}$$

其中 r ≪ min ⁡ { p , q } r \ll \min \{p, q\} r≪min{p,q} 是一个控制低秩矩阵大小和可训练参数数量的超参数，称为“内在维度”。在本节中，我们主要关注最后一项，记为 $\mathbf{z}\leftarrow {\mathbf{W}}_u{\mathbf{W}}_d\mathbf{x}$。

LoRA 上的自适应排序。尽管在可处理性和效率方面取得了很大进步，但 LoRA 仍然受到其在选择最佳排序 r 方面的不灵活性的限制。与可以在训练过程中自适应地在线调整的连续超参数（例如学习率和权重衰减）不同，LoRA 排序 r 采用离散值 - 其变化将直接改变模型结构。排序的最佳选择可能因不同的主干模型和下游任务而异。保守地选择较大的排序 r 会浪费训练时间和计算资源，而逐渐将 r 设置为很小可能会降低模型性能并导致从头开始重新训练。这些限制凸显了使用自适应排序选择插件升级 LoRA 的重要性。

近年来，提出了多种方法以实现 LoRA 秩的灵活调整。例如，与固定秩不同，Valipour 等人（Valipour et al., 2022）提出了 DyLoRA，其中在一系列秩选择上引入了一个预定义的离散分布 $p_B(\cdot )$。这种方法与嵌套 Dropout（Rippel et al., 2014）相关但有所不同，可被视为一种对不同秩的 LoRA 模块的混合模型进行优化的方式。

然而，直接且确定性地调整 LoRA 秩似乎是一种更具吸引力的方法。为设计这种方法，我们首先从矩阵的秩与奇异值分解（SVD）之间的关系中获得了重要的启示。将 LoRA 中可调整的增量权重矩阵记为 $\Delta :={\mathbf{W}}_u{\mathbf{W}}_d$，我们可以将其 SVD 表示为：

$${\mathbf{\Delta }}_{p\times q}={\mathbf{U}}_{p\times p}{\mathbf{\Sigma }}_{p\times q}{\mathbf{V}}_{q\times q}^{\top }$$

其中 $\mathbf{U}$ 和 $\mathbf{V}$ 分别是正交矩阵，$\mathbf{\Sigma }$ 是一个（矩形）对角矩阵，其对角元素为 $\Delta$ 的奇异值： σ ( Δ ) = { σ 1 ≥ σ 2 ≥ ⋯ ≥ σ min ⁡ { p , q } ≥ 0 } \sigma(\Delta) = \left\{\sigma_1 \geq \sigma_2 \geq \cdots \geq \sigma_{\min \{p, q\}} \geq 0\right\} σ(Δ)={σ1​≥σ2​≥⋯≥σmin{p,q}​≥0}。为了方便表示，我们将 $\Sigma$ 的对角线元素重塑为一个列向量：

g : = ( σ 1 , σ 2 , ⋯   , σ min ⁡ { p , q } ) ⊤ \mathbf{g} := \left(\sigma_1, \sigma_2, \cdots, \sigma_{\min \{p, q\}}\right)^{\top} g:=(σ1​,σ2​,⋯,σmin{p,q}​)⊤

令 d = min ⁡ { p , q } d=\min \{p, q\} d=min{p,q}，我们可以将 LoRA 的前向传播公式重新表述为：

$$\mathbf{z}\leftarrow \mathbf{\Delta x}={\mathbf{U}}_{\cdot ,1:d}\left(\mathbf{g}\odot {\mathbf{V}}_{\cdot ,1:d}^{\top }\mathbf{x}\right)$$

其中，$\odot$ 表示逐元素点积（Hadamard 乘积）。注意，$\mathrm{rank}(\mathbf{\Delta })=\Vert \mathbf{g}{\Vert }_0$，即 $\mathbf{g}$ 的 ${\ell }_0$ 范数。因此，调整 LoRA 的秩足以控制向量 $\mathbf{g}$ 的稀疏性。

Zhang 等人基于这一 SVD 方法提出了 AdaLoRA 方法（Zhang et al., 2023）。在 AdaLoRA 中，向量 $\mathbf{g}$ 中的元素经过调整，使得非零项的数量小于预定义的预算 $b$。具体来说，他们仅保留重要性评分（由权重-梯度乘积启发性地构造的一种新度量）排名前 $b$ 的条目。由于负值 ${\mathbf{g}}_i$ 可以通过翻转 ${\mathbf{u}}_i$ 或 ${\mathbf{v}}_i$ 的符号转换为正值，因此合理地去掉了 $\mathbf{g}$ 条目非负性的限制。

此外，他们将约束优化问题转化为其无约束版本，通过将正交性条件 ${\mathbf{U}}^{\top }\mathbf{U}={\mathbf{I}}_p$ 和 ${\mathbf{V}}^{\top }\mathbf{V}={\mathbf{I}}_q$ 替换为以下正则化项：

$$R(\mathbf{U},\mathbf{V})={\Vert {\mathbf{U}}^{\top }\mathbf{U}-{\mathbf{I}}_p\Vert }_F^2+{\Vert {\mathbf{V}}^{\top }\mathbf{V}-{\mathbf{I}}_q\Vert }_F^2$$

尽管实验证明了其有效性，但 AdaLoRA 中仍然存在两个问题，需要重新思考方法论并等待进一步改进。首先，AdaLoRA 中的稀疏性选择标准基于他们新提出的重要性得分，该得分依赖于权重梯度乘积的移动平均值。尽管它在实证研究中是有效的，但该标准在很大程度上是启发式的，缺乏理论动机。其次，重要性得分的移动平均运算和正交正则化的梯度 (5) 都会增加额外的计算成本。与具有上述局限性的 AdaLoRA 相比，我们的方法 SoRA 是一种改进，具有高度简化的更新规则，并以稀疏正则化和近端梯度方法的理论为后盾。SoRA 的详细方法将在下一节中详细阐述。

3 Our Approach

我们的方法的核心思想是稀疏低秩自适应 (SoRA)，即使用近端梯度法训练的稀疏门控单元在训练过程中动态调整固有秩。SoRA 采用之前介绍的低秩分解框架，因为它的有效性和参数效率已得到广泛验证。

3.1 Sparse Low-rank Adaptation

模块结构

在构建 SoRA 模块时，我们首先根据实际需求或研究考虑预定义一个最大可接受秩 $r_{\text{max}}$。然后，每个 SoRA 模块会从 LoRA 继承两个矩阵 ${\mathbf{W}}_d\in {\mathbb{R}}^{r_{\text{max}}\times q}$ 和 ${\mathbf{W}}_u\in {\mathbb{R}}^{p\times r_{\text{max}}}$，分别用于降维投影和升维投影。最大秩 $r_{\text{max}}$ 被设定为相对较大的值，但我们将在接下来的段落中展示如何通过稀疏化机制高效控制它。实际上，这通过在投影矩阵之间引入一个门控单元 $g\in {\mathbb{R}}^{r_{\text{max}}}$ 实现，其公式类似于 SVD 的形式。SoRA 模块的前向传播过程如下：

$$\begin{array}{rl}& \mathbf{h}\stackrel{\text{降维投影}}{\leftarrow }{\mathbf{W}}_d\mathbf{x}\\ & {\mathbf{h}}^{\prime }\stackrel{\text{门控操作}}{\Leftarrow }\mathbf{g}\odot \mathbf{h}\\ & \mathbf{z}\stackrel{\text{升维投影}}{\leftarrow }{\mathbf{W}}_u{\mathbf{h}}^{\prime }\end{array}$$

更简洁地表示为：

$$\mathbf{z}\leftarrow {\mathbf{W}}_u\left(\mathbf{g}\odot \left({\mathbf{W}}_d\mathbf{x}\right)\right)$$

优化

我们使用与 LoRA 相同的随机梯度方法优化降维投影和升维投影矩阵，而门控单元 $\mathbf{g}$ 采用一种促进稀疏性的不同优化方式：

$${\mathbf{g}}_{t+1}\leftarrow {\mathcal{T}}_{{\eta }_t\cdot \lambda }\left({\mathbf{g}}_t-{\eta }_t{\nabla }_{\mathbf{g}}{\mathcal{L}}_0\left({\mathbf{\Delta }}_t\right)\right)$$

其中：

• ${\mathcal{L}}_0(\cdot )$ 是语言模型的原始损失函数，
• $\Delta$ 表示完整的可调参数（包括门控单元），
• ${\eta }_t>0$ 是第 $t$ 次迭代的步长，
• $\lambda >0$ 是促进稀疏性的正则化强度超参数。

此外，表达式中的 ${\mathcal{T}}_{{\eta }_t\cdot \lambda }(\cdot )$ 表示以下软阈值函数的逐元素广播：

$${\mathcal{T}}_{\xi }(x):=\{\begin{array}{l}x-\xi ,x>\xi \\ 0,-\xi \le x\le \xi \\ x+\xi ,x<-\xi \end{array}$$

其中 $\xi ={\eta }_t\cdot \lambda$ 是阈值。在实际操作中，(10) 中的真实梯度 ${\nabla }_{\mathbf{g}}{\mathcal{L}}_0$ 被其小批量随机近似所代替。

后剪枝 (Post-pruning)

当训练完成后，我们对 SoRA 权重进行进一步剪枝，以删除为零的秩并将模块还原为 LoRA 形式。具体来说，对于第 $k$ 个 SoRA 模块，令：

$${\mathcal{I}}^{(k)}=\left\{i\in \left[1:r_{\max }\right]\mid {\mathbf{g}}_i^{(k)}=0\right\}$$

表示第 $k$ 个门控向量 ${\mathbf{g}}^{(k)}$ 中零条目的索引。我们对第 $k$ 个模块进行以下处理：

1. 删除降维投影矩阵 ${\mathbf{W}}_d^{(k)}$ 中索引为 ${\mathcal{I}}^{(k)}$ 的行，得到 ${\tilde{\mathbf{W}}}_d^{(k)}$；
2. 删除升维投影矩阵 ${\mathbf{W}}_u^{(k)}$ 中索引为 ${\mathcal{I}}^{(k)}$ 的列，得到 ${\tilde{\mathbf{W}}}_u^{(k)}$；
3. 删除门控向量 ${\mathbf{g}}^{(k)}$ 中索引为 ${\mathcal{I}}^{(k)}$ 的条目，得到 ${\tilde{\mathbf{g}}}^{(k)}$。

通过上述步骤，在推理阶段，第 $k$ 个 SoRA 模块将作为一个通常的 LoRA 模块运行，其秩为 $r_{\text{max}}-\left|{\mathcal{I}}^{(k)}\right|$。此时，降维投影矩阵为 ${\tilde{\mathbf{W}}}_d^{(k)}$，升维投影矩阵为 ${\tilde{\mathbf{W}}}_u^{(k)}\cdot \mathrm{diag}\left({\tilde{\mathbf{g}}}^{(k)}\right)$。

3.2 Interpretation and Comparison

理论解释

更新规则 (10) 实际上是 ${\ell }_1$ 损失的 近端梯度方法 的应用（Chambolle et al., 1998; Beck and Teboulle, 2009）。这一点可以通过将 (10) 等价地重写为以下公式直接得出：

$$\begin{array}{rl}{\mathbf{g}}_{t+1}& \leftarrow \underset{\mathbf{g}}{\mathrm{argmin}}{\eta }_t\cdot \lambda \Vert \mathbf{g}{\Vert }_1\\ & +\frac{1}{2}{\Vert \mathbf{g}-\left({\mathbf{g}}_t-{\eta }_t\nabla {\mathcal{L}}_0\left({\mathbf{g}}_t\right)\right)\Vert }_2^2\end{array}$$

上式 (13) 正是以下 带 ${\ell }_1$ 正则化损失函数 的近端梯度更新：

$$\mathcal{L}(\mathbf{\Delta }):={\mathcal{L}}_0(\mathbf{\Delta })+\lambda \sum _{k=1}^K{\Vert {\mathbf{g}}^{(k)}\Vert }_1$$

其中，${\mathbf{g}}^{(k)}$ 表示第 $k$ 个 SoRA 模块的门控向量。该促进稀疏性的策略可以追溯到 LASSO 回归估计器（Tibshirani, 1996）和 压缩感知（Candes et al., 2006），并已被许多深度学习领域的工作所采用（Wen et al., 2016; Scardapane et al., 2017）。

与 AdaLoRA 进行比较。我们的方法 SoRA 受到 SVD 分解的启发，与之前的 AdaLoRA (Zhang et al, 2023) 有以下不同之处。首先，我们不应用 AdaLoRA 中使用的正交性正则化 (5)。原因是，对于秩选择目的，稀疏门 g 就足够了。 坚持 SVD 的原始要求可能会导致额外的计算开销。其次，AdaLoRA 中的移动平均重要性得分可作为相应条目归零时损失变化的近似值，这被视为参数“敏感性”的启发式测量。然而，模型对某个参数的时间敏感性并不意味着应该保留该参数，因为没有严格的理论来做到这一点。相比之下，我们基于软阈值操作 (10) 的秩选择以更清晰的形式进行，并且由近端梯度迭代理论充分证明。如本节前面所述，SoRA 模块的更新规则通过最小化正则化损失目标 (14)，完全遵循插值-复杂性权衡的第一原则。 除了形式简单性和理论清晰性之外，SoRA 还以更少的参数和更短的时钟时间实现了卓越的实验性能，这将在第 4 节中介绍。

图 1：稀疏低秩自适应 (SoRA) 的示意图。 训练阶段：门控单元 $g$ 控制 ${\mathbf{W}}_d$ 和 ${\mathbf{W}}_u$ 的稀疏性。 推理阶段：根据 $g$ 中零条目的索引，将 ${\mathbf{W}}_d$ 和 ${\mathbf{W}}_u$ 中的零向量移除。