arXiv:2505.11893cs.CLcs.AI2025-05被引 1

用强化学习动态规划LLM解多步NLP任务,更智能选步骤顺序。

RLAP: A Reinforcement Learning Enhanced Adaptive Planning Framework for Multi-step NLP Task Solving

  • 将任务建模为马尔可夫决策过程,用轻量级智能体学判断语言序列好坏。
  • 在多个数据集上提升任务成功率,尤其对复杂任务效果更明显。
  • 适合需要精细规划的NLP应用,如复杂问答或文本推理。

多步规划已被广泛用于提升大语言模型(LLMs)在下游自然语言处理(NLP)任务上的表现,通过将原任务分解为多个子任务并按序求解,无需额外训练。现有方法要么预设步骤顺序,要么在每一步尝试多种路径,但忽略了任务实例的语言特征,且依赖LLM自身能力评估中间反馈并选择子任务,导致结果不优。为此,本文提出强化学习增强的自适应规划框架(RLAP)。在该框架中,将NLP任务建模为马尔可夫决策过程(MDP),直接将LLM作为环境。特别地,训练一个轻量级智能体(Actor)通过强化学习估计包含状态与动作的自然语言序列的Q值。因此,在顺序规划过程中,能充分考虑每个序列的语言特征,且由智能体与LLM交互,为每个任务实例确定最优子任务顺序。我们在三类不同NLP任务上应用RLAP,并在多个数据集上进行大量实验,验证其有效性与鲁棒性。

原文摘要 · Abstract (English)

Multi-step planning has been widely employed to enhance the performance of large language models (LLMs) on downstream natural language processing (NLP) tasks, which decomposes the original task into multiple subtasks and guide LLMs to solve them sequentially without additional training. When addressing task instances, existing methods either preset the order of steps or attempt multiple paths at each step. However, these methods overlook instances' linguistic features and rely on the intrinsic planning capabilities of LLMs to evaluate intermediate feedback and then select subtasks, resulting in suboptimal outcomes. To better solve multi-step NLP tasks with LLMs, in this paper we propose a Reinforcement Learning enhanced Adaptive Planning framework (RLAP). In our framework, we model an NLP task as a Markov decision process (MDP) and employ an LLM directly into the environment. In particular, a lightweight Actor model is trained to estimate Q-values for natural language sequences consisting of states and actions through reinforcement learning. Therefore, during sequential planning, the linguistic features of each sequence in the MDP can be taken into account, and the Actor model interacts with the LLM to determine the optimal order of subtasks for each task instance. We apply RLAP on three different types of NLP tasks and conduct extensive experiments on multiple datasets to verify RLAP's effectiveness and robustness.

强化学习多步规划LLMNLP

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