arXiv:2508.15868cs.CLcs.AI2025-08EMNLP被引 4

通过对比学习提升大模型推理能力,解决训练不稳定与思维链利用不充分问题。

CARFT: Boosting LLM Reasoning via Contrastive Learning with Annotated Chain-of-Thought-based Reinforced Fine-Tuning

  • 引入对比学习构建思维链表征,生成稳定训练信号。
  • 在两个数据集上实现最高10.15%的推理性能提升,训练效率提高30.62%。
  • 适合需要增强推理能力的大模型应用,如数学解题、逻辑判断。

推理能力在大语言模型的广泛应用中至关重要。为提升大模型的推理性能,已有多种基于强化学习的微调方法被提出,以克服仅通过监督微调训练时泛化能力不足的问题。然而,现有方法存在两大局限:一是原始强化学习方法忽略标注的思维链(CoT),且采用不稳定的推理路径采样,易导致模型崩溃、训练过程不稳和性能欠优;二是现有监督微调方法过度依赖标注的思维链,可能因未能充分挖掘潜在思维链而造成性能下降。本文提出一种基于标注思维链的对比学习强化微调方法(CARFT),旨在同时解决上述问题。具体而言,我们为每个思维链学习一个表征,并基于此设计新的对比信号以指导微调过程。该方法不仅充分利用了已有的标注思维链,还通过引入额外的无监督学习信号,稳定了微调过程。我们在三种基线方法、两个基础模型和两个数据集上进行了全面实验与深入分析,结果表明,CARFT在鲁棒性、性能(最高提升10.15%)和效率(最高提升30.62%)方面均表现出显著优势。代码已公开于https://github.com/WNQzhu/CARFT。

原文摘要 · Abstract (English)

Reasoning capability plays a significantly critical role in the the broad applications of Large Language Models (LLMs). To enhance the reasoning performance of LLMs, diverse Reinforcement Learning (RL)-based fine-tuning approaches have been proposed to address the limited generalization capability of LLMs trained solely via Supervised Fine-Tuning (SFT). Despite their effectiveness, two major limitations hinder the advancement of LLMs. First, vanilla RL-based approaches ignore annotated Chain-of-Thought (CoT) and incorporate unstable reasoning path sampling, which typically results in model collapse, unstable training process, and suboptimal performance. Second, existing SFT approaches generally overemphasize the annotated CoT, potentially leading to performance degradation due to insufficient exploitation of potential CoT. In this paper, we propose a Contrastive learning with annotated CoT-based Reinforced Fine-Tuning approach, i.e., \TheName{}, to enhance the reasoning performance of LLMs while addressing the aforementioned limitations. Specifically, we propose learning a representation for each CoT. Based on this representation, we design novel contrastive signals to guide the fine-tuning process. Our approach not only fully exploits the available annotated CoT but also stabilizes the fine-tuning procedure by incorporating an additional unsupervised learning signal. We conduct comprehensive experiments and in-depth analysis with three baseline approaches, two foundation models, and two datasets to demonstrate significant advantages of \TheName{} in terms of robustness, performance (up to 10.15\%), and efficiency (up to 30.62\%). Code is available at https://github.com/WNQzhu/CARFT.

大模型推理对比学习强化微调思维链

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