让大模型像侦探一样一步步推理,靠结构化知识和真实世界模型提升逻辑一致性。
Deliberate Reasoning in Language Models as Structure-Aware Planning with an Accurate World Model
- 用蕴含图构建结构化知识,支持符号验证中间步骤
- 多候选生成+判别器重排序,使推理更新更可信
- 适合数学、编程等复杂推理任务,对模型要求高
提升语言模型的推理能力仍是关键挑战,尤其在需要多步决策的任务中,现有链式思维(CoT)方法常因不一致性和缺乏验证而失效。本文提出一种新框架SWAP(结构感知规划与精准世界模型),将结构化知识表示与学习型规划结合。不同于依赖自然语言推理的方法,SWAP利用蕴含图编码结构依赖关系,实现中间步骤的符号验证。为系统构建和更新图谱,采用策略模型生成候选扩展,并通过世界模型预测结构更新;世界模型生成多个替代更新方案,由判别器根据合理性重新排序。为促进多样化探索,引入基于多样性的建模(DM),从原策略分布中移除已采样项后,从剩余概率质量中采样。此外,通过对比排序(CR)直接在提示中比较候选,结合元知识提升排名质量。在数学推理、逻辑推理和编程等多样化推理基准上进行评估,实验表明SWAP显著优于基线模型,且持续超越现有推理方法。
原文摘要 · Abstract (English)
Enhancing the reasoning capabilities of language models (LMs) remains a key challenge, especially for tasks that require complex, multi-step decision-making where existing Chain-of-Thought (CoT) approaches struggle with consistency and verification. In this paper, we propose a novel reasoning framework, referred to as Structure-aware Planning with an Accurate World Model (SWAP), that integrates structured knowledge representation with learned planning. Unlike prior methods that rely purely on natural language reasoning, SWAP leverages entailment graphs to encode structured dependencies and enable symbolic verification of intermediate steps. To systematically construct and update the graph, SWAP employs a policy model to propose candidate expansions and a world model to predict structural updates. To improve accuracy, the world model generates multiple alternative updates, and a discriminator re-ranks them based on plausibility. To encourage diverse exploration, we introduce Diversity-based Modelling (DM), which samples candidates from the remaining probability mass after removing previously sampled candidates from the original policy distribution. Additionally, SWAP improves the discrimination accuracy through Contrastive Ranking (CR), which directly compares candidates within prompts and incorporates meta-knowledge to improve ranking quality. We evaluate SWAP across diverse reasoning-intensive benchmarks including math reasoning, logical reasoning, and coding tasks. Extensive experiments demonstrate that SWAP significantly improves upon the base models and consistently outperforms existing reasoning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。