通義千問開源Qwen2-Math,成為最先進的數學專項模型

新闻助手發表於2024-08-09

8月9日訊息,阿里通義團隊開源新一代數學模型Qwen2-Math,包含1.5B、7B、72B三個引數的基礎模型和指令微調模型。Qwen2-Math基於通義千問開源大語言模型Qwen2研發,旗艦模型 Qwen2-Math-72B-Instruct在權威測評集MATH上的得分超越GPT-4o、Claude-3.5-Sonnet、Gemini-1.5-Pro、Llama-3.1-405B等,以84%的準確率處理了代數、幾何、計數與機率、數論等多種數學問題,成為最先進的數學專項模型。

通義千問開源Qwen2-Math,成為最先進的數學專項模型

注:在MATH基準測評中,通義千問數學模型的旗艦款Qwen2-Math-72B-Instruct取得了84%的準確率,超過GPT-4o、Claude-3.5-Sonnet、Gemini-1.5-Pro 和 Llama-3.1-405B等開閉源模型。

Qwen2-Math 基礎模型使用 Qwen2大語言模型進行初始化,並在精心設計的數學專用語料庫上進行預訓練,訓練資料包含大規模高質量的數學網路文字、書籍、程式碼、考試題目,以及由 Qwen2 模型合成的數學預訓練資料。所有預訓練和微調資料集都進行了去汙染處理。

隨後,研發團隊訓練了指令微調版本模型:首先,基於Qwen2-Math-72B 訓練一個數學專用的獎勵模型;接著,將密集的獎勵訊號與指示模型是否正確回答問題的二元訊號結合,用作學習標籤,再透過拒絕取樣構建監督微調(SFT)資料;最後在SFT模型基礎上使用 GRPO 方法最佳化模型。

據悉,Qwen2-Math系列模型目前主要支援英文,通義團隊很快就將推出中英雙語版本,多語言版本也在開發中。

通義團隊在多箇中英文數學基準測評集對指令微調模型作了效能評估,除了 GSM8K 和 MATH等常見的測評基準 ,還引入了更具挑戰性的考試競賽類測試,如奧林匹克級別的基準測評OlympiadBench、大學數學級別的基準測評CollegeMath、高考(GaoKao)、美國數學邀請賽(AIME)2024 賽題、美國數學競賽( AMC)2023賽題,中文測評則有CMATH測評集、2024年中國高考和中考數學題。最終,Qwen2-Math-72B-Instruct表現優異,在十大測評中都獲得了遠超其他開源數學模型的成績。

通義千問開源Qwen2-Math,成為最先進的數學專項模型

注:研發團隊在greedyRM@8 的條件下對模型作了測評,表中為每款Qwen2-Math-72B-Instruct模型列出了三個得分結果,分別是第1次回答得分(無下標數字)、8次回答中出現最多次數的答案的得分,8次回答中reward model所選答案的得分。

“大模型能不能做數學題”,不僅是社交平臺的熱門話題,也是業界非常關注的研究課題。處理高階數學問題,需要模型具備複雜多步邏輯推理能力。通義團隊在技術部落格中表示,希望透過開源“為科學界解決高階數學問題做出貢獻”,未來將持續增強模型數學能力。

附:Qwen2-Math解題示例

Qwen2-Math解題示例


相關文章