改进前缀调优,让模型更高效地适应新任务。
PrefixMemory-Tuning: Modernizing Prefix-Tuning by Decoupling the Prefix from Attention
- 将前缀模块从注意力头中分离,提升表达能力
- 在多个基准上表现优于传统前缀调优方法
- 适合想用少量参数微调大模型的研究者
参数高效微调(PEFT)已成为快速适配大语言模型(LLMs)到下游任务的关键技术。前缀调优作为早期且有效的PEFT方法,能在显著降低计算与内存开销的前提下达到接近全量微调的性能。然而,尽管早期成功,其在现代先进大模型上的效果仍非常有限。本文通过实证发现,前缀调优在大模型上表现不佳的根本原因在于输入提示与可学习前缀在注意力头中存在固有权衡。为此,我们提出PrefixMemory-Tuning,通过将前缀模块从注意力头中移出并增强其表达能力,从而扩展前缀调优原理并克服其缺陷。实验表明,该方法在多种基准上均持续优于现有前缀调优方法,并在多个通用基准上实现与现代PEFT方法相当的性能,表明前缀调优有望发展为最先进的微调范式。研究结果表明,通过突破其内在限制,前缀调优仍可在参数高效模型适配领域保持竞争力。
原文摘要 · Abstract (English)
Parameter-Efficient Fine-Tuning (PEFT) methods have become crucial for rapidly adapting large language models (LLMs) to downstream tasks. Prefix-Tuning, an early and effective PEFT technique, demonstrated the ability to achieve performance comparable to full fine-tuning with significantly reduced computational and memory overhead. However, despite its earlier success, its effectiveness in training modern state-of-the-art LLMs has been very limited. In this work, we demonstrate empirically that prefix-tuning underperforms on LLMs because of an inherent tradeoff between the contribution of the input prompt and the parameterized prefix within the attention head. This motivates us to introduce PrefixMemory-Tuning, an architecture that generalizes the principles of prefix-tuning while addressing its shortcomings by shifting the prefix module out of the attention head itself and improving its expressiveness. Our experiments show that, across diverse benchmarks, PrefixMemory-Tuning consistently outperforms existing prefix-tuning methods. Notably, it achieves competitive performance with modern PEFTs on several general benchmarks, highlighting a potential extension of prefix-tuning approaches to become state-of-the-art. Our findings suggest that by overcoming its inherent limitations, prefix-tuning can remain a competitive and relevant research direction in the landscape of parameter-efficient LLM adaptation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。