arXiv:2502.01118cs.LGcs.AI2025-02ACL被引 7

用大模型预测收益,让经典算法更好做决策

Large Language Model-Enhanced Multi-Armed Bandits

  • 用大模型预测奖励,结合经典算法做决策
  • 在合成和真实文本任务中均优于直接选臂方法
  • 适合需要精准预测但无语义可利用的复杂场景

大型语言模型(LLMs)被用于解决多臂赌博机(MAB)等序列决策任务,传统方法是直接指令大模型选择要拉的臂。然而该方法在许多任务中表现不佳。本文提出新方法:以经典MAB算法为高层框架,利用大模型的上下文学习能力进行奖励预测。首先将基于大模型的奖励预测器融入经典的汤普森采样(TS)算法,并采用温度衰减策略实现从探索到利用的过渡;其次,将大模型预测器(温度设为0)引入基于回归预言机的MAB算法,该算法具有显式探索机制。还扩展了基于TS的算法至对决赌博机场景,仅需成对偏好反馈,需非平凡的算法修改。在合成任务和基于真实文本数据集的实验中,结果表明所提算法始终优于基于直接选臂的基线方法。有趣的是,在臂无语义可利用的挑战性任务中,本方法性能显著优于直接选臂的大模型方法。

原文摘要 · Abstract (English)

Large language models (LLMs) have been adopted to solve sequential decision-making tasks such as multi-armed bandits (MAB), in which an LLM is directly instructed to select the arms to pull in every iteration. However, this paradigm of direct arm selection using LLMs has been shown to be suboptimal in many MAB tasks. Therefore, we propose an alternative approach which combines the strengths of classical MAB and LLMs. Specifically, we adopt a classical MAB algorithm as the high-level framework and leverage the strong in-context learning capability of LLMs to perform the sub-task of reward prediction. Firstly, we incorporate the LLM-based reward predictor into the classical Thompson sampling (TS) algorithm and adopt a decaying schedule for the LLM temperature to ensure a transition from exploration to exploitation. Next, we incorporate the LLM-based reward predictor (with a temperature of 0) into a regression oracle-based MAB algorithm equipped with an explicit exploration mechanism. We also extend our TS-based algorithm to dueling bandits where only the preference feedback between pairs of arms is available, which requires non-trivial algorithmic modifications. We conduct empirical evaluations using both synthetic MAB tasks and experiments designed using real-world text datasets, in which the results show that our algorithms consistently outperform previous baseline methods based on direct arm selection. Interestingly, we also demonstrate that in challenging tasks where the arms lack semantic meanings that can be exploited by the LLM, our approach achieves considerably better performance than LLM-based direct arm selection.

强化学习大模型应用决策优化上下文学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。