arXiv:2505.17454cs.LGcs.CL2025-05EMNLP被引 13

用推理置信度提升大模型自训练效果,更关注推理过程质量。

Self-Training Large Language Models with Confident Reasoning

  • 基于推理阶段置信度筛选优质推理路径进行自训练。
  • 在6个基准上准确率超越现有自训练方法,包括分布内和分布外数据。
  • 适合希望提升模型推理可靠性、减少偶然正确答案的研究者。

大型语言模型(LLMs)通过生成推理路径来提升性能,但这一过程需要昂贵的人工标注。为解决此问题,近期研究探索了使用模型自身生成伪标签的自训练方法。现有方法多依赖最终答案的置信度(通过多数投票估算),但仅关注答案正确性,可能忽略推理路径的质量——即使错误推理也可能恰好得出正确答案。本文提出新方法CORE-PO,通过策略优化,使模型偏好高置信度的推理路径。实验表明,该方法在四个分布内与两个分布外基准上均优于现有自训练方法。

原文摘要 · Abstract (English)

Large language models (LLMs) have shown impressive performance by generating reasoning paths before final answers, but learning such a reasoning path requires costly human supervision. To address this issue, recent studies have explored self-training methods that improve reasoning capabilities using pseudo-labels generated by the LLMs themselves. Among these, confidence-based self-training fine-tunes LLMs to prefer reasoning paths with high-confidence answers, where confidence is estimated via majority voting. However, such methods exclusively focus on the quality of the final answer and may ignore the quality of the reasoning paths, as even an incorrect reasoning path leads to a correct answer by chance. Instead, we advocate the use of reasoning-level confidence to identify high-quality reasoning paths for self-training, supported by our empirical observations. We then propose a new self-training method, CORE-PO, that fine-tunes LLMs to prefer high-COnfidence REasoning paths through Policy Optimization. Our experiments show that CORE-PO improves the accuracy of outputs on four in-distribution and two out-of-distribution benchmarks, compared to existing self-training methods.

自训练推理质量大模型

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