新方法以块级连接实现高效微调,参数量减少超99%。
Solo Connection: A Parameter Efficient Fine-Tuning Technique for Transformers
- 通过块间长连接替代权重矩阵修改,实现参数高效微调
- 在端到端生成任务上优于LoRA,参数量比LoRA少59%
- 基于同伦理论设计平滑适应机制,适合大模型微调场景
参数高效微调(PEFT)是适配大语言模型(LLM)应对新任务的通用方法。主流方法低秩适配(LoRA)主要调整GPT2解码器块内的注意力权重矩阵。本文提出全新方法Solo Connection,改在解码器块级别适配表示,而非修改单个权重矩阵。Solo Connection不仅在端到端自然语言生成基准上超越LoRA,还相较LoRA减少59%可训练参数,相比完整微调GPT2减少超过99%。该方法受同伦理论启发:引入可训练线性变换,渐进地在零向量与任务特异性表示之间插值,实现稳定平滑的适应过程。原始12层GPT2中跳连通常局限于单个解码器块,但后续变体扩展至48层,更大模型可达128层以上,凸显重思微调中跳连应用的必要性。本文聚焦跨解码器块的长跳连,可能增强模型适应新任务能力并利用预训练知识。
原文摘要 · Abstract (English)
Parameter efficient fine tuning (PEFT) is a versatile and extensible approach for adapting a Large Language Model (LLM) for newer tasks. One of the most prominent PEFT approaches, Low Rank Adaptation (LoRA), primarily focuses on adjusting the attention weight matrices within individual decoder blocks of a Generative Pre trained Transformer (GPT2). In contrast, we introduce Solo Connection a novel method that adapts the representation at the decoder-block level rather than modifying individual weight matrices. Not only does Solo Connection outperform LoRA on E2E natural language generation benchmarks, but it also reduces the number of trainable parameters by 59% relative to LoRA and by more than 99% compared to full fine-tuning of GPT2, an early version of Large Language Models (LLMs). Solo Connection is also motivated by homotopy theory: we introduce a trainable linear transformation that gradually interpolates between a zero vector and the task-specific representation, enabling smooth and stable adaptation over time. While skip connections in the original 12 layer GPT2 are typically confined to individual decoder blocks, subsequent GPT2 variants scale up to 48 layers, and even larger language models can include 128 or more decoder blocks. These expanded architectures underscore the need to revisit how skip connections are employed during fine-tuning. This paper focuses on long skip connections that link outputs of different decoder blocks, potentially enhancing the model's ability to adapt to new tasks while leveraging pre-trained knowledge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。