用融合错误轨迹提升大模型数学推理能力
Beyond Rejection Sampling: Trajectory Fusion for Scaling Mathematical Reasoning
- 将错误推理路径与正确路径交织生成新训练样本
- 在复杂问题上显著优于传统筛选方法,准确率提升显著
- 无需改架构或目标函数,适合想提升推理能力的研究者
大型语言模型在数学推理方面取得显著进展,通常通过拒绝采样微调,仅保留正确推理路径。然而,该方法将监督视为二值过滤器,系统性排除教师生成的错误,导致训练中对推理失败建模不足。本文提出TrajFusion,将拒绝采样重构为结构化监督构建过程:通过交替融合选定的错误路径、反思提示和正确路径,形成融合轨迹。每条融合样本长度基于教师错误的频率与多样性自适应调整,在难题上提供更丰富监督,而在错误信号无信息时自动退化为标准拒绝采样微调(RFT)。TrajFusion无需修改模型架构或训练目标。在多个数学基准上的大量实验表明,其性能持续优于RFT,尤其在复杂和长序列推理任务中表现突出。
原文摘要 · Abstract (English)
Large language models (LLMs) have made impressive strides in mathematical reasoning, often fine-tuned using rejection sampling that retains only correct reasoning trajectories. While effective, this paradigm treats supervision as a binary filter that systematically excludes teacher-generated errors, leaving a gap in how reasoning failures are modeled during training. In this paper, we propose TrajFusion, a fine-tuning strategy that reframes rejection sampling as a structured supervision construction process. Specifically, TrajFusion forms fused trajectories that explicitly model trial-and-error reasoning by interleaving selected incorrect trajectories with reflection prompts and correct trajectories. The length of each fused sample is adaptively controlled based on the frequency and diversity of teacher errors, providing richer supervision for challenging problems while safely reducing to vanilla rejection sampling fine-tuning (RFT) when error signals are uninformative. TrajFusion requires no changes to the architecture or training objective. Extensive experiments across multiple math benchmarks demonstrate that TrajFusion consistently outperforms RFT, particularly on challenging and long-form reasoning problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。