本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

オフライン推論訓練の重み空間幾何学

記事の要約

本論文は、推論蒸留に用いられる6種類のオフライン強化学習損失(SFT、RFT、DFT、RIFT、Offline GRPO、DPO)の重み更新幾何学を比較する。Qwen3-4Bをベースに同一の数学データで訓練し、SFT・RFT・RIFTはほぼ共線的な更新、DFTは発散、Offline GRPOは直交成分を追加、DPOはほぼ直交部分空間に位置し最高精度を示すがモード接続障壁があることを明らかにした。

ソースarXiv Machine Learning著者: Aleksandr Nikolich, Igor Kiselev, Vladimir Platonov, Karina Romanova
オフライン推論訓練の重み空間幾何学
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

オフライン強化学習損失(RFT、RIFT、DFT、Offline GRPO、DPO)は、大規模教師モデルから小規模生徒モデルへ推論能力を蒸留するために広く使われている。しかし、これらの手法は通常、下流タスクの精度のみで比較され、メカニズムの違いは不明である。本研究では、6つの手法(SFT、RFT、DFT、RIFT、Offline GRPO、DPO)を同一の数学データで訓練し、重み更新の幾何学的性質を直接比較した。

研究者らは、ベースモデルとしてQwen3-4Bを使用し、アテンションのみのLoRAアダプタを適用した。全ての手法に同一の数学推論データ(教師モデルが生成)を用い、コサイン類似度、主角度部分空間分析、線形モード接続性、CKA(Centered Kernel Alignment)を用いて訓練前後の重み差分を分析した。その結果、顕著な構造的差異が明らかになった。

まず、SFT(教師ありファインチューニング)、RFT(拒否ファインチューニング)、RIFT(報酬加重拒否ファインチューニング)の重み差分はほぼ共線的であり、コサイン類似度は0.97以上、主角度の中央値は約7度(144モジュール中)であった。これらの手法のGSM8K数学ベンチマーク精度は同等(約87-88%)で、ペアワイズMcNemar検定のp値は0.15以上であり、有意差はなかった。これは、RFTやRIFTが報酬加重によって独立したメカニズムを導入していない可能性を示唆する。

これに対し、DFT(直接選好ファインチューニング)は同じデータを使用しているにもかかわらず、更新方向が報酬加重法よりも大きく乖離した。Offline GRPO(グループ相対政策最適化のオフライン変種)は独自の特性を示した。SFT損失盆地を維持しつつ、SFT方向に直交する成分を大量に追加した(全体で約67%、後期層では最大86%)が、損失収束を妨げなかった。

最も注目すべきはDPO(直接選好最適化)である。その重み差分幾何学は他の手法とほぼ直交しており、後期層ではCKAが約0.46に低下し、モード接続障壁も観測された。幾何学的に孤立しているにもかかわらず、DPOは実験で最高精度を達成した。GSM8Kで93.5%、AIME26で30.0%(他の手法は3.3-10.0%)である。ただし、DPOは他の手法より10倍小さい学習率(標準的な慣行)を使用しているため、精度差は損失関数と最適化器の選択に起因する可能性がある。著者らは、学習率を一致させた比較を今後の課題としている。

この研究は、オフライン推論訓練手法が重み空間において顕著な幾何学的差異を持つことを強調している。その差異の一部は損失関数の設計に由来し、純粋な教師信号によるものではない。これらの幾何学的性質を理解することは、蒸留戦略やハイパーパラメータの最適化に役立つだろう。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • SFT、RFT、RIFTの重みコサイン類似度は0.97以上で、GSM8K精度は約87-88%で同等。
  • DFTの更新方向は報酬重み付け法よりも乖離が大きい。
  • Offline GRPOはSFT損失盆地を維持しつつ、約67%の直交成分を追加(後期層では86%)。
  • DPOはほぼ直交部分空間に位置し、GSM8Kで93.5%、AIME26で30%の最高精度だがモード接続障壁を示す。

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。