【大模型上下文长度扩展】LongLoRA：长序列大模型微调新方式

 

---

论文：https://arxiv.org/pdf/2309.12307.pdf

代码：https://github.com/dvlab-research/LongLoRA

 

核心问题

核心问题是如何有效地扩展大型语言模型（LLMs）的上下文窗口大小，以便处理长文本输入。

这一挑战主要由于标准的自注意力机制在处理大量上下文时计算成本高昂所导致。

子问题1: 上下文窗口限制

• 背景: 大型语言模型如LLaMA和Llama2分别被训练以处理2048和4096个令牌的上下文。
• 这种预定义的大小限制了模型在处理长文档摘要或回答长问题等应用中的能力。
• 解法: LongLoRA通过引入效率更高的微调方法来扩展预训练LLMs的上下文窗口。
• 例子: 若以Llama2为例，通过LongLoRA方法，可以在单个8×A100机器上将7B模型的上下文扩展到100k，或将70B模型扩展到32k。

子问题2: 计算资源限制

• 背景: 从头训练或微调LLMs以适应更长的上下文需要大量的计算资源，例如，使用32个A100 GPUs扩展LLaMA模型的上下文从2k到8k，这对普通研究者来说通常是不可承受的。
• 解法: LongLoRA通过利用低秩矩阵更新（LoRA）和短程注意力机制（S2-Attn）来减少所需的计算资源。
• 例子: 使用S2-Attn可以通过将上下文长度分为几个组，并在每个组内单独进行注意力操作，从而有效减少计算成本。

子问题3: 高效微调方法的缺乏

• 背景: 传统的低秩适应（LoRA）方法在扩展上下文时既不够有效也不够高效，导致长上下文模型的困惑度高。
• 解法: LongLoRA结合了低秩权重更新和S2-Attn，后者通过分组和令牌偏移来实现高效的长上下文训练。
• 例子: 在Llama2模型上应用LongLoRA，可以在保留原始注意力架构的同时，通过S2-Attn实现对长上下文的有效近似。

低秩权重更新（Low-rank Adaptation, LoRA）和S2-Attn（Shifted Sparse Attention）是两种技术，它们在提高大型语言模型（LLMs）处理长上下文时的效率和效果方面起着关键作用。

低秩权重更新（LoRA）

低秩权重更新是一种模型微调技术，旨在优化模型的参数更新过程，从而减少所需的计算资源。

在大型语言模型中，LoRA的核心思想是对模型中的线性投影层（通常存在于自注意力机制中）进行修改，而不是直接修改整个模型参数。

这种修改涉及到使用低秩矩阵来近似原始的高维权重矩阵。

• 工作原理: LoRA通过将原始的权重矩阵分解为两个较小的矩阵的乘积，从而实现对模型的低秩更新。
• 这两个较小的矩阵乘积的结果能够近似原始的高维权重矩阵，但参数数量大大减少，从而减轻了计算和存储的负担。
• 优点: LoRA通过减少微调过程中需要优化的参数数量，显著降低了计算成本，同时仍然保持或甚至提高了模型性能。

S2-Attn（Shifted Sparse Attention）

S2-Attn是一种自注意力机制的变体，旨在通过对输入序列的高效处理来扩展模型的上下文窗口。

它通过将长序列分组，并在这些组内以及组之间应用稀疏注意力模式来实现这一点。

• 工作原理: 在S2-Attn中，序列被分割成多个固定大小的组。在每组内部，模型执行标准的自注意力计算。
• 为了确保信息能够在相邻组之间流动，一半的注意力头会对令牌进行偏移处理，即将它们在序列中的位置向前或向后移动半个组的大小。
• 这种偏移允许模型捕获更长范围内的依赖关系，同时减少了计算量。
• 优点: S2-Attn通过减少需要计算的注意力对，显著降低了处理长序列时的计算复杂度。
• 这种方法特别适合于那些需要模型理解和生成长文本的应用场景。

结合使用LoRA和S2-Attn的优势在于，它们共同提供了一种高效且有效的方式来扩展大型语言模型的上下文处理能力。

LoRA通过低秩更新减少了模型微调的计算成本，而S2-Attn通过改进的注意力机制有效处理长序列，这两种技术的结合使得在资源受限的情况下也能实现对长上下文的支持。

这种方法不仅提高了模型处理长文本的能力，还保持了计算上的可行性，提供了更多的灵活性和效率。

分析不足

尽管LongLoRA为扩展LLMs的上下文窗口提供了一种有效且高效的方法，但存在几个潜在的限制和考虑因素：

• 泛化能力: LongLoRA的效果如何在不同类型的任务和数据集上泛化还需进一步验证。
• 复杂度与效率的平衡: 虽然S2-Attn减少了计算成本，但其对模型复杂度的影响，以及在极大上下文长度下的性能表现，需要详细的实验分析。
• 与其他技术的兼容性: LongLoRA与其他优化技术和架构的兼容性，如Flash-Attention2，虽然在文中提到，但具体的实现细节和性能影响也值得进一步探索。

LongLoRA为扩展LLMs的上下文窗口提供了一条可行之路，但每个解决方案都应当在不同的场景下被细致评估，以确保最佳的性能和效率。

 

---

扩展大模型处理长上下文能力不同方法

子问题1: 处理长上下文的高计算成本

• 背景: 传统的全注意力（full attention）机制在处理长上下文时，面临计算成本呈二次方增长的问题。
• 解法: 使用稀疏注意力机制，如Longformer和BigBird，通过限制注意力的范围来减少计算量。
• 例子: Longformer设计了一种稀疏注意力模式，只计算序列中特定位置的注意力分数，有效降低了处理长序列的计算复杂度。

子问题2: 预训练LLMs微调的不可行性

• 背景: 许多扩展上下文方法通过对模型架构进行较大改动来实现，这使得在已有预训练模型上进行微调变得不可行。
• 解法: 通过S2-Attn等方法，提出在微调阶段使用近似但形状相似的注意力机制，以便在推理阶段保持全注意力架构。
• 例子: S2-Attn通过分组和令牌偏移实现高效处理长上下文，同时在模型推理时仍使用原始的全注意力机制，保证了模型的灵活性和效果。

子问题3: 位置嵌入的长上下文适应性

• 背景: LLMs通常采用预定义的上下文长度进行预训练，对于超出此范围的上下文处理存在局限。
• 解法: 通过位置插值和其他位置嵌入修改方法，如NTK-aware、Yarn等，来扩展模型对长上下文的适应性。
• 例子: 位置插值通过修改旋转位置编码，扩展了LLaMA模型的上下文长度至32768，从而使模型能够处理更长的输入序列。

子问题4: 参数高效微调的需求

• 背景: 给定的计算资源和存储限制要求微调方法不仅要有效，还要参数高效。
• 解法: 采用LoRA等参数高效微调方法，通过局部更新模型的小部分，如输入嵌入层或特定的权重，来实现长上下文的扩展。
• 例子: LoRA通过在自注意力块中使用低秩矩阵更新来微调模型，减少了所需的参数数量和计算资源。

分析不足

虽然上述方法在扩展LLMs处理长上下文的能力方面取得了进展，但每种技术都有其局限性。

例如，稀疏注意力机制虽然减少了计算成本，但可能牺牲了模型对于上下文中某些关键信息的捕捉能力。

位置嵌入方法的扩展能力虽好，但可能需要对模型架构进行较大的修改，影响模型的通用性。

参数高效微调技术虽然在资源受限的情况下非常有用，但可能需要仔细选择微调的层次，以确保模型性能不受影响。

此外，当前的研究主要集中在模型架构和训练方法的改进上，较少考虑到数据侧的优化，如通过更智能的数据预处理和选择机制来减轻长上下文处理的负担。

可以探索结合模型和数据两方面的优化，以进一步提高处理长上下文的效率和效果。

 

---

LongLoRA方法

子问题1: 高计算成本和内存需求

• 背景: 标准的自注意力机制在处理长序列时，由于计算复杂度和内存需求与序列长度的二次方成正比，导致了高计算成本和内存需求。
• 解法: Shifted Sparse Attention (S2-Attn)。
• 特征: S2-Attn通过将输入序列分组并在每个组内进行自注意力计算，以及通过在一半的注意力头中对分组进行偏移，实现了计算成本的大幅度降低，同时保持了不同组之间信息的流动。
• 例子: 在处理8192个令牌的输入时，通过将自注意力限制在2048大小的各个组内，并在一半的注意力头中将分组偏移1024个令牌，有效降低了计算复杂度，同时通过偏移实现组间信息交流。

LongLoRA技术的示意概览。它包括两个子图：

• (a) 移位稀疏注意力（S^2-Attn）：展示了如何将注意力划分为两种模式——无移位（模式1）和有移位（模式2）——以及这些模式如何结合以允许不同令牌组之间的通信。
• (b) 低秩适应：展示了LongLoRA如何适应Transformer模型中的注意力权重。
• 虽然注意力层中的LoRA权重是可训练的，嵌入层和归一化层也被设为可训练，这对于扩展模型处理的上下文至关重要。

此图提供了S^2-Attn工作方式的逐步视觉解释。分为三个步骤：

1. 拆分：特征沿头部维度被拆分为两个块。
2. 移位：其中一个块中的令牌被移动半个组大小。
3. 分组和注意力：令牌被分组并重新塑形，并在这些组内计算注意力。
4. 该图展示了两种注意力模式：一种没有移位和一种有移位，指出如何通过这种机制实现组间信息流动。

子问题2: 微调与长上下文适应性的差距

• 背景: 通过低秩适应（LoRA）微调预训练模型时，随着目标上下文长度的增加，其与全面微调之间存在明显的性能差距。
• 解法: LoRA+（改进的LoRA方法）。
• 特征: LoRA+通过在训练过程中开放嵌入层和归一化层的参数，解决了仅通过低秩权重更新无法有效适应长上下文的问题。
• 例子: 在Llama2 7B模型上，尽管归一化层参数仅占模型总参数的0.004%，通过训练这些层，LoRA+显著缩小了与全面微调之间的性能差距。

此图表比较了三种不同的方法：完全微调（Full FT）、传统的LoRA以及LongLoRA，它们基于在不同上下文大小下的困惑度、GPU内存使用和训练时长。

• 困惑度图表：显示了随着上下文大小增加，困惑度（衡量模型性能的指标，数值越低越好）的变化。
• LongLoRA似乎在性能和资源使用之间提供了一种平衡，其困惑度低于LoRA，并且与全微调相当。
• GPU内存图表：说明了使用每种方法进行训练的内存成本。
• LongLoRA比全微调使用的内存要少得多，表明它更为高效。
• 训练时长图表：描述了使用每种方法训练模型所需的时间。
• LongLoRA减少了与全微调相比的训练时间，突显了其效率。

分析不足

虽然LongLoRA通过S2-Attn和LoRA+提供了处理长上下文的有效策略，但仍有一些潜在的限制和改进空间：

• 泛化能力: LongLoRA方法在不同类型和规模的LLMs上的泛化能力需要进一步研究和验证。
• 长序列处理的极限: 尽管S2-Attn在减少计算成本方面非常有效，但其在处理极长序列时的效率和效果如何，特别是在超过模型原设计上下文长度很多倍的情况下，仍需进一步探索。
• 优化与调整: LoRA+虽然提高了长上下文微调的效果，但如何选择最优的层（嵌入层和归一化层）参数，以及这些参数对不同任务影响的深入理解，可能会进一步提高微调效率和模型性能。

总结

问题：如何提高大型语言模型（LLMs）处理长上下文的能力，同时保持高效率和精确度？

子问题1: 高计算成本

• 背景: 标准的自注意力机制在处理长序列时，会因为计算复杂度随序列长度平方增长而导致计算资源消耗巨大。
• 解法: 移位稀疏注意力（S²-Attn）。
• 特征: S²-Attn通过将注意力分布在不同的组中，并在组内进行自注意力计算，减少了计算量。
• 通过移位一半的注意力头部，实现了组间的信息交流，保持了上下文的连贯性，同时减少了计算成本。
• 例子: 假设有一个包含8192个令牌的序列，通过S²-Attn，我们将其分为四个2048令牌的组，并且在一半的头部中进行移位，使每个组能够与相邻组交换信息，而不是单独计算。

子问题2: 微调效率低

• 背景: 全参数微调（Full Fine-Tuning）需要大量的计算资源，这对于资源受限的研究者来说是不切实际的。
• 解法: 改进的低秩适应（LoRA+）。
• 特征: LoRA+通过在微调过程中只更新一小部分参数，减少了参数的数量并降低了训练成本。
• 特别是，它通过对嵌入层和归一化层的参数进行训练，提高了对长上下文的适应性。
• 例子: 对于一个7B参数的LLM，LoRA+能够在不牺牲性能的前提下，只微调嵌入层和归一化层参数，从而实现对长达32768个令牌上下文的处理。

全流程分析不足

虽然LongLoRA通过S²-Attn和LoRA+有效提高了LLMs处理长上下文的能力和效率，但可能在极端长上下文或特定任务上的表现仍有待验证。

此外，对于全新的上下文结构，这些方法可能需要进一步的调整或优化。

当前的方法可能还未充分考虑到模型的泛化能力，即在不同类型的数据或任务上的表现。