arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2510.16670cs.CLcs.AIcs.LG
收藏

你只需要一个:使用单个向量的 Capsule Prompt Tuning(胶囊提示调优)

All You Need is One: Capsule Prompt Tuning with a Single Vector

University of Missouri-Kansas City(密苏里大学堪萨斯城分校);U.S. Naval Research Laboratory(美国海军研究实验室);University of North Texas(北卡罗来纳州立大学);Lamar University(拉马尔大学);ByteDance(字节跳动);Case Western Reserve University(凯斯西储大学);University of California, Davis(加州大学戴维斯分校);Rochester Institute of Technology(罗切斯特理工学院);Meta AI

Yiyang Liu, James C. Liang, Heng Fan, Wenhao Yang, Yiming Cui, Xiaotian Han, Lifu Huang, Dongfang Liu, Qifan Wang, Cheng Han

AI总结:

针对提示调优需搜索长度且缺乏实例感知的问题,提出 CaPT,用单个胶囊提示融合任务与实例语义,以极低参数量提升多任务性能。

摘要:

基于提示的学习已成为一种参数高效微调(PEFT)方法,通过用任务感知的引导来条件化生成,从而促进大型语言模型(LLM)对下游任务的适配。尽管取得了成功,但当前基于提示的学习方法严重依赖繁琐的网格搜索来确定最佳提示长度,并且通常需要大量提示,从而带来额外的计算负担。更糟糕的是,我们的开创性发现表明,任务感知提示设计固有地受限于其缺乏实例感知信息,导致其与输入序列之间存在微妙的注意力相互作用。相比之下,只需将实例感知信息作为引导的一部分纳入,即可在无需额外微调的情况下提升经提示调优的模型性能。此外,我们发现了一个有趣的现象,即“注意力锚点”:在序列的最早位置纳入实例感知 token,可以成功保持对关键结构信息的强注意力,并与所有输入 token 表现出更活跃的注意力交互。基于这些观察,我们提出了 Capsule Prompt-Tuning(CaPT,胶囊提示调优),这是一种高效且有效的解决方案,将现成且信息丰富的实例语义用于基于提示的学习。我们的方法以近乎无参数的方式(即一个单一的胶囊提示)创新性地融合了实例感知和任务感知信息。实验结果表明,我们的方法能够在多种语言任务上表现出优越性能(例如,在 T5-Large 上达到 84.03% 的平均准确率),充当“注意力锚点”,同时享有很高的参数效率(例如,在 Llama3.2-1B 上仅占模型参数的 0.003%)。

英文摘要:

Prompt-based learning has emerged as a parameter-efficient finetuning (PEFT) approach to facilitate Large Language Model (LLM) adaptation to downstream tasks by conditioning generation with task-aware guidance. Despite its successes, current prompt-based learning methods heavily rely on laborious grid searching for optimal prompt length and typically require considerable number of prompts, introducing additional computational burden. Worse yet, our pioneer findings indicate that the task-aware prompt design is inherently limited by its absence of instance-aware information, leading to a subtle attention interplay with the input sequence. In contrast, simply incorporating instance-aware information as a part of the guidance can enhance the prompt-tuned model performance without additional fine-tuning. Moreover, we find an interesting phenomenon, namely "attention anchor", that incorporating instance-aware tokens at the earliest position of the sequence can successfully preserve strong attention to critical structural information and exhibit more active attention interaction with all input tokens. In light of our observation, we introduce Capsule Prompt-Tuning (CaPT), an efficient and effective solution that leverages off-the-shelf, informative instance semantics into prompt-based learning. Our approach innovatively integrates both instance-aware and task-aware information in a nearly parameter-free manner (i.e., one single capsule prompt). Empirical results demonstrate that our method can exhibit superior performance across various language tasks (e.g., 84.03\% average accuracy on T5-Large), serving as an "attention anchor," while enjoying high parameter efficiency (e.g., 0.003\% of model parameters on Llama3.2-1B).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑