arXiv:2410.02200cs.LG2024-10ICLR被引 16

重审前缀提示:共享结构提升参数效率

Revisiting Prefix-tuning: Statistical Benefits of Reparameterization among Prompts

  • 通过重参数化显式构建前缀键值向量的共享结构
  • 共享结构使参数估计样本效率显著提升
  • 适用于理解提示调优机制的研究者与工程师

提示调优类方法(如前缀提示、提示调优)因其在微调大模型时的高效性而广受关注。尽管应用广泛,其理论基础仍不充分。研究发现,前缀提示性能接近全参数微调的关键在于重参数化策略。本文揭示该策略不仅为工程技巧,更具有深层理论依据:它隐式编码了前缀键与值向量间的共享结构。结合前缀提示与专家混合模型(MoE)的最新关联,进一步表明此共享结构相比非共享设计能显著提升参数估计的样本效率。跨视觉与语言任务的大量实验证实,共享结构增强了前缀提示在多样化任务中的有效性。此外,类似结构优势亦存在于提示调优中,为理解其成功提供了新视角。本研究从理论与实证层面推进了对提示调优方法及其内在机制的认知。

原文摘要 · Abstract (English)

Prompt-based techniques, such as prompt-tuning and prefix-tuning, have gained prominence for their efficiency in fine-tuning large pre-trained models. Despite their widespread adoption, the theoretical foundations of these methods remain limited. For instance, in prefix-tuning, we observe that a key factor in achieving performance parity with full fine-tuning lies in the reparameterization strategy. However, the theoretical principles underpinning the effectiveness of this approach have yet to be thoroughly examined. Our study demonstrates that reparameterization is not merely an engineering trick but is grounded in deep theoretical foundations. Specifically, we show that the reparameterization strategy implicitly encodes a shared structure between prefix key and value vectors. Building on recent insights into the connection between prefix-tuning and mixture of experts models, we further illustrate that this shared structure significantly improves sample efficiency in parameter estimation compared to non-shared alternatives. The effectiveness of prefix-tuning across diverse tasks is empirically confirmed to be enhanced by the shared structure, through extensive experiments in both visual and language domains. Additionally, we uncover similar structural benefits in prompt-tuning, offering new perspectives on its success. Our findings provide theoretical and empirical contributions, advancing the understanding of prompt-based methods and their underlying mechanisms.

提示调优前缀提示重参数化参数效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。