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

在少样本提示中利用训练数据实现数值推理

Leveraging Training Data in Few-Shot Prompting for Numerical Reasoning

ByteDance Research(字节跳动研究院);Singapore University of Technology and Design(新加坡科技设计大学)

Zhanming Jie, Wei Lu

AI总结:

针对数学应用题求解中提示泛化难、训练数据缺CoT标注的问题,提出动态程序提示与程序蒸馏两种利用训练数据的少样本方法,在三个标准数据集上显著优于基线,提升了提示泛化性与小模型性能。

摘要:

思维链(Chain-of-thought, CoT)提示结合大语言模型已在众多自然语言处理任务中被证明有效,但设计能良好泛化到多样问题类型的提示颇具挑战,尤其是在数学应用题(math word problem, MWP)求解场景中。此外,通常存在大量覆盖多样性更好的训练数据,但缺少CoT标注,这限制了监督学习技术的应用。为解决这些问题,我们研究了两种在少样本提示场景下利用训练数据的方法:动态程序提示和程序蒸馏。我们的方法很大程度上受Gao等人(2022)的启发,他们提出用程序作为中间推理步骤来替代CoT。这种提示策略让我们能够在MWP求解中通过程序执行准确验证答案的正确性。我们的动态程序提示包括通过从大语言模型中采样正确程序来标注训练数据,而程序蒸馏则是让一个更小的模型适配带有程序标注的训练数据。我们在三个标准MWP数据集上的实验证明了这些方法的有效性,相较于之前的提示和微调基线取得了显著提升。我们的结果表明,利用大量训练数据可以提高提示的泛化能力,并提升微调后的小模型在MWP求解中的性能。

英文摘要:

Chain-of-thought (CoT) prompting with large language models has proven effective in numerous natural language processing tasks, but designing prompts that generalize well to diverse problem types can be challenging, especially in the context of math word problem (MWP) solving. Additionally, it is common to have a large amount of training data that have a better diversity coverage but CoT annotations are not available, which limits the use of supervised learning techniques. To address these issues, we investigate two approaches to leverage the training data in a few-shot prompting scenario: dynamic program prompting and program distillation. Our approach is largely inspired by Gao et al., (2022), where they proposed to replace the CoT with the programs as the intermediate reasoning step. Such a prompting strategy allows us to accurately verify the answer correctness through program execution in MWP solving. Our dynamic program prompting involves annotating the training data by sampling correct programs from a large language model, while program distillation involves adapting a smaller model to the program-annotated training data. Our experiments on three standard MWP datasets demonstrate the effectiveness of these approaches, yielding significant improvements over previous baselines for prompting and fine-tuning. Our results suggest that leveraging a large amount of training data can improve the generalization ability of prompts and boost the performance of fine-tuned small models in MWP solving.

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

↑