arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2305.01879cs.CL

SCOTT:自一致思维链蒸馏

SCOTT: Self-Consistent Chain-of-Thought Distillation

University of Southern California(南加州大学);Amazon.com Inc(亚马逊公司)

Peifeng Wang, Zhengyang Wang, Zheng Li, Yifan Gao, Bing Yin, Xiang Ren

AI总结:

提出SCOTT方法,通过对比解码和反事实推理目标,从大型教师模型蒸馏出小型自一致思维链模型,生成更忠实的推理并提升决策性能。

摘要:

超过一定规模的大型语言模型(LMs)展现出通过思维链(CoT)提示为其预测生成自由文本理由的新兴能力。虽然CoT可以带来显著提升的性能,但这种增益仅对足够大的语言模型可见。更令人担忧的是,几乎没有保证生成的推理与语言模型的预测一致,或忠实地证明其决策。在这项工作中,我们提出了一种忠实的知识蒸馏方法,从规模大数个数量级的教师模型学习一个小的、自一致的CoT模型。为了形成更好的监督,我们通过对比解码从大型语言模型(教师)中引出支持黄金答案的理由,这鼓励教师生成只有在考虑答案时才变得更合理的标记。为了确保忠实的蒸馏,我们使用教师生成的推理来学习一个具有反事实推理目标的学生语言模型,这防止学生忽略推理而做出不一致的预测。实验表明,在产生可比的最终任务性能的同时,我们的方法可以生成比基线更忠实的CoT推理。进一步的分析表明,这样的模型在做出决策时更尊重推理;因此,我们可以通过改进其推理来进一步提高其性能。

英文摘要:

Large language models (LMs) beyond a certain scale, demonstrate the emergent capability of generating free-text rationales for their predictions via chain-of-thought (CoT) prompting. While CoT can yield dramatically improved performance, such gains are only observed for sufficiently large LMs. Even more concerning, there is little guarantee that the generated rationales are consistent with LM's predictions or faithfully justify the decisions. In this work, we propose a faithful knowledge distillation method to learn a small, self-consistent CoT model from a teacher model that is orders of magnitude larger. To form better supervision, we elicit rationales supporting the gold answers from a large LM (teacher) by contrastive decoding, which encourages the teacher to generate tokens that become more plausible only when the answer is considered. To ensure faithful distillation, we use the teacher-generated rationales to learn a student LM with a counterfactual reasoning objective, which prevents the student from ignoring the rationales to make inconsistent predictions. Experiments show that, while yielding comparable end-task performance, our method can generate CoT rationales that are more faithful than baselines do. Further analysis suggests that such a model respects the rationales more when making decisions; thus, we can improve its performance more by refining its rationales.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑