Research map / Large language models / Instruction and fine-tuning
EVOLvE: Evaluating and Optimizing LLMs For In-Context Exploration
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.
Introduces BanditBench, a natural-language multi-armed and contextual bandit benchmark for LLM in-context exploration, and improves LLMs via algorithm-guided UCB support and distillation of UCB trajectories through few-shot demos and fine-tuning.
Problem
LLMs are under-studied in sequential decision-making requiring exploration under uncertainty; it is unclear whether they can explore in-context to solve multi-armed and contextual bandit tasks.
Approach
Provide LLMs with UCB-style exploitation values and exploration bonuses in context (algorithm-guided support), and distill optimal UCB/LinUCB trajectories into LLMs via in-context demonstrations or oracle behavior fine-tuning (OFT), framed as next-action likelihood maximization.
Key results
- Gemini-1.5 Pro with algorithm-guided support reaches 59.6% MAB overall win-rate and 64.3% CB win-rate, versus 45.5% and 7.1% with raw history, while UCB scores 90.6% (MAB) and LinUCB 96.4% (CB).
- Oracle behavior fine-tuning of Gemini-1.5 Flash with raw history reaches 65.6% MAB win-rate, and fine-tuned Flash surpasses Gemini-1.5 Pro as stated by the authors.
- On CB, fine-tuning Flash with AG trajectories raises win-rate from 28.6% (RH) to 89.3% (AG at inference is used in evaluation as well).
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| provide UCB exploitation values and exploration bonuses instead of Raw History | Gemini-1.5 Flash · Contextual Bandit | overall win-rate: 0.0 -> 46.4 | Helped |
| provide UCB exploitation values and exploration bonuses instead of Raw History | Gemini-1.5 Pro · Contextual Bandit | overall win-rate: 7.1 -> 64.3 | Helped |
| use Algorithm-Guided support instead of Raw History | Gemini-1.5 Flash · MAB, K=20 | overall win-rate: 21.9 -> 37.9 | Helped |
| use Algorithm-Guided support instead of Raw History | Gemini-1.5 Flash · MAB, K=5 | overall win-rate: 33.6 -> 26.6 | Hurt |
| add UCB few-shot demonstrations | Gemini-1.5 Flash · MAB | overall win-rate: 27.7 -> 50.2 | Helped |
| add UCB few-shot demonstrations | Gemini-1.5 Pro · MAB and Contextual Bandit | overall win-rate | Hurt |
| fine-tune on UCB trajectories | Gemini-1.5 Flash · MAB and Contextual Bandit | overall win-rate | Helped |
| use easy rather than hard trajectories for few-shot demonstrations (ablation) | Gemini-1.5 Flash · MAB | overall win-rate: 43.0 -> 50.2 | Helped |
| fine-tune on hard rather than easy trajectories (ablation) | Gemini-1.5 Flash · MAB | overall win-rate: 54.5 -> 65.6 | Helped |
| use summarized rather than raw-history oracle demonstrations (ablation) | Gemini-1.5 Flash few-shot · MAB | overall win-rate: 27.5 -> 50.2 | Helped |
| use raw-history rather than summarized-history OFT trajectories (ablation) | Gemini-1.5 Flash · MAB | overall win-rate: 28.3 -> 65.6 | Helped |
| include Algorithm-Guided information in OFT trajectories (ablation) | Gemini-1.5 Flash · Contextual Bandit | overall win-rate: 28.6 -> 89.3 | Helped |
Surprising findings
- Few-shot demonstrations helped Gemini-1.5 Flash but hurt Gemini-1.5 Pro in both MAB and CB.
- Fine-tuning on the hard (Clothes, K=20) trajectories generalized better than on easy trajectories, while few-shot demonstrations showed the opposite trend.
Limitations
- A substantial optimality gap remains between LLMs and UCB/LinUCB.
- Fine-tuning may overfit to training task distribution, with weaker generalization for easy-trained models on some settings.
- Evaluation cost is high for large API models, limiting scale of benchmark use.
Experimental setup
Datasets: MovieLens-1M UCB-generated bandit trajectories LinUCB-generated contextual bandit trajectories
Benchmarks: BanditBench MAB (16 configurations) BanditBench CB (MovieLens-based, K=10 and K=30)
Metrics: overall win-rate (pairwise Student's t-test, p<0.05) cumulative regret MinFrac OptFrac fitted regret function alpha and beta
Baselines: Raw History prompting Summarized History prompting UCB LinUCB
Builds on: UCB LinUCB Algorithm Distillation Behavior Cloning Summarized history from prior LLM bandit work
Closest papers by meaning
- Can large language models explore in-context? (NeurIPS 2024 · similarity 0.93)
- Efficient Exploration for LLMs (ICML 2024 · similarity 0.90)
- From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systems (ICML 2024 · similarity 0.90)
- Expanding LLM Agent Boundaries with Strategy-Guided Exploration (NeurIPS 2026 · similarity 0.90)
- Efficient Reinforcement Learning with Large Language Model Priors (ICLR 2025 · similarity 0.90)
- LLM-Explorer: A Plug-in Reinforcement Learning Policy Exploration Enhancement Driven by Large Language Models (NeurIPS 2025 · similarity 0.90)
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.
