跳到內容

Technology

新聞|MagicBeanStalk:把 RLVR 解不出的難題拆開再長回來,合成難度剛好的訓練題

華盛頓大學、CMU、NVIDIA 與 Stanford 的研究者發現,一份 7.8 萬題的開源數學 RLVR 語料對 gpt-oss-120b 有約 88% 提供不了學習訊號。整理他們怎麼用難題的驗證解法拆出子題樹、再由下而上合成新題,以及數學與金融上的結果和限制。

AI
Published: at 09:00 AM
編輯本頁下載圖卡

重點

題目多,訊號不一定多

GRPO 這類方法對同一題取樣 NN 個回答,用二元驗證器打分 ri∈{0,1}r_i \in \{0, 1\},再拿每個回答跟組內平均比較:

rˉπ,L(p)=1N∑j=1Nrj,A^i(p)=ri(p)−rˉπ,L(p)\bar{r}_{\pi,L}(p) = \frac{1}{N}\sum_{j=1}^{N} r_j, \qquad \hat{A}_i(p) = r_i(p) - \bar{r}_{\pi,L}(p)

π\pi 是目標模型,LL 是每次回答的 token 上限。一組回答全對或全錯時,每個 A^i\hat{A}_i 都是 0,這題對訓練沒有貢獻。論文把這稱為零優勢陷阱(zero-advantage trap),把 0<rˉ<10 < \bar{r} < 1 的題稱為 Goldilocks 區間。

研究者合併 DAPO-Math-17K、Skywork-OR1-RL-Data、Turing Open-RL 與部分 StackMathQA,得到約 7.8 萬題。對 gpt-oss-120b 在 32K 預算下各取樣 8 次:

對照實驗也支持這個判斷。從一份 6,922 題、難度分布與原語料相同的代理集裡,只取 862 題 Goldilocks 題,或再加上「預算內可解」條件的 696 題來訓練,結果都比整份代理集好,題數少一個數量級。

篩選的問題在於它只能挑,會把模型還解不了的難題丟掉,而這些題剛好代表模型還缺的能力。

難度取決於模型與預算

同一題有沒有訊號,要看拿哪個模型、給多少 token:

所以回答被截斷,只代表預算內沒寫完,不代表模型不會。論文另加一個條件:至少要有一個答對的回答是正常結束、沒有碰到上限,稱為「預算內可解」(in-budget solvability)。一題只有同時落在 Goldilocks 區間、又預算內可解,才算乾淨的訓練訊號。

難題可以當規格用

難題沒有訊號,但它的驗證解法說明了需要哪些推理操作、怎麼組合,論文稱為推理藍圖(reasoning blueprint)。研究者用 98 題難題測這件事:

直接叫教師改題為什麼不夠

最直覺的做法,是把難題、解題過程與答案交給強教師,請它出變形題,再篩出 Goldilocks 題。論文稱為 seed synthesis。在同一批種子題的對照實驗裡,4 項核心基準(AIME 24/25、HMMT-F/N)的平均是:

訓練資料平均
只用種子題65.3
seed synthesis63.9
MBS69.2

seed synthesis 甚至比不上只用種子題。研究者找到兩個原因。第一,它保留了解題技巧,但常常只是改數字、換情境的淺層變形。第二,事後篩選能調難度,却無法把淺層變形變成新題,也沒有辦法替一個題都沒產生的種子補出候選。

給 seed synthesis 跟 MBS 差不多的生成預算(約 1,600 萬對 1,430 萬 token),在同樣 64 題種子上比較。「保留藍圖、夠新、落在 Goldilocks」三項同時成立的種子數:

多取樣可以提高 Goldilocks 的比例,新題比例的差距仍然在。

MBS 怎麼做

第一步:取得驗證解法。 教師池包括 Gemini 3 Flash、Claude Opus 4.6、DeepSeek V4 Pro、GPT 5.2 與 Kimi K2.6。依序讓各教師解題,取第一個答案正確的解法;全部失敗就丟掉。

第二步:由上而下拆題。 教師根據解法列出需要的技能,產生最多 3 個較簡單、可獨立作答的子題,每個子題也要通過驗證。接著做有效性測試:把子題與解法當範例給一組強弱不同的學生模型,再讓它們解父題。至少一個學生的正確率上升,這次拆解才算有效;不然就把失敗紀錄當反例請教師重拆,最多 3 次。拆解最深 3 層。消融實驗裡,教師認為合格的 22 個拆解只有 6 個通過有效性測試。

第三步:由下而上合成。 在葉節點明確要求教師出「更難」的變形,最弱學生正確率超過 0.8 的丟掉。接著照原本父題與子題的結構,把合成出來的子題組成新的父題,最弱學生正確率超過 0.625 的丟掉,每個父題最多試 10 種組合。一路往上,得到一棵跟原題平行的合成樹。論文觀察到,沒有要求「更難」時,教師傾向只改寫字面。

最後篩選。 只留下對目標模型 8 次 rollout 對 1 到 7 次、且至少一個正確回答在預算內正常結束的題。學生模型只用來判斷拆解有沒有幫助,最後的篩選只看目標模型。

金融題必須能從文件回答。MBS 依種子文件摘要的嵌入相似度找出未用過的文件,每題 35 或 75 份,把整棵合成樹改建在新文件的片段上。

結果

從難題產出可用資料。 取約 1,250 題 gpt-oss-120b 在 16K 下 8 次全錯的種子:

方法產生題數Goldilocks 比例合格比例至少產出一題合格題的種子
seed synthesis2,49436.8%31.7%43.8%
MBS3,37055.4%42.8%74.7%

數學 RLVR。 gpt-oss-120b 用 GRPO、LoRA(rank 32)、16 張 H100、16K 回答上限訓練。8 項基準為 AIME 24/25、AMC、HMMT-F/N、Minerva、Olympiad、Omega-500,平均如下:

訓練資料8 項平均變化
基礎模型61.2—
整份原始語料62.3+1.1
Golden Goose(使用已發布資料集)63.1+1.9
原始語料的預算內 Goldilocks 子集67.0+5.8
seed synthesis66.8+5.6
MBS69.8+8.6

同一份 MBS 資料不重新生成、不重新篩選,拿去對 DeepSeek v4 Flash 做全參數 GRPO(80 張 H100),平均從 72.4 升到 77.9。

超過訓練預算時。 模型都在 16K 訓練,推論時放寬上限:

訓練資料24K(相對基礎)32K(相對基礎)
預算內 Goldilocks 子集−2.9−3.9
seed synthesis0.0−0.3
MBS+1.0+1.5

在更難的 MathArena Apex-Shortlist(47 題)上,MBS 在 16K、24K、32K 各比基礎模型高 11.0、13.3、10.9 分,32K 時是 27.1 對 16.2。

回頭看原來的難題。 同樣那約 1,250 題,重新取樣的基礎模型能解出 9.5%,seed synthesis 訓練後 23.2%,MBS 訓練後 25.0%;8 次全對的題各有 0、6、14 題。

金融。 340 題種子來自 17 份企業信用協議,MBS 產出約 2.5 萬個候選、1.42 萬題 Goldilocks 題。Qwen3-4B-Instruct 經 RLVR 後:

基準訓練前 pass@1訓練後 pass@1
SecQue32.439.3
FinanceBench65.968.3

跟既有做法的關係

限制

參考資料

註釋

編輯本頁

Podcast

關於語音摘要

此語音摘要由 Google NotebookLM 提供,不完全代表作者本人對文章的理解。