AISubHub/AI ニュース
開発者

詳しい解説 · 読了目安 9 分

AWSの複数ターンRLガイド:報酬設計、デコード条件、総費用まで含めて結果を読む

SageMaker AIでWordleを学習するgpt-oss-20bの改善を報告した技術ガイドは、評価設定、報酬の失敗、AgentCore費用を分けて見る必要も示す。

当サイトの公開日
原典の公開日
出典確認日

AWSはSageMaker AIでの複数ターン強化学習について、実装を詳しく扱う続編ガイドを公開した。10月23日の技術記事はgpt-oss-20bにWordleを学習させ、報酬設計の失敗、運用問題、評価条件を説明する。既存サービスに関する新たな実装上の証拠であり、学習セッションの一般提供や新しい基盤モデルを再発表したものではない。

ニュースを聴く

この記事を読み上げるか、ニュースを続けて聴けます。

AWSの複数ターンRLガイド:報酬設計、デコード条件、総費用まで含めて結果を読む

読み上げを準備中…

端末の音声

読み上げはブラウザーまたは端末が処理します。AISubHub は文章を音声 API に送りません。利用できる音声やロック画面での再生は端末によって異なります。

再開時に短い部分を繰り返す場合があります。

要点

  • 未使用の評価用128単語で、AWSは推論努力medium・温度1.0の正解率61%→84%を報告する。同じ温度でlowにすると15%→69%となる。
  • 100ステップの実行は12h40mで105,486ロールアウトを処理し、モデルのprefill・生成・学習の3区分のトークン費用は合計$338.50。AgentCoreの約$79は別にかかる。
  • 重み付けを誤った報酬が遅い解答を優遇した経緯と、学習中の検証終点37%が提供時の評価84%とは異なる理由を示す。

学習対象は模範回答ではなく、エピソード全体

複数段階で動くエージェントには、回答例の模倣だけでなく意思決定の連続に対するフィードバックが必要になる。AWSの構成ではSageMaker AIが学習ジョブと方策モデルを管理し、AgentCoreランタイムにStrandsエージェント、環境、採点を配置する。エージェントはモデルに候補を求め、Wordleの手掛かりを受け取り、終了時に報酬を1つ返す。GRPOは同じ単語の複数ロールアウトを比較し、LoRAアダプターを更新する。5文字と6回の推測で採点しやすい教材であり、本番業務の有効性を示すものではない。

例では学習用640単語と未使用の評価用128単語を分ける。各学習行は環境に秘密の単語を渡すが、方策への正解ヒントにはしない。目標回答やラベル付き軌跡はない。推論努力low・温度1.0で、100ステップ、各ステップ128単語、単語ごとに8ゲームを使う。この分離が重要なのは、機能する報酬と環境が学習信号になる一方、その用意は顧客の責任だからである。

報酬が上がっても、狙った行動は悪化し得る

最初の採点は推測のたびにボーナスを加算した。その結果、敗北が+0.49、初回正解が+0.99、3回目の正解が+1.61となり、良い結果より長い経路を優遇し得た。最終設計はエピソードの成否を優先する。正解時の結果報酬は初回の1.55から6回目の1.05までで、敗北は−0.5、有効な推測がない場合は−1.5。手掛かりとの整合性、反復、情報利得は小さな整形報酬として加える。

整形値は−1から1に制限し、正解ゲームは0.04、それ以外は0.15で重み付けする。以前は正解時も0.15だったため、なお遅い完了を優遇し得た。重みを小さくすると隣接する成功範囲が分離され、最も低い正解報酬も成功閾値を上回る。AISubHubの解釈では、実軌跡を調べ、報酬の順位を意図した行動と照合することが運用上の教訓である。平均報酬の上昇だけでは正しい目的を学んだと証明できない。

大きな改善値は推論努力とサンプリングの条件に依存する

評価用128単語で、推論努力medium・温度1.0の場合、基礎モデルは61%、学習後は84%を解き、差は23パーセントポイントとなる。84%は80%と88%の2実行の平均である。low・温度1.0では15%→69%で54ポイント差となる。一方、温度0の図はlowで2%→37%、mediumで16%→30%を示す。これらは4つの異なる条件であり、普遍的な改善を表す交換可能な値ではない。

学習中の検証はlowとサービス固定の温度0を使う。正解率は90ステップ時に48%で最高となり、終点は37%。提供時の見出し値はmediumとサンプリングを使うため、原文も差をデコード設定に結び付けている。著者らの評価ではlowのゲーム当たり生成トークンはmediumより63–83%少ないが、プロジェクト総費用も同じだけ減る保証ではない。評価集合は小さく、任意のツール、言語、業務についての結果をAWSが示したわけでもない。

学習トークンだけでは請求全体にならない

報告された100ステップの実行時間は12h40mで、105,486ロールアウトを処理した。表にはprefill 491,616,806トークンを100万当たり$0.12で$58.99、生成159,737,098トークンを$0.30で$47.92、学習643,308,554トークンを$0.36で$231.59と記載する。合計は$338.50、丸めて約$339。記事執筆時のus-east-1におけるGPT-OSS-20Bの表示料金である。当サイトは原文の$記号を保ち、通貨を独立に確定したり、現在の全地域共通見積もりとして扱ったりしない。

AgentCoreのCPUとメモリは別に約$79、評価ジョブは約$1.20を加える。Prefillには会話履歴の再読み込みが、生成トークンには推論が含まれ、列挙された最大費目は逆伝播を伴う学習である。長い軌跡は複数の計量項目に影響する。AgentCore、MLflow、S3のリソースが学習後も残り得る点も注意される。AISubHubは有用な会計例と見るが、請求書を検証しておらず、残存リソースや後段の配備まで含む総費用を確定してはいない。

ランタイム障害は妥当な実験も失敗させ得る

ガイドは正解率の数字だけでは見えない障害を記録する。ジョブのロールには信頼プリンシパルjob.sagemaker.amazonaws.comと、AssumeRole・TagSessionの両権限が必要になる。ハンドラーのデコレーターだけではランタイムサーバーは起動せず、例ではBedrockAgentCoreAppと最も外側のentrypointデコレーターが要る。データ、ランタイム、ジョブは同一リージョンを使う。学習中のランタイム再配備はロールアウトを中断させ、ジョブを失敗させ得る。

32並列ロールアウトではキャッシュ済み認証情報の期限切れも表面化し、著者らは事前更新と認可失敗後の再署名を説明する。トークン予算にツール呼び出し分の余地がなくなる場合や、学習率が大きすぎて構造化呼び出しが通常テキストになる場合もあった。対処は実装上の知見であり信頼性保証ではない。当サイトは掲載コードを読んだが、依存関係の導入、AWS API呼び出し、スモークテスト、アダプター学習、修正の再現は行っていない。

Wordleの手順から他の仕事へ移せるもの

モデル表は執筆時点に限定される。LoRAを用い、Nova2LiteとGPT-OSS-20Bはus-east-1・us-west-2、指示調整済みGemma4 31BとQwen3.6 27Bはus-west-2とされる。全モデル、全リージョン、全アカウントへの対応を示さない。続編ガイドはこれらのモデルや、以前別途扱われた学習セッションサービスの提供開始を再発表したものでもない。

他の仕事に移せる設計は、課題の成否、整形報酬の誘因、提供時の設定、リソース総費用を別々に点検することだ。業務フローで利益を主張するなら、その環境、根拠のある報酬、代表性のある未使用評価が必要になる。当サイトは記事全体、全表、必要な静止図7点を読んだ。リンク先文書、論文、リポジトリ、ライセンスは監査せず、性能、料金、安全性の独立テストも行っていない。結果はAWSが報告した限定的なWordle実験のままである。

注目する理由

ガイドは複数ターン学習の点検方法を具体化する。報酬の順序を試し、同一デコード条件で比較し、モデルの処理だけでなくエージェントの実行環境も予算に入れる。他の業務を改善するかは、その業務で測定する必要がある。

制約と提供状況

AWSが報告したのは評価用128単語による単一のWordle実験で、主要な学習後正解率は2実行の平均である。記事全体と静止図7点を確認したが、外部文書、リポジトリ、ライセンスは未監査。学習、API、請求、配備のテストはせず、料金とモデル・リージョン条件は原文時点の記述である。

出典と根拠

AWS Artificial Intelligence Blog — Daniel Quang, Hadrien Almela and Vinay Bhat

原典の公開日: · 出典確認日:

AI を活用した編集記事を、記載した一次情報で確認しています。企業の主張は出典を明示し、掲載は推奨を意味しません。