paper-with-me

홈 › Papers

Learning to Select In-Context Demonstration Preferred by Large Language Model

2025-05-26 · Zheng Zhang, Shaocheng Lan, Lei Song, Jiang Bian, Yexin Li, Kan Ren

In-context learning (ICL) enables large language models (LLMs) to adapt to new tasks during inference using only a few demonstrations. However, ICL performance is highly dependent on the selection of these demonstrations. Recent work explores retrieval-based methods for selecting query-specific demonstrations, but these approaches often rely on surrogate objectives such as metric learning, failing to directly optimize ICL performance. Consequently, they struggle to identify truly beneficial demonstrations. Moreover, their discriminative retrieval paradigm is ineffective when the candidate pool lacks sufficient high-quality demonstrations. To address these challenges, we propose GenICL, a novel generative preference learning framework that leverages LLM feedback to directly optimize demonstration selection for ICL. Experiments on 19 datasets across 11 task categories demonstrate that GenICL achieves superior performance than existing methods in selecting the most effective demonstrations, leading to better ICL performance.

📄 PDF Abstract BibTeX arXiv:2505.19966

Code (0)

등록된 구현이 없습니다.

Tasks

In-Context LearningLanguage ModelingLanguage ModellingLarge Language ModelMetric LearningRetrieval

Similar Papers 제목 키워드 기반

In-Context Demonstration Selection with Cross Entropy Difference

2023-05-24 · Dan Iter, Reid Pryzant, Ruochen Xu, Shuohang Wang 외

Large language models (LLMs) can use in-context demonstrations to improve performance on zero-shot tasks. However, selecting the best in-context examples is challenging because model performance can vary widely depending…

Language ModelingLanguage ModellingText Generation

Learn to Select: Exploring Label Distribution Divergence for In-Context Demonstration Selection in Text Classification

2025-11-10 · Ye Jiang, Taihang Wang, Youzheng Liu, Yimin Wang 외 arxiv

In-context learning (ICL) for text classification, which uses a few input-label demonstrations to describe a task, has demonstrated impressive performance on large language models (LLMs). However, the selection of in-con…

Semantic SimilarityText Classification

SafeDICE: Offline Safe Imitation Learning with Non-Preferred Demonstrations

2023-09-21 · NeurIPS 2023 11

We consider offline safe imitation learning (IL), where the agent aims to learn the safe policy that mimics preferred behavior while avoiding non-preferred behavior from non-preferred demonstrations and unlabeled demonst…

Curriculum Demonstration Selection for In-Context Learning

2024-11-27 · Duc Anh Vu, Nguyen Tran Cong Duy, Xiaobao Wu, Hoang Minh Nhat 외

Large Language Models (LLMs) have shown strong in-context learning (ICL) abilities with a few demonstrations. However, one critical challenge is how to select demonstrations to elicit the full potential of LLMs. In this …

In-Context Learning

Self-Generated In-Context Learning: Leveraging Auto-regressive Language Models as a Demonstration Generator

2022-06-16 · Hyuhng Joon Kim, Hyunsoo Cho, Junyeob Kim, Taeuk Kim 외

Large-scale pre-trained language models (PLMs) are well-known for being capable of solving a task simply by conditioning a few input-label pairs dubbed demonstrations on a prompt without being explicitly tuned for the de…

In-Context Learningtext-classificationText ClassificationZero-Shot Learning