让大模型学会在不确定中主动探索,提升决策能力。
EVOLvE: Evaluating and Optimizing LLMs For In-Context Exploration
- 用上下文引导和算法蒸馏增强模型探索能力
- 小模型经优化后超越大模型的探索表现
- 适用于需要主动学习的推荐与医疗场景
尽管大语言模型在多个领域取得成功,但在需在不确定性下做出最优决策的场景中仍研究不足。这类能力对个性化推荐、医疗干预等现实应用至关重要。本文通过多臂赌博机(bandits)这一无状态强化学习设置,评估大模型的决策能力,构建涵盖不同难度的上下文无关与上下文相关赌博机环境以进行基准测试。受最优探索算法启发,提出两种高效集成方法:推理时提供显式算法指导,以及通过合成数据进行算法蒸馏(基于算法生成的演示和微调)。令人惊喜的是,这些方法使小型模型在多种任务上超越大型模型。我们进行了详尽消融实验,分析任务难度与数据表示对探索效率的影响,并利用“遗憾”(regret)概念严格评估探索效率,揭示模型规模与底层算法对探索能力的影响。
原文摘要 · Abstract (English)
Despite their success in many domains, large language models (LLMs) remain under-studied in scenarios requiring optimal decision-making under uncertainty. This is crucial as many real-world applications, ranging from personalized recommendations to healthcare interventions, demand that LLMs not only predict but also actively learn to make optimal decisions through exploration. In this work, we measure LLMs' (in)ability to make optimal decisions in bandits, a state-less reinforcement learning setting relevant to many applications. We develop a comprehensive suite of environments, including both context-free and contextual bandits with varying task difficulties, to benchmark LLMs' performance. Motivated by the existence of optimal exploration algorithms, we propose efficient ways to integrate this algorithmic knowledge into LLMs: by providing explicit algorithm-guided support during inference; and through algorithm distillation via in-context demonstrations and fine-tuning, using synthetic data generated from these algorithms. Impressively, these techniques allow us to achieve superior exploration performance with smaller models, surpassing larger models on various tasks. We conducted an extensive ablation study to shed light on various factors, such as task difficulty and data representation, that influence the efficiency of LLM exploration. Additionally, we conduct a rigorous analysis of the LLM's exploration efficiency using the concept of regret, linking its ability to explore to the model size and underlying algorithm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。