小米官方今天正式宣布,開源首個為推理(Reasoning)而生的大模型——Xiaomi MiMo。
據(jù)介紹,Xiaomi MiMo在數(shù)學推理(AIME 24-25)和代碼競賽(LiveCodeBench v5)公開測評集上,MiMo僅用7B的參數(shù)規(guī)模,超越了OpenAI的閉源推理模型o1-mini和阿里Qwen更大規(guī)模的開源推理模型QwQ-32B-Preview。
官方表示,MiMo 推理能力的提升,由預訓練和后訓練階段中數(shù)據(jù)和算法等多層面的創(chuàng)新聯(lián)合驅(qū)動。包括:
預訓練:核心是讓模型見過更多推理模式
數(shù)據(jù):著重挖掘富推理語料,并合成約200B tokens推理數(shù)據(jù)。
訓練:進行了三階段訓練,逐步提升訓練難度,總訓練25T tokens。
后訓練:核心是高效穩(wěn)定的強化學習算法和框架
算法:提出 Test Difficulty Driven Reward 來緩解困難算法問題中的獎勵稀疏問題,并引入 Easy Data Re-Sampling 策略,以穩(wěn)定 RL 訓練。
框架:設計了Seamless Rollout系統(tǒng),使得RL訓練加速2.29倍,驗證加速1.96倍。
所有技術細節(jié)已經(jīng)Open,見技術報告。
MiMo-7B全系列已開源MiMo-7B 已開源4個模型至HuggingFace。