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

ソルバーボトルネックの解消:学習可能フロンティアにおけるタスク生成器の訓練

記事の要約

強化学習におけるエージェントの能力向上に伴い、固定タスク分布は飽和し、単純な合成生成では無効または簡単すぎるタスクが生じる。PROPELフレームワークは、軽量な活性化プローブを用いてソルバーの成功率を予測し、直接評価を回避することで、学習可能フロンティアに位置するタスクを効率的に生成する。数学、コード、ソフトウェア工学の各領域で、目標成功率でのタスク生成割合が大幅に向上した。

ソースarXiv Machine Learning著者: Lorenz Wolf, Connor Watts, Roger Creus Castanyer, Geoffrey Bradway, Maxwill Lin, Augustine N. Mavor-Parker, Matthew Daborn-Sargent
ソルバーボトルネックの解消:学習可能フロンティアにおけるタスク生成器の訓練
誤りを報告

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

訂正案内
本文へ

強化学習(RL)を用いたエージェントの訓練において、推論能力や自律性が向上するにつれ、固定されたタスク分布ではすぐに飽和が生じます。一方、単純に合成されたタスクは、容易すぎたり、解けない問題であったり、条件が不適切であることが多く、有効な学習の妨げとなります。このボトルネックを解消するため、タスク生成器をRLで訓練し、タスクの有効性と学習可能性を最適化するアプローチが提案されています。しかし、直接最適化するには各候補タスクに対してソルバーを繰り返し実行する必要があり、特にソフトウェア工学(SWE)タスクでは1回の実行に数十分かかることもあり、実用的ではありません。

そこで、Lorenz Wolfら7名の著者による本研究では、PROPELというフレームワークを提案します。PROPELは、ソルバー評価を償却するアプローチを採用し、目標とする解決率でタスクを生成する生成器を訓練します。具体的には、あらかじめ生成されたタスクとソルバーの結果からなるラベル付きコーパスを用いて、軽量な活性化プローブを訓練します。このプローブは、凍結された生成器の参照モデルから目標ソルバーのパス率を予測し、生成器最適化中の解決率の代理として機能するため、生成器の評価は単一の順伝播に削減されます。この償却戦略により、ソルバー呼び出しのコストが大幅に削減され、複雑なタスクにおいても効果的なタスク生成器の訓練が可能になります。

実験は、数学、コード、ソフトウェア工学の複数のモデル規模で実施されました。コードタスクでは、Qwen2.5-3B-Instructソルバーに対して、学習可能フロンティアのタスク生成割合が10.1%から20.0%に増加し、Qwen2.5-7B-Instructソルバーでは5.3%から12.6%に向上しました。SWEタスクでは、プローブと生成器の訓練時に見られなかったリポジトリに対してQwen3.5-27Bを用いた場合、目標解決率での生成割合が9.8%から19.6%に上昇しました。これらの結果は、PROPELが目標解決率に向けてタスク生成をシフトさせ、学習可能フロンティアのタスク供給を効果的に増加させることを示しています。

さらに、PROPELの主要な革新は、軽量な活性化プローブの設計にあります。このプローブは、一度の順伝播でソルバーのパス率を予測できるため、従来の各タスクに対する完全なソルバー推論の巨大なオーバーヘッドを回避します。この設計により、生成器の訓練は効率的に実行でき、単一のソルバー推論に数十分を要するソフトウェア工学タスクにも拡張可能です。PROPELのフレームワークは、現在の大規模言語モデルだけでなく、将来のより強力なエージェントシステムにおける訓練データの自動生成にもスケーラブルなソリューションを提供します。

本研究は、自動化された訓練データの生成に新たな道を開き、特に高コストなソルバーシナリオにおいて、PROPELの償却手法は最適化のオーバーヘッドを劇的に削減します。今後、この手法はさらに複雑なタスク領域に拡大され、知的モデルの訓練効率を一層向上させることが期待されます。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • PROPELは一度だけラベル付けされたコーパスを用いて活性化プローブを訓練し、ソルバーのパス率を代理評価する。
  • コードタスクでは、Qwen2.5-3Bソルバーに対する学習可能フロンティアタスクの割合が10.1%から20.0%に増加。
  • ソフトウェア工学タスクでは、未見のリポジトリに対してQwen3.5-27Bソルバーの目標成功率タスクが9.8%から19.6%に向上。

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