提出任务特定方向新方法,显著提升高效微调性能
Task-Specific Directions: Definition, Exploration, and Utilization in Parameter Efficient Fine-Tuning

- 定义任务特定方向(TSD),指导高效微调
- LoRA-Dash 和 LoRA-Init 提升模型性能,最高达+4.2%
- 适合关注参数高效微调的从业者和研究者
大语言模型在下游任务中表现优异,但全参数微调消耗大量资源。为缓解此问题,提出了参数高效微调(PEFT)策略,如LoRA。本文深入探讨任务特定方向(TSD),这是将大模型从预训练状态转向任务特化的核心。我们提出框架以明确定义TSD并探索其性质与应用挑战。进而提出新方法LoRA-Dash,旨在最大化TSD在微调中的影响,提升目标任务表现。此外,针对LoRA初始化这一关键问题,我们提出LoRA-Init:基于TSD识别微调中最需调整的方向,并以此初始化LoRA矩阵,显著提升性能。结合两者形成基于TSD的最终版本LoRA-TSD。大量实验验证了方法有效性,深入分析揭示其成功机制。
原文摘要 · Abstract (English)
Large language models demonstrate impressive performance on downstream tasks, yet they require extensive resource consumption when fully fine-tuning all parameters. To mitigate this, Parameter Efficient Fine-Tuning (PEFT) strategies, such as LoRA, have been developed. In this paper, we delve into the concept of task-specific directions (TSDs), which are critical for transitioning large models from pretrained states to task-specific enhancements in PEFT. We propose a framework to clearly define these directions and explore their properties and practical utilization challenges. We then introduce a novel approach, LoRA-Dash, which aims to maximize the impact of TSDs during the fine-tuning process, thereby enhancing model performance on targeted tasks. Additionally, based on our exploration of TSD, we focus on an important issue in PEFT: the initialization of LoRA. While some works have pointed out the significance of initialization for LoRA's performance and proposed various strategies, these methods are often empirical and not task-specific. To address this issue, we propose LoRA-Init. Starting from TSD, we identify the directions that require the most adjustment during fine-tuning for downstream tasks. By initializing the matrices in LoRA with these directions, LoRA-Init significantly enhances LoRA's performance. Moreover, we can combine LoRA-Dash and LoRA-Init to create the final version of LoRA based on TSDs, which we refer to as LoRA-TSD. Extensive experiments have conclusively demonstrated the effectiveness of these methods, and in-depth analyses further reveal the underlying mechanisms behind their success.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。