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

探索大型语言模型在合成数据上的数学外推能力

Exploring Mathematical Extrapolation of Large Language Models with Synthetic Data

ByteDance(字节跳动);East China Normal University(华东师范大学)

Haolong Li, Yu Ma, Yinqi Zhang, Chen Ye, Jie Chen

AI总结:

本文通过新提出的算术谜题问题,利用高质量合成数据微调 open-llama-3B 模型,显著提升了大型语言模型在多步数学推理任务上的表现,并在领域外数据集上展现出良好的泛化能力。

摘要:

大型语言模型(LLMs)在语言理解、文本生成、代码合成及许多其他任务中表现出色,但在复杂的数学推理等多步推理问题上仍然存在困难。在本文中,通过一个新提出的算术谜题问题,我们展示了模型可以通过在高质量合成数据上进行微调,在多步推理任务中表现良好。使用 open-llama-3B 模型在三个不同测试数据集上的实验结果表明,该模型不仅能在领域内数据集上达到 0.44 的零样本 pass@1,还在领域外数据集上展现出一定的泛化能力。具体而言,本文分别通过扩展算术谜题问题的数值范围和组合组件,设计了两个领域外数据集。微调后的模型在这两个难度大得多的任务上表现令人鼓舞,零样本 pass@1 分别达到 0.33 和 0.35。

英文摘要:

Large Language Models (LLMs) have shown excellent performance in language understanding, text generation, code synthesis, and many other tasks, while they still struggle in complex multi-step reasoning problems, such as mathematical reasoning. In this paper, through a newly proposed arithmetical puzzle problem, we show that the model can perform well on multi-step reasoning tasks via fine-tuning on high-quality synthetic data. Experimental results with the open-llama-3B model on three different test datasets show that not only the model can reach a zero-shot pass@1 at 0.44 on the in-domain dataset, it also demonstrates certain generalization capabilities on the out-of-domain datasets. Specifically, this paper has designed two out-of-domain datasets in the form of extending the numerical range and the composing components of the arithmetical puzzle problem separately. The fine-tuned models have shown encouraging performance on these two far more difficult tasks with the zero-shot pass@1 at 0.33 and 0.35, respectively.

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

↑