英語記事
FlexFloor:専門家の弾性並列化で強化学習の後学習を効率化
見出しと要約は、英語の元記事をもとに当サイトが日本語で作成しています。
原題:Resource-Efficient Reinforcement Learning Post-Training through Elastic Expert Parallelism
三行要約
- 強化学習による後学習は大規模言語モデルの能力を高めるが、学習用レスポンスの生成が大きなボトルネックになっている。
- 本論文は、同期処理中のリソースを早期に解放できるシステム「FlexFloor」を提案している。
記事の概要
強化学習による後学習は大規模言語モデルの能力を高めるが、学習用レスポンスの生成が計算上の大きなボトルネックになっている。Mixture-of-Expertsモデルでは、割り当てられたレスポンスを生成し終えたワーカーも、その専門家がより長いレスポンスを処理するワーカーにまだ必要とされるため、占有されたままになることがある。ベースラインでは、占有されたワーカー時間の約3分の1が、ローカルでのレスポンス生成が終わった後に発生している。本論文が提案するFlexFloorは、同期処理の最中にリソースをより早く解放できる仕組みで、専門家の弾性並列化によって後学習の効率化を図る。