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

基于语言模型行列式点过程的高效上下文学习演示标注

Effective Demonstration Annotation for In-Context Learning via Language Model-Based Determinantal Point Process

Zhejiang University(浙江大学);Alibaba Group(阿里巴巴集团);ShanghaiTech University(上海科技大学)

Peng Wang, Xiaobin Wang, Chao Lou, Shengyu Mao, Pengjun Xie, Yong Jiang

AI总结:

本文提出LM-DPP方法,在上下文学习的演示检索前通过同时考虑未标注实例的不确定性与多样性进行选择性标注,实验证明其能在多个语言模型和数据集上有效选出典型示例,使LLMs从低不确定性且高多样性的子集中获益最大。

摘要:

上下文学习(ICL)是一种少样本学习范式,通过输入-输出对学习映射关系,并将其适当地应用于新实例。尽管大型语言模型(LLMs)展现了卓越的ICL能力,但现有工作高度依赖大规模标注支持集,这在实际场景中并不总是可行。为了改进这一方法,我们主要聚焦于一种创新的选择性标注机制,该机制先于标准的演示检索。我们提出了基于语言模型的行列式点过程(LM-DPP),它同时考虑未标注实例的不确定性和多样性,以实现最优选择。由此产生一个用于标注的子集,在两者之间取得权衡。我们将LM-DPP应用于多种语言模型,包括GPT-J、LlaMA和GPT-3。在9个自然语言理解(NLU)数据集和2个生成数据集上的实验结果表明,LM-DPP能够有效选择典型示例。进一步分析表明,大型语言模型从低不确定性且高多样性的子集中获益最为显著。

英文摘要:

In-context learning (ICL) is a few-shot learning paradigm that involves learning mappings through input-output pairs and appropriately applying them to new instances. Despite the remarkable ICL capabilities demonstrated by Large Language Models (LLMs), existing works are highly dependent on large-scale labeled support sets, not always feasible in practical scenarios. To refine this approach, we focus primarily on an innovative selective annotation mechanism, which precedes the standard demonstration retrieval. We introduce the Language Model-based Determinant Point Process (LM-DPP) that simultaneously considers the uncertainty and diversity of unlabeled instances for optimal selection. Consequently, this yields a subset for annotation that strikes a trade-off between the two factors. We apply LM-DPP to various language models, including GPT-J, LlaMA, and GPT-3. Experimental results on 9 NLU and 2 Generation datasets demonstrate that LM-DPP can effectively select canonical examples. Further analysis reveals that LLMs benefit most significantly from subsets that are both low uncertainty and high diversity.

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

↑