🚀 MetaMath-Llemma-7B項目
MetaMath-Llemma-7B是基於強大的Llemma-7B模型,在MetaMathQA數據集上進行全量微調的模型。該模型在數學任務上表現出色,能有效提升解題性能。
🚀 快速開始
你可以通過以下鏈接查看項目相關信息:
✨ 主要特性
- 數據來源可靠:所有MetaMathQA數據均從GSM8K和MATH的訓練集擴充而來,無測試集數據。
- 性能提升顯著:使用MetaMathQA數據集,並將基礎模型從llama-2-7B更換為Llemma-7B後,MATH任務的Pass@1指標從19.8提升到了30.0。
📦 安裝指南
使用以下命令安裝所需依賴:
pip install transformers==4.35.0
pip install torch==2.0.1
pip install sentencepiece==0.1.99
pip install tokenizers==0.13.3
pip install accelerate==0.21.0
pip install bitsandbytes==0.40.0
pip install vllm
pip install fraction
pip install protobuf
💻 使用示例
基礎用法
模型的提示模板如下:
"Below is an instruction that describes a task. "
"Write a response that appropriately completes the request.\n\n"
"### Instruction:\n{instruction}\n\n### Response: Let's think step by step."
你需要將{instruction}
替換為你的查詢問題。
📚 詳細文檔
數據說明
所有MetaMathQA數據均從GSM8K和MATH的訓練集擴充而來,你可以查看meta-math/MetaMathQA
中的original_question
,每個條目都來自GSM8K或MATH的訓練集。
實驗結果
以下是不同模型在GSM8k和MATH任務上的Pass@1指標對比:
模型 |
GSM8k Pass@1 |
MATH Pass@1 |
MPT-7B |
6.8 |
3.0 |
Falcon-7B |
6.8 |
2.3 |
LLaMA-1-7B |
11.0 |
2.9 |
LLaMA-2-7B |
14.6 |
2.5 |
MPT-30B |
15.2 |
3.1 |
LLaMA-1-13B |
17.8 |
3.9 |
GPT-Neo-2.7B |
19.5 |
-- |
Falcon-40B |
19.6 |
2.5 |
Baichuan-chat-13B |
23.9 |
-- |
Vicuna-v1.3-13B |
27.6 |
-- |
LLaMA-2-13B |
28.7 |
3.9 |
InternLM-7B |
31.2 |
-- |
ChatGLM-2-6B |
32.4 |
-- |
GPT-J-6B |
34.9 |
-- |
LLaMA-1-33B |
35.6 |
3.9 |
LLaMA-2-34B |
42.2 |
6.24 |
RFT-7B |
50.3 |
-- |
LLaMA-1-65B |
50.9 |
10.6 |
Qwen-7B |
51.6 |
-- |
WizardMath-7B |
54.9 |
10.7 |
LLaMA-2-70B |
56.8 |
13.5 |
WizardMath-13B |
63.9 |
14.0 |
MAmmoTH-7B (COT) |
50.5 |
10.4 |
MAmmoTH-7B (POT+COT) |
53.6 |
31.5 |
Arithmo-Mistral-7B |
74.7 |
25.3 |
MetaMath-7B |
66.5 |
19.8 |
MetaMath-13B |
72.3 |
22.4 |
🔥 MetaMath-Llemma-7B |
69.2 |
30.0 |
🔥 MetaMath-Mistral-7B |
77.7 |
28.2 |
📄 許可證
本項目使用的許可證為llama2。
📚 引用信息
如果你使用了本項目的相關內容,請引用以下論文:
@article{yu2023metamath,
title={MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Models},
author={Yu, Longhui and Jiang, Weisen and Shi, Han and Yu, Jincheng and Liu, Zhengying and Zhang, Yu and Kwok, James T and Li, Zhenguo and Weller, Adrian and Liu, Weiyang},
journal={arXiv preprint arXiv:2309.12284},
year={2023}
}