arXiv:2605.09009cs.LGcs.AI2026-05被引 1

通过微调让大模型学会从离线数据中做序列决策,效果优于直接上下文学习。

Large Language Models for Sequential Decision-Making: Improving In-Context Learning via Supervised Fine-Tuning

  • 用监督微调让大模型模仿离线轨迹中的策略。
  • 在长时序、部分可观测等复杂场景下,决策差距显著缩小。
  • 适合医疗等离线数据丰富但需安全决策的领域使用。

大型语言模型(LLMs)展现出强大的上下文学习(ICL)能力,但在序列决策领域的潜力尚未充分探索。本文研究了LLMs在马尔可夫决策过程(MDPs)、部分可观测MDPs(POMDPs)和模糊POMDPs(APOMDPs)中的ICL能力。通过在离线、带标注的轨迹数据上对预训练LLM进行监督微调(SFT),实现了灵活的策略模仿。理论上,针对线性MDPs,我们解释了微调后的注意力层可隐式从上下文数据中估计最优Q函数,并推导出端到端的次优性上界,将上下文估计误差与训练长度偏差分离。实验表明,在合成的MDP、POMDP和APOMDP环境中,微调后的LLM相比仅依赖ICL和随机基线,显著减小了最优性差距,尤其在长时序、部分可观测和模型模糊环境下提升明显。结果表明,监督微调是赋予预训练LLM从离线数据中获得序列决策能力的有效路径,这在医疗等离线数据丰富的领域具有重要意义。

原文摘要 · Abstract (English)

Large language models (LLMs) have shown remarkable in-context learning (ICL) capabilities, yet their potential for sequential decision-making remains underexplored. In this paper, we study the ICL capabilities of LLMs in sequential decision-making settings, including Markov Decision Processes (MDPs), Partially Observable MDPs (POMDPs), and Ambiguous POMDPs (APOMDPs). We fine-tune pretrained LLMs to perform few-shot decision-making directly from offline, oracle-labeled trajectories. Our framework enables flexible imitation of policies through supervised fine-tuning (SFT). Theoretically, we focus on linear MDPs and interpret a fine-tuned attention layer as implicitly estimating optimal Q-functions from in-context data. Building on this interpretation, we derive an end-to-end suboptimality bound for the induced policy that separates the in-context estimation error from the training-length bias. Empirically, across synthetic MDP, POMDP, and APOMDP settings, we find that fine-tuned LLMs achieve substantially smaller optimality gaps than in-context-only and random baselines, with especially large gains in longer-horizon, partially observed, and model-ambiguous environments. Together, these results show that supervised fine-tuning provides an effective route to endowing pretrained LLMs with sequential decision-making capabilities from offline data, which is an important advantage in domains such as healthcare where offline data are abundant.

序列决策监督微调大模型应用强化学习

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