arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2212.10071cs.CLcs.AIcs.LG
收藏

大语言模型是推理教师

Large Language Models Are Reasoning Teachers

KAIST(韩国科学技术院)

Namgyu Ho, Laura Schmid, Se-Young Yun

AI总结:

针对基于提示的思维链方法依赖超大规模模型、部署成本高的问题,提出Fine-tune-CoT方法,用大模型生成的推理样本微调小模型,结合多样化推理数据增强,大幅提升小模型推理能力,甚至在多任务上超过教师模型。

摘要:

近期研究表明,思维链(CoT)提示能够引导语言模型逐步解决复杂推理任务。然而,基于提示的CoT方法依赖于GPT-3 175B这类超大规模模型,这类模型的规模化部署成本极高,难以推广。在本文中,我们将这类大模型用作推理教师,为更小的模型赋予复杂推理能力,同时将模型规模的要求降低数个数量级。我们提出Fine-tune-CoT方法:利用超大规模教师模型生成推理样本,用以微调较小的学生模型。我们在多种公开模型和复杂任务上评估了该方法,发现Fine-tune-CoT能让小模型获得显著的推理能力,在许多任务上的表现远超基于提示的基线模型,甚至超过教师模型。此外,我们还利用教师模型为每个原始样本生成多个不同推理依据的能力对方法进行扩展,用这类多样化推理数据丰富微调数据集后,即使是极小的模型,在各数据集上的性能也能得到大幅提升。我们还开展了消融实验和样本分析,以探究学生模型推理能力的涌现机制。我们的代码实现和数据可在https://github.com/itsnamgyu/reasoning-teacher获取。

英文摘要:

Recent works have shown that chain-of-thought (CoT) prompting can elicit language models to solve complex reasoning tasks, step-by-step. However, prompt-based CoT methods are dependent on very large models such as GPT-3 175B which are prohibitive to deploy at scale. In this paper, we use these large models as reasoning teachers to enable complex reasoning in smaller models and reduce model size requirements by several orders of magnitude. We propose Fine-tune-CoT, a method that generates reasoning samples from very large teacher models to fine-tune smaller models. We evaluate our method on a wide range of public models and complex tasks. We find that Fine-tune-CoT enables substantial reasoning capability in small models, far outperforming prompt-based baselines and even the teacher model in many tasks. Additionally, we extend our method by leveraging the teacher model's ability to generate multiple distinct rationales for each original sample. Enriching the fine-tuning data with such diverse reasoning results in a substantial performance boost across datasets, even for very small models. We conduct ablations and sample studies to understand the emergence of reasoning capabilities of student models. Our code implementation and data are available at https://github.com/itsnamgyu/reasoning-teacher.

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

↑