Research / AIエージェント · 基盤技術

報酬と探索を同時に考える Soft Actor-Critic

行動の報酬だけでなく、方針の多様さも目的に加える強化学習法。連続した動作を選ぶゲーム型の課題で既存法と比べ、学習速度と乱数による成績の揺れを調べた。報酬の高い試行だけでなく、複数回のばらつきも記事で確認する。

論文初公開(arXiv)
参照版
v2
TRAIL公開
  • Soft Actor-Critic(SAC)は、報酬と行動の多様さを同時に重視する。
  • 連続した動作を選ぶ複数課題で、既存法との学習速度と最終成績を比較した。
  • 高得点の一回だけではなく、乱数を変えた複数試行の安定性が重要だ。

何を確かめた研究か

強化学習では、今まで高得点だった行動だけを繰り返すと、新しい良い行動を見つけにくい。SACの論文は、期待される報酬と、選べる行動の多様さを示すエントロピーを同時に最大化する方法を提案した。過去の経験を再利用するoff-policy学習と、確率的な行動方針を組み合わせる。

どう検証したか

制御器が小刻みに連続値の動作を選ぶHopper、Walker2d、Ant、Humanoidなどのシミュレーションで試した。DDPG、PPO、SQLなどと比べ、環境を何ステップ使ってどれだけの報酬を得たかを曲線で示す。五つの乱数条件で実行し、平均だけでなく最小・最大の広がりを見る。確率的な方針からエントロピー目的を外した比較も行い、多様さが安定性にどう関係するかを調べた。

確率的な方針は、同じ状態でも複数の行動を選び得る。探索の早い段階で一つの行動に固まり過ぎないようにする一方、報酬の高い行動へ学習を進める。SACの「soft」はこの報酬とエントロピーを同時に考える目的に由来する。ランダムに動くだけの方針とは違う。

何が分かったか

図1ではSACは容易な課題で既存法と同程度、難しい課題では学習速度や最終報酬で優位な例が多かった。特にDDPGがAntやHumanoidで進まない条件があり、SACは学習を続けられた。図2のHumanoid比較では、確率的な方針の方が、決定的な方針へ変えた条件より乱数を変えた試行間の揺れが小さかった。著者らは、エントロピーを含む探索が早い段階で行動を狭め過ぎるのを防ぐと解釈する。

図1の曲線は、試した環境ステップ数に対する評価時の報酬を表す。五つの乱数条件の平均と幅を見れば、一つの条件だけで偶然良かった結果を見分けやすい。図2は特定のHumanoid課題における確率的・決定的方針の比較で、そこでの安定性を全課題の保証としない。

限界は何か

主な証拠はシミュレーション内の連続制御で、現実のロボットの安全性を測ったわけではない。基準法との比較は実装、探索範囲、ハイパーパラメータの設定に依存する。エントロピーの強さにも感度があり、行動を多様にすれば常に得点が上がるわけではない。五つの乱数条件は安定性を調べる一歩だが、あらゆる環境の変化を表さない。

仕事との接点

行動を選ぶAIの試験では、最高の試行だけでなく、一定の経験予算での平均、ばらつき、失敗例を残す。探索的な行動が許される範囲も用途に合わせて決める。これは論文の評価方法から導く編集部の提案で、シミュレーションの報酬を実機の安全性へ読み替えない。

01

出典

Research の一覧へ戻る