让小模型学会真推理,避免被错误理由带偏
Mitigating Spurious Correlations Between Question and Answer via Chain-of-Thought Correctness Perception Distillation
- 设计正确性感知任务,引导模型用对的理由推答案
- 动态加权训练样本,强化有支持力的推理路径
- 提升小模型在常见和罕见题上的推理可靠性
大语言模型(LLMs)在推理任务上表现优异但部署成本高。因此,常通过大模型生成的思维链(CoT)数据微调小语言模型(SLMs),以复制其推理能力。然而这些CoT数据可能包含无效或冗余的推理过程,导致小模型学习到问题与答案之间的虚假关联,损害推理质量。本文提出链式思考正确性感知蒸馏(CoPeD),从任务设置与数据利用两方面改进学生模型。首先引入正确性感知任务,鼓励学生模型基于正确推理路径预测答案,并在推理错误时进行修正,从而提升推理忠实度并从错误中学习。其次提出正确性感知加权损失函数,根据推理与答案的联合损失动态调整每条样本的贡献,使模型更关注那些推理能有效支持正确答案的样本。实验表明,CoPeD在分布内(IND)与分布外(OOD)的多个推理基准数据集上均有效。
原文摘要 · Abstract (English)
Large language models (LLMs) excel at reasoning tasks but are expensive to deploy. Thus small language models (SLMs) are fine-tuned on CoT data generated by LLMs to copy LLMs' abilities. However, these CoT data may include noisy rationales that either fail to substantiate the answers or contribute no additional information to support answer prediction, which leads SLMs to capture spurious correlations between questions and answers and compromise the quality of reasoning. In this work, we propose Chain-of-Thought Correctness Perception Distillation (CoPeD), which aims to improve the reasoning quality of the student model from the perspectives of task setting and data utilization. Firstly, we introduce a correctness-aware task setting that encourages the student model to predict answers based on correct rationales and revise them when they are incorrect. This setting improves the faithfulness of reasoning and allows the model to learn from its mistakes. Then, we propose a Correctness-Aware Weighted loss, which dynamically adjusts the contribution of each training instance based on the combined loss of the rationale and the answer. This strategy encourages the model to focus more on samples where the rationale offers stronger support for the correct answer. Experiments have shown that CoPeD is effective on both in-distribution (IND) and out-of-distribution (OOD) benchmark reasoning datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。