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

オンライン政策蒸留によるデータ効率的な自己回帰から拡散言語モデルへの変換

記事の要約

研究者らはOPDLMを提案し、オンライン政策蒸留を用いて自己回帰言語モデルを拡散言語モデルへ効率的に変換する。訓練と推論の不一致と知識喪失の課題を解決し、訓練トークン数を15倍から7,000倍削減する。

ソースarXiv Computational Linguistics著者: Xingyu Su, Jacob Helwig, Shubham Parashar, Atharv Chagi, Lakshmi Jotsna, Degui Zhi, James Caverlee, Dileep Kalathil, Shuiwang Ji
オンライン政策蒸留によるデータ効率的な自己回帰から拡散言語モデルへの変換
誤りを報告

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

訂正案内
本文へ

最近、arXivに発表された論文では、自己回帰言語モデル(ARLM)を拡散言語モデル(DLM)へ効率的に変換する新しい手法「オンライン政策拡散言語モデル(OPDLM)」が提案された。この研究はXingyu Suら9名の著者によるもので、論文番号はarXiv:2606.06712v1、計算と言語(cs.CL)分野に属し、人工知能(cs.AI)にも関連している。

従来、自己回帰モデルを拡散モデルに変換する方法では、ARLMの因果的注意を双方向注意に置き換え、その結果得られたモデルをDLMの目的関数で訓練していた。しかし、このアプローチには二つの分布シフトの問題が生じる。第一に、次トークン予測目的からDLM目的への移行により、ARLMが訓練中に獲得した知識が失われる可能性がある。第二に、標準的なDLMは訓練と推論の不一致に悩まされる。訓練損失はランダムにマスクされた系列上で定義されるが、推論時には信頼度ベースのデコーディングによって生成される軌跡上で encountered されるためである。

これらの課題を解決するため、研究チームはOPDLMを導入した。ここではオンライン政策蒸留(OPD)がARLMからDLMへの変換に用いられる。具体的には、OPDLMは自己OPDによって訓練される。学生モデルは双方向注意を備えたARLMであり、自身の軌跡を生成する。教師モデルは元の凍結されたARLMであり、これらの軌跡上で目標ロジットを提供することで知識を蒸留する。オンライン政策方式で直接訓練することで、OPDLMはDLMにおける訓練と推論の不一致を解消し、同時に元のモデルからの蒸留によりARLMからの知識保持を強化する。

実証結果は、OPDLMが多様なタスクで強い性能を示しながら、従来の方法と比較して15倍から7,000倍少ない訓練トークンで済むことを示している。この大幅なデータ効率の向上により、DLMの事前訓練にかかる prohibitive なコストが回避され、DLM変換がARLMの事後訓練の一形態として位置づけられる。本論文は2026年6月4日に提出され、モデル変換の効率化における重要な進展を示している。

要点と分析を開く

記事インテリジェンス

投資家上級

要点

  • OPDLMは自己オンライン政策蒸留によりARLMをDLMへ変換し、ゼロからの事前訓練コストを回避する。
  • オンライン政策訓練は拡散言語モデルに固有の訓練と推論の不一致を解消する。
  • 元の凍結ARLMからの蒸留により、自己回帰訓練で獲得した知識を保持する。
  • 実験により、多様なタスクで強い性能を維持しつつ、訓練トークン数を15倍から7,000倍削減できることが示された。

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