港中大开发MathCoder,超越GPT-4的数学大模型

新知榜官方账号

2023-10-09 01:22:22

港中大开发MathCoder,超越GPT-4的数学大模型

MathCoder是一个新的数学大模型,可以在竞赛级“题库”Math上超过GPT-4。它可以无缝集成代码的能力,在遇到数学问题时,不仅能用自然语言推理,还能自动编写和执行代码来建模、推导公式与方程。

对标GPT-4代码解释器

为了提高大模型的数学推理能力,港大这项研究学习了GPT-4代码解释器的优点和工作原理,提出了一种微调开源语言模型的方法。该方法最终使大模型无缝集成代码,利用代码来解决数学问题。

数据集的准备

他们首先提出了一个可以生成高质量数学题的数据集:MathCodeInstruct。该数据集由两部分组成:种子数据(D0)和插值数据(D1)。种子数据主要基于GSM8K和MATH,并利用GPT-4收集答案。插值数据让GPT-4基于他们提出的一种叫做“问题插值提示”的方法生成。最终MathCodeInstruct数据集一共收集了8万道数学题。

微调方法

他们提出了一种定制的监督微调和推理方法,最终在Llama-2和CodeLlama上微调出了MathCoder。该方法使用特殊的token(<|text|>、<|code|>、<|execution|>)来识别训练数据集中哪一部分是自然语言、代码还是结果,让模型学习生成由这些特殊标记划分的自然语言和代码。在推理期间,该方法还会将动态执行的结果附加到模型的先前预测中。然后,继续基于这个新版本的输入自回归预测下一个token,以及最后的执行结果。作者表示,通过这种方式,模型将能够“看到”执行结果,并不断地继续推理。

评测结果

在评测中,MathCoder凭此直接在MATH和GSM8K这俩数据集上取得了45.2%和83.9%的好成绩。该成绩证明:其一,它超过了ChatGPT-3.5和PaLM-2等9个闭源模型,并在以数学竞赛题为主的MATH集上超过GPT-4。其二,它打败了此前数学领域里最强的开源模型WizardMath,成为新的开源之最。

论文地址:https://arxiv.org/abs/2310.03731

本页网址:https://www.xinzhibang.net/article_detail-15960.html

寻求报道,请 点击这里 微信扫码咨询

关键词

MathCoder GPT-4 数学 大模型 竞赛题 数据集

分享至微信: 微信扫码阅读

相关工具

相关文章

相关快讯