arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.14201cs.LGcs.AIcs.CL
收藏

Goat:在算术任务上微调的LLaMA优于GPT-4

Goat: Fine-tuned LLaMA Outperforms GPT-4 on Arithmetic Tasks

National University of Singapore(新加坡国立大学)

Tiedong Liu, Bryan Kian Hsiang Low

AI总结:

Goat通过微调LLaMA在算术任务上超越GPT-4,利用一致数字分词和任务分解策略实现高精度,并支持低成本训练复现。

摘要:

我们引入了Goat,这是一个经过微调的LLaMA模型,在一系列算术任务上显著优于GPT-4。通过在合成生成的数据集上进行微调,Goat在BIG-bench算术子任务上达到了最先进的性能。特别是,零样本的Goat-7B匹配甚至超越了少样本的PaLM-540B所达到的准确率。令人惊讶的是,Goat仅通过监督微调就能在大数加减法上实现近乎完美的准确率,这对于之前的预训练语言模型(如Bloom、OPT、GPT-NeoX等)几乎是不可能的。我们将Goat的卓越性能归因于LLaMA对数字的一致分词。为了解决更具挑战性的任务,如大数乘法和除法,我们提出了一种基于可学习性对任务进行分类的方法,随后利用基本算术原理将不可学习的任务(如多位数乘法和除法)分解为一系列可学习的任务。我们彻底检查了模型的性能,对我们提出的分解步骤的有效性进行了全面评估。此外,Goat-7B可以在24GB VRAM GPU上使用LoRA轻松训练,便于其他研究人员复现。我们发布了我们的模型、数据集以及用于数据集生成的Python脚本。

英文摘要:

We introduce Goat, a fine-tuned LLaMA model that significantly outperforms GPT-4 on a range of arithmetic tasks. Fine-tuned on a synthetically generated dataset, Goat achieves state-of-the-art performance on BIG-bench arithmetic sub-task. In particular, the zero-shot Goat-7B matches or even surpasses the accuracy achieved by the few-shot PaLM-540B. Surprisingly, Goat can achieve near-perfect accuracy on large-number addition and subtraction through supervised fine-tuning only, which is almost impossible with previous pretrained language models, such as Bloom, OPT, GPT-NeoX, etc. We attribute Goat's exceptional performance to LLaMA's consistent tokenization of numbers. To tackle more challenging tasks like large-number multiplication and division, we propose an approach that classifies tasks based on their learnability, and subsequently decomposes unlearnable tasks, such as multi-digit multiplication and division, into a series of learnable tasks by leveraging basic arithmetic principles. We thoroughly examine the performance of our model, offering a comprehensive evaluation of the effectiveness of our proposed decomposition steps. Additionally, Goat-7B can be easily trained using LoRA on a 24GB VRAM GPU, facilitating reproducibility for other researchers. We release our model, dataset, and the Python script for dataset generation.

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

↑