アーカイブ / LLM
正誤判定を回答の修正に生かす自己蒸留:数学・コード課題での検証
Self-Distillation Zero
Self-Distillation Zeroの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1: We introduce SD-Zero , an on-policy self-distillation paradigm where the teacher ( reviser ) can condition on incorrect student ( generator ) attempts.
概要
- 一つのモデルが初回答を正誤に応じて修正し、その修正役の予測を生成役の学習に使う。
- 著者らの8ベンチマークでのavg@8全体平均は、Qwenで49.8から60.3、Olmoで41.1から51.5になった。
- 応用には正誤を判定できる問題と正解値またはテストが必要で、未使用問題での成績と回答の長さを確かめたい。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、Self-Distillation Zeroの仕組み
- 01問題と正解値
- 02初回答を判定
- 03自己修正を学習
- 04予測を生成役へ反映
- 05新たな初回答
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
数学の最終答案やコードの実行結果は、正誤を判定できる場合がある。ただし、回答全体への二択の判定だけでは、長い解答のどこを直すべきかは分からない。そのため、学習には多くの回答を試すか、細かな手本を用意する必要が生じる。
SD-Zeroは、モデル自身の初回答とその正誤を使って修正方法を学び、修正する力を次の初回答に反映する。学習には正解の最終値を使う一方、模範解答の途中の推論過程は必要としない。
手法
まずモデルに問題を解かせ、初回答を正解の最終値と照合する。誤答なら問題・初回答・誤りという判定を見せて書き直させ、修正後に正解した例を残す。正答には言い換えを促す。集めた例を使い、初回答を修正する役と、問題から回答を作る役を同じモデルに学習させる。
次に生成役が新しい回答を作り、修正役はその回答と正誤を見て、各トークンの次に来る候補の確率を出す。生成役の予測をその確率に近づける工程が自己蒸留である。例えばコード問題では、初回答の合否を受けた修正役の予測を使い、次は問題文だけから回答する生成役を更新する。
新規性
SD-Zeroは、一つのモデルを初回答を作る生成役と、その回答を正誤判定に応じて見直す修正役として学習させる。正解の最終値は必要だが、外部の教師モデルや模範的な解答過程は前提にしない。
著者らは、誤答に対する修正役と生成役の予測の差が、一部のトークンに集中すると分析した。トークンは文章を処理する小さな単位である。更新したモデルを修正役に同期し、学習を続ける方法も検証している。
従来手法との違い
比較対象は、正解した自己生成回答を選んで学ぶRFT、正誤を報酬に使うGRPO、良質な解答例を使う自己蒸留のSDFTなどである。RFTは誤答を学習対象から除くが、SD-Zeroの第1段階では誤答を修正時の文脈に残す。
第2段階では、生成役が今作った回答に対し、修正役が各トークンの予測を学習信号として返す。著者らの比較では、問題集合とモデルが生成する回答数の予算をそろえた条件でSD-Zeroの全体平均が上回った。
実験結果
著者らはQwen3-4B-InstructとOlmo-3-7B-Instructを、数学・コードの計8ベンチマークで評価した。指標は8回生成した回答の平均成績を示すavg@8。表1の全体平均は、Qwenが基礎モデル49.8からSD-Zero 60.3、Olmoが41.1から51.5となった。
学習には数学のOpenR1-Mathから1万5000問、コードのCodeforcesからC++とPythonで各7500例を選び、分野別に学習した。第1段階は6000件の自己修正例を使い、全体平均がQwen 57.6、Olmo 50.3だった。著者らは比較手法と問題集合・生成回答数の予算をそろえたとしている。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:社内のコード演習問題で、初回答の修正を学習する試験を行う。入力は問題文とモデルが生成したコード、出力は次に問題文だけを与えた際のコード案とする。用意したテストで初回答の正誤を判定し、その結果を修正役の入力に使う。
元データには問題文と正誤を判定できるテストが必要で、正誤ラベルはテスト結果から作る。追加センサーは使わない。コードの取り出し方と修正指示を問題形式に合わせて調整し、演習問題での結果を実務コードへそのまま当てはめない。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:正誤を自動判定できるコード演習問題を選び、問題文と判定用テストを元データとして整える。テストの合否を正誤ラベルにし、追加センサーは使わない。問題形式に応じて回答からコードを取り出す方法と修正指示を調整する。
同じ問題集合と回答生成数で、元モデル、修正学習後のモデル、自己蒸留後のモデルを比べる。学習に使わない問題でテスト通過率と回答の長さを測り、修正例の収集に要した生成数も記録する。テストを通ったコードも人が確認し、合否だけでは見えない欠点を調べる。
限界と課題
著者らは、修正例を集める第1段階では問題ごとに複数の初回答を生成する必要があり、学習後の回答が長くなりやすいと報告する。第2段階で修正役を固定すると性能が頭打ちになり、更新したモデルとの同期後に追加の改善が見られた。
編集上の検証事項は、正誤を自動判定しにくい課題にも適用できるか、テストを通っても不適切なコードを増やさないかである。確認できた本文の範囲では、実務環境での品質や運用費用を判断するための評価は示されていない。
出典
元のタイトル・要旨を確認する(英語)
Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervision
Current post-training methods in verifiable settings fall into two categories. Reinforcement learning (RLVR) relies on binary rewards, which are broadly applicable and powerful, but provide only sparse supervision during training. Distillation provides dense token-level supervision, typically obtained from an external teacher or using high-quality demonstrations. Collecting such supervision can be costly or unavailable. We propose Self-Distillation Zero (SD-Zero), a method that is substantially more training sample-efficient than RL and does not require an external teacher or high-quality demonstrations. SD-Zero trains a single model to play two roles: a Generator, which produces an initial response, and a Reviser, which conditions on that response and its binary reward to produce an improved response. We then perform on-policy self-distillation to distill the reviser into the generator, using the reviser's token distributions conditioned on the generator's response and its reward as supervision. In effect, SD-Zero trains the model to transform binary rewards into dense token-level self-supervision. On math and code reasoning benchmarks with Qwen3-4B-Instruct and Olmo-3-7B-Instruct, SD-Zero improves performance by at least 10% over the base models and outperforms strong baselines, including Rejection Fine-Tuning (RFT), GRPO, and Self-Distillation Fine-Tuning (SDFT), under the same question set and training sample budget. Extensive ablation studies show two novel characteristics of our proposed algorithm: (a) token-level self-localization, where the reviser can identify the key tokens that need to be revised in the generator's response based on reward, and (b) iterative self-evolution, where the improving ability to revise answers can be distilled back into generation performance with regular teacher synchronization. Code: https://github.com/princeton-pli/Self-Distillation-Zero.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01




