arXiv:2505.00031cs.CLcs.AI2025-05ICLR被引 5

让大模型先拟计划再解题,提升推理泛化能力

Learning to Plan Before Answering: Self-Teaching LLMs to Learn Abstract Plans for Problem Solving

  • 训练模型先生成抽象解题计划,再据此求解
  • 在多个推理基准上优于传统方法,提升泛化性能
  • 适合需要复杂逻辑推理的AI系统开发者

在大语言模型(LLM)后训练领域,利用模型自动生成的合成数据已被证明有效。然而关键问题仍未解决:此类自生成数据应包含何种核心信息?现有方法仅生成逐步解题过程,未能捕捉跨相似问题所需的抽象元知识。受认知科学启发,人类在解决复杂问题前会进行高层抽象,我们提出新型自训练算法LEPA(Learning to Plan before Answering)。LEPA训练模型在接触具体问题前,先制定前瞻性的解题计划,作为解决问题的抽象元知识。该方法不仅明确解题路径,还帮助模型避开无关细节干扰。数据生成阶段,先根据问题构建前瞻计划,再生成与计划和问题一致的解答;通过自我反思优化计划,使其更有利于得出正确答案。模型优化阶段,训练模型同时预测优化后的计划和对应解答。通过高效提取并利用前瞻计划,LEPA在多个挑战性自然语言推理基准上显著优于传统算法。

原文摘要 · Abstract (English)

In the field of large language model (LLM) post-training, the effectiveness of utilizing synthetic data generated by the LLM itself has been well-presented. However, a key question remains unaddressed: what essential information should such self-generated data encapsulate? Existing approaches only produce step-by-step problem solutions, and fail to capture the abstract meta-knowledge necessary for generalization across similar problems. Drawing insights from cognitive science, where humans employ high-level abstraction to simplify complex problems before delving into specifics, we introduce a novel self-training algorithm: LEarning to Plan before Answering (LEPA). LEPA trains the LLM to formulate anticipatory plans, which serve as abstract meta-knowledge for problem-solving, before engaging with the intricacies of problems. This approach not only outlines the solution generation path but also shields the LLM from the distraction of irrelevant details. During data generation, LEPA first crafts an anticipatory plan based on the problem, and then generates a solution that aligns with both the plan and the problem. LEPA refines the plan through self-reflection, aiming to acquire plans that are instrumental in yielding correct solutions. During model optimization, the LLM is trained to predict both the refined plans and the corresponding solutions. By efficiently extracting and utilizing the anticipatory plans, LEPA demonstrates remarkable superiority over conventional algorithms on various challenging natural language reasoning benchmarks.

大模型推理自训练计划生成

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