atlas

Research map / Large language models / Language, safety and interpretability

Many-Shot In-Context Learning

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: In-context learning analysis

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Studies many-shot in-context learning (up to ~8192 shots, up to 1M tokens) with Gemini 1.5 Pro across translation, summarization, planning, reward modeling, reasoning and classification, and introduces reinforced and unsupervised ICL to reduce reliance on human-written rationales.

Problem

Few-shot in-context learning is limited by short context windows, so it is unclear how LLM performance scales when prompted with hundreds to thousands of examples, and whether many-shot ICL can reduce dependence on human-written demonstrations.

Approach

Scales ICL to the many-shot regime with long-context LLMs and introduces Reinforced ICL (model-generated, correctness-filtered rationales as demonstrations) and Unsupervised ICL (only problems, no solutions) as alternatives to human-written shots.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use many-shot reinforced ICL with model-generated, answer-correct rationales instead of human-written CoTGemini 1.5 Pro · BIG-Bench Hard (8 tasks)success rate: 72.1% -> 83%Helped
Use many-shot unsupervised ICL with problems only instead of human-written CoTGemini 1.5 Pro · BIG-Bench Hard (8 tasks)success rate: 72.1% -> 77.1%Helped
Replace problems-only unsupervised ICL with reinforced ICL using model-generated rationalesGemini 1.5 Pro · BIG-Bench Hard (8 tasks)success rate: 77.1% -> 83%Helped
Increase the prompt from few-shot to many-shot examplesGemini 1.5 Pro · Logistics planningsuccess rate: 42% to 62%Helped
Increase translation prompt to the full 997-example dev setGemini 1.5 Pro · English to Bemba (FLORES-200)chrF2++: +15.3%Helped
Increase translation prompt to the full 997-example dev setGemini 1.5 Pro · English to Kurdish (FLORES-200)chrF2++: +4.5%Helped
Use model-generated rationales instead of ground-truth human-written rationalesReinforced ICL · MATH500accuracyHelped
Use problems only rather than problem-solution pairsUnsupervised ICL · MATH500accuracyHelped
Use model-generated rationales rather than ground-truth rationalesReinforced ICL · GSM8K test (MATH transfer)accuracyHelped
Use model-generated rationales instead of ground-truth rationales at up to 25 shotsReinforced ICL · GPQA diamondaccuracyHelped
Increase ground-truth-rationale prompt from 5 to 125 shots (ablation)ICL · GPQA diamondaccuracyHelped
Increase ground-truth-rationale prompt from 125 to 250 shots (ablation)ICL · GPQA diamondaccuracyHurt

Surprising findings

Limitations

Experimental setup

Datasets: MATH GSM8K GPQA BIG-Bench Hard FLORES-200 XSum XLSum Financial PhraseBank PDDL Logistics generated problems

Benchmarks: MATH500 GSM8K GPQA diamond BIG-Bench Hard (8 tasks) FLORES-200 English-Bemba and English-Kurdish XSum XLSum Financial PhraseBank Logistics PDDL planning

Metrics: accuracy chrF2++ ROUGE-L success rate negative log-likelihood best-of-4 accuracy

Baselines: few-shot ICL with human-written solutions Gemini 1.5 Pro 1-shot GPT-4-Turbo Claude-3-Opus Gemini 1.5 Flash PEGASUS mT5 Fast-Downward GPT-2 Medium trained from scratch k-nearest neighbors full supervised fine-tuning

Builds on: Reinforced Self-Training (ReST) chain-of-thought prompting KV caching STaR-style self-generated rationales

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy