Saved in:
Bibliographic Details
Main Authors: Wang, Peng, Wang, Xiaobin, Lou, Chao, Mao, Shengyu, Xie, Pengjun, Jiang, Yong
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.02103
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929449455321088
author Wang, Peng
Wang, Xiaobin
Lou, Chao
Mao, Shengyu
Xie, Pengjun
Jiang, Yong
author_facet Wang, Peng
Wang, Xiaobin
Lou, Chao
Mao, Shengyu
Xie, Pengjun
Jiang, Yong
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2408_02103
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Effective Demonstration Annotation for In-Context Learning via Language Model-Based Determinantal Point Process
Wang, Peng
Wang, Xiaobin
Lou, Chao
Mao, Shengyu
Xie, Pengjun
Jiang, Yong
Computation and Language
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.
title Effective Demonstration Annotation for In-Context Learning via Language Model-Based Determinantal Point Process
topic Computation and Language
url https://arxiv.org/abs/2408.02103