$Se^2$: Sequential Example Selection for In-Context Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Liu, Haoyu, Liu, Jianfeng, Huang, Shaohan, Zhan, Yuefeng, Sun, Hao, Deng, Weiwei, Wei, Furu, Zhang, Qi
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929375793905664
author Liu, Haoyu
Liu, Jianfeng
Huang, Shaohan
Zhan, Yuefeng
Sun, Hao
Deng, Weiwei
Wei, Furu
Zhang, Qi
author_facet Liu, Haoyu
Liu, Jianfeng
Huang, Shaohan
Zhan, Yuefeng
Sun, Hao
Deng, Weiwei
Wei, Furu
Zhang, Qi
contents The remarkable capability of large language models (LLMs) for in-context learning (ICL) needs to be activated by demonstration examples. Prior work has extensively explored the selection of examples for ICL, predominantly following the "select then organize" paradigm, such approaches often neglect the internal relationships between examples and exist an inconsistency between the training and inference. In this paper, we formulate the problem as a $Se$quential $Se$lection problem and introduce $Se^2$, a sequential-aware method that leverages the LLM's feedback on varying context, aiding in capturing inter-relationships and sequential information among examples, significantly enriching the contextuality and relevance of ICL prompts. Meanwhile, we utilize beam search to seek and construct example sequences, enhancing both quality and diversity. Extensive experiments across 23 NLP tasks from 8 distinct categories illustrate that $Se^2$ markedly surpasses competitive baselines and achieves 42\% relative improvement over random selection. Further in-depth analysis shows the effectiveness of proposed strategies, highlighting $Se^2$'s exceptional stability and adaptability across various scenarios. Code available at https://github.com/microsoft/LMOps.
format Preprint
id arxiv_https___arxiv_org_abs_2402_13874
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle $Se^2$: Sequential Example Selection for In-Context Learning
Liu, Haoyu
Liu, Jianfeng
Huang, Shaohan
Zhan, Yuefeng
Sun, Hao
Deng, Weiwei
Wei, Furu
Zhang, Qi
Computation and Language
The remarkable capability of large language models (LLMs) for in-context learning (ICL) needs to be activated by demonstration examples. Prior work has extensively explored the selection of examples for ICL, predominantly following the "select then organize" paradigm, such approaches often neglect the internal relationships between examples and exist an inconsistency between the training and inference. In this paper, we formulate the problem as a $Se$quential $Se$lection problem and introduce $Se^2$, a sequential-aware method that leverages the LLM's feedback on varying context, aiding in capturing inter-relationships and sequential information among examples, significantly enriching the contextuality and relevance of ICL prompts. Meanwhile, we utilize beam search to seek and construct example sequences, enhancing both quality and diversity. Extensive experiments across 23 NLP tasks from 8 distinct categories illustrate that $Se^2$ markedly surpasses competitive baselines and achieves 42\% relative improvement over random selection. Further in-depth analysis shows the effectiveness of proposed strategies, highlighting $Se^2$'s exceptional stability and adaptability across various scenarios. Code available at https://github.com/microsoft/LMOps.
title $Se^2$: Sequential Example Selection for In-Context Learning
topic Computation and Language
url https://arxiv.org/abs/2402.13874