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

InternLM-Math:面向可验证推理的开源数学大语言模型

InternLM-Math: Open Math Large Language Models Toward Verifiable Reasoning

Shanghai AI Laboratory(上海人工智能实验室);Fudan University(复旦大学);University of Southern California(南加州大学);Shanghai Jiaotong University(上海交通大学);Tsinghua University(清华大学)
Huaiyuan Ying, Shuo Zhang, Linyang Li, Zhejian Zhou, Yunfan Shao, Zhaoye Fei, Yichuan Ma, Jiawei Hong, Kuikun Liu, Ziyi Wang, Yudong Wang, Zijian Wu, Shuaibin L… 展开作者

Huaiyuan Ying, Shuo Zhang, Linyang Li, Zhejian Zhou, Yunfan Shao, Zhaoye Fei, Yichuan Ma, Jiawei Hong, Kuikun Liu, Ziyi Wang, Yudong Wang, Zijian Wu, Shuaibin Li, Fengzhe Zhou, Hongwei Liu, Songyang Zhang, Wenwei Zhang, Hang Yan, Xipeng Qiu, Jiayu Wang, Kai Chen, Dahua Lin

AI总结:

本文开源了基于 InternLM2 继续预训练的数学大模型 InternLM-Math,统一了思维链、奖励建模、形式推理、数据增强和代码解释器,在多个基准上取得开源最优性能,并探索了 LEAN 作为统一数学求解与证明平台的可能性。

摘要:

大语言模型的数学能力可以代表其抽象推理能力。在本文中,我们介绍并开源了我们的数学推理大语言模型 InternLM-Math,该模型是在 InternLM2 的基础上继续预训练得到的。我们将思维链推理、奖励建模、形式推理、数据增强和代码解释器统一为一种序列到序列的格式,并训练我们的模型成为一个多功能的数学推理者、验证者、证明者和增强者。这些能力可用于开发下一代数学大语言模型或进行自我迭代。在各种非正式和正式基准测试中,包括 GSM8K、MATH、匈牙利数学考试、MathBench-ZH 和 MiniF2F,InternLM-Math 在上下文学习、监督微调和代码辅助推理的设置下取得了开源最优性能。我们的预训练模型在 MiniF2F 测试集上无需微调即可达到 30.3 的分数。我们进一步探索了如何使用 LEAN 来解决数学问题,并研究了其在多任务学习设置下的性能,这表明了使用 LEAN 作为数学求解和证明的统一平台的可能性。我们的模型、代码和数据发布在 \url{https://github.com/InternLM/InternLM-Math}。

英文摘要:

The math abilities of large language models can represent their abstract reasoning ability. In this paper, we introduce and open-source our math reasoning LLMs InternLM-Math which is continue pre-trained from InternLM2. We unify chain-of-thought reasoning, reward modeling, formal reasoning, data augmentation, and code interpreter in a unified seq2seq format and supervise our model to be a versatile math reasoner, verifier, prover, and augmenter. These abilities can be used to develop the next math LLMs or self-iteration. InternLM-Math obtains open-sourced state-of-the-art performance under the setting of in-context learning, supervised fine-tuning, and code-assisted reasoning in various informal and formal benchmarks including GSM8K, MATH, Hungary math exam, MathBench-ZH, and MiniF2F. Our pre-trained model achieves 30.3 on the MiniF2F test set without fine-tuning. We further explore how to use LEAN to solve math problems and study its performance under the setting of multi-task learning which shows the possibility of using LEAN as a unified platform for solving and proving in math. Our models, codes, and data are released at \url{https://github.com/InternLM/InternLM-Math}.

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

↑