HomeLocal LLM / Open ModelsACRL:訓練-推論エンジン乖離の適応制御でFP8量子化下のRL学習を安定化
ACRL:訓練-推論エンジン乖離の適応制御でFP8量子化下のRL学習を安定化

ACRL:訓練-推論エンジン乖離の適応制御でFP8量子化下のRL学習を安定化Huawei's ACRL framework monitors the discrepancy between training…

AI要点サマリSummary highlight

HuaweiのACRLは、LLMのRL学習でFSDP/vLLM間の精度差(BF16 vs FP8)による実質的なoff-policy化を適応的に補正し、トークン単位の勾配重み調整でBF16基線を上回る精度をわずか0.1%のオーバーヘッドで実現する。

Huawei's ACRL framework monitors the discrepancy between training (FSDP/Megatron) and inference (vLLM/SGLang) engines caused by FP8 vs BF16 precision gaps, then adjusts per-token gradient weights to prevent training collapse while outperforming BF16 baselines across 3B–32B Dense and MoE models.

要約と収集メタデータをもとに生成した AI 解説本文です。元記事全文の転載・翻訳ではありません。This AI explainer is generated from the summaries and collected metadata, not from a reproduction or translation of the full source article.

大規模言語モデル(LLM)の強化学習(RL)では、学習と推論で異なるソフトウェアと数値精度を使うことに起因する「訓練-推論乖離」が、学習崩壊の一因になりうる。Huaweiが提案したACRLは、この乖離を適応的に監視して補正し、FP8量子化の下でも安定した学習を可能にするフレームワークだとされる。

背景を整理すると、近年のLLMのRLでは、勾配を計算する訓練エンジン(FSDPやMegatron)と、応答を生成する推論エンジン(vLLMやSGLang)を別々に用いる構成が一般的だ。両者は実装が異なるうえ、訓練側がBF16、推論側がFP8といったように数値精度も食い違うことがある。その結果、本来は最新の方策で生成したデータを使う「on-policy」学習のはずが、実質的には古い方策を用いる「off-policy」に近い状態になり、学習が不安定化すると説明されている。

ACRLはこの問題に対し、トークンごとに訓練と推論の乖離度を測り、その大きさに応じて勾配の重みを調整するという。乖離が大きいトークンの影響を抑えることで、崩壊を避けつつ学習を進める狙いと見られる。

Huaweiによれば、ACRLはFP8量子化を用いながらBF16を基線とする精度を上回り、追加の計算オーバーヘッドはわずか0.1%にとどまるという。検証はGRPOPPO・DAPOの3アルゴリズム、3B〜32Bの4モデル、DenseとMoEの双方にわたるとされる。

FP8はメモリ使用量と計算コストを抑えられるため、推論の高速化・低コスト化の手段として注目が集まっている。一方で、量子化に伴う精度低下がRL学習を不安定にする懸念は以前から指摘されてきた。ACRLのようなアルゴリズム側での補正は、量子化推論とRL学習を両立させる一つの方向性を示す可能性がある。もっとも、これは提案段階の報告であり、実運用での効果や他手法との比較については、今後の検証が求められるだろう。

Reinforcement learning has become one of the primary levers for improving large language models, yet a subtle piece of infrastructure plumbing can quietly undermine it. Huawei researchers have proposed a framework called ACRL that tackles the discrepancy between the engine used to train a model and the engine used to generate its training samples. The problem matters because when these two systems disagree at the numerical level, a run that is meant to be on-policy becomes effectively off-policy, and that mismatch can escalate into full training collapse.

The issue stems from how contemporary RL pipelines for LLMs are assembled. Gradient updates are typically handled by training frameworks such as FSDP or Megatron, while the rollouts—the model's own generated responses that the algorithm learns from—are produced by dedicated high-throughput inference engines like vLLM or SGLang. These two halves use different implementations and, increasingly, different numerical precisions. A common configuration generates rollouts in FP8 for speed while computing gradients in BF16. As a result, the probability distribution the inference engine actually sampled from is not quite the distribution the trainer assumes it sampled from.

That gap is consequential for on-policy algorithms. Methods such as PPO assume the data being learned from came from the current policy, so any divergence between the sampling and training distributions miscalibrates the implicit importance weights in the update. The gradients become biased or excessively noisy, and over many steps that error can compound until the model's performance degrades sharply. FP8 quantization makes the effect worse because its lower precision widens the numerical distance between the inference and training passes.

ACRL's response is to measure the discrepancy as it happens and compensate at fine granularity. According to the description, the framework adaptively monitors the degree of divergence and adjusts per-token gradient weights, reweighting tokens where the inference and training probabilities diverge most. This is broadly in the spirit of importance-

  • 出典SourceZenn LLMコミュニティCommunity
  • 直近30件の平均重要度Avg importance, last 301=Info · 2=Medium · 3=High
  • 配信形式FormatブログBlog
  • 重要度Importance重要度 MediumMedium priority(Local LLM / Open Models 230件中、同等以上 207件)(207 of 230 Local LLM / Open Models entries are equal or higher)
  • 情報の寿命Half-life📘 中期 (チュートリアル)Medium-term (tutorial)
  • 原文言語Source languageJA
  • 収集日時Collected2026/07/31 05:10

本ページの本文と要約は AI による自動生成です。日本語版と英語版は言語ごとに独立して生成されるため、表現や詳しさが異なる場合があります。正確性は元記事 (zenn.dev) をご確認ください。The body and summaries are AI-generated independently for each language, so wording and detail may differ. Verify accuracy at the original source (zenn.dev).

🏠Local LLM / Open Models の他の記事More from Local LLM / Open Modelsもっと見る →View more →