arXiv:2506.06470cs.AIcs.LG2025-06NeurIPS被引 4

通过利用被丢弃的推理分支提升大模型逻辑能力

SIGMA: Refining Large Language Model Reasoning via Sibling-Guided Monte Carlo Augmentation

  • 用兄弟节点对比增强最优路径的推理
  • 7B模型仅用3万样本达54.92%准确率
  • 适合想降低数据成本的推理优化研究者

单纯扩大数据集已难带来显著收益,数据质量成为关键。蒙特卡洛树搜索(MCTS)能生成高质量思维链数据,但传统方法仅保留最优路径,舍弃了包含部分洞察、错误模式和替代策略的兄弟节点。这些被丢弃的信息往往蕴含丰富价值。本文提出SIGMA框架,重新整合这些兄弟节点,构建路径间语义关联,并采用两阶段精炼:批判模型识别兄弟集合中的优劣,修订模型基于此反馈进行文本反向传播,优化最优路径。通过挖掘非最优分支中被忽视的有效信号,SIGMA显著提升推理表现。在挑战性MATH基准上,经SIGMA微调的7B模型仅使用30,000样本即达到54.92%准确率,优于使用590,000样本训练的先进模型,证明该方法既能大幅减少数据用量,又能有效提升大模型推理能力。

原文摘要 · Abstract (English)

Enhancing large language models by simply scaling up datasets has begun to yield diminishing returns, shifting the spotlight to data quality. Monte Carlo Tree Search (MCTS) has emerged as a powerful technique for generating high-quality chain-of-thought data, yet conventional approaches typically retain only the top-scoring trajectory from the search tree, discarding sibling nodes that often contain valuable partial insights, recurrent error patterns, and alternative reasoning strategies. This unconditional rejection of non-optimal reasoning branches may waste vast amounts of informative data in the whole search tree. We propose SIGMA (Sibling Guided Monte Carlo Augmentation), a novel framework that reintegrates these discarded sibling nodes to refine LLM reasoning. SIGMA forges semantic links among sibling nodes along each search path and applies a two-stage refinement: a critique model identifies overlooked strengths and weaknesses across the sibling set, and a revision model conducts text-based backpropagation to refine the top-scoring trajectory in light of this comparative feedback. By recovering and amplifying the underutilized but valuable signals from non-optimal reasoning branches, SIGMA substantially improves reasoning trajectories. On the challenging MATH benchmark, our SIGMA-tuned 7B model achieves 54.92% accuracy using only 30K samples, outperforming state-of-the-art models trained on 590K samples. This result highlights that our sibling-guided optimization not only significantly reduces data usage but also significantly boosts LLM reasoning.

大模型推理MCTS数据效率思维链

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