ソフト actor-critic (SAC) エージェント

ソフト actor-critic (SAC) アルゴリズムは、モデルフリーのオンライン Off-Policy actor-critic 強化学習手法です。SAC アルゴリズムは、長期的に期待される報酬と、方策のエントロピーの両方を最大化する最適な方策を計算します。方策エントロピーは、特定の状態に関する方策の不確実性の尺度です。エントロピー値が高いほど、より多くの探索が促進されます。累積長期報酬の期待値とエントロピーの両方を最大化することは、環境の利用と探索のバランスを取るのに役立ちます。

さまざまな種類の強化学習エージェントの詳細については、強化学習エージェントを参照してください。

Reinforcement Learning Toolbox™ ソフトウェアでの SAC エージェントの実装では、価値関数の過大推定を防ぐ 2 つの Q 値関数のクリティックが使用されます。SAC アルゴリズムの他の実装では、追加の価値関数クリティックを使用します。

SAC エージェントは、次の観測空間と行動空間をもつ環境で学習させることができます。

観測空間	行動空間
離散または連続	連続

SAC エージェントは次のアクターとクリティックを使用します。

クリティック	アクター
Q 値関数クリティック Q(S,A)。rlQValueFunction を使用して作成	確率的方策アクター π(S)。rlContinuousGaussianActor を使用して作成

学習が実行されている間、SAC エージェントは次を行います。

• 学習時にアクターとクリティックのプロパティを定期的に更新する。

• 連続行動空間のガウス確率分布の平均と標準偏差を推定し、分布に基づいてアクションをランダムに選択する。

• 収益の期待値と方策のエントロピーのバランスを取るエントロピー重み項を更新する。

• リング経験バッファーを使用して過去の経験を保存する。エージェントは、バッファーからランダムにサンプリングされた経験のミニバッチを使用して、アクターとクリティックを更新する。

エージェントの UseExplorationPolicy オプションが false に設定されている場合、最大尤度のアクションが sim および generatePolicyFunction で常に使用されます。その結果、シミュレーションされたエージェントと生成された方策は確定的に動作します。

UseExplorationPolicy が true に設定されている場合、エージェントは確率分布をサンプリングすることによってアクションを選択します。結果として、方策は確率的となり、エージェントはその観測空間を探索します。

このオプションはシミュレーションと展開にのみ影響します。学習には影響しません。

アクター関数近似器およびクリティックの関数近似器

方策と価値関数を推定するために、SAC エージェントは次の関数近似器を維持します。

• 確率的アクター π(A|S;θ) — アクターはパラメーター θ をもち、状態 S にあるときに、各連続行動 A を実行する条件付きガウス確率の平均と標準偏差を出力します。

• 1 つまたは 2 つの Q 値クリティック Q_k(S,A;ϕ_k) — それぞれパラメーター ϕ_k をもつクリティックは、観測値 S とアクション A を入力として受け取り、長期報酬とエントロピーの両方を含む、対応する価値関数の期待値を返します。

• 1 つまたは 2 つのターゲット クリティック Q_tk(S,A;ϕ_tk) — 最適化の安定性を向上させるために、エージェントはターゲット クリティック パラメーター ϕ_tk を最新の対応するクリティック パラメーター値に周期的に設定します。ターゲット クリティックの数はクリティックの数と一致します。

Q₁(S,A;ϕ₁) と Q₂(S,A;ϕ₂) の 2 つのクリティックを使用する場合、各クリティックは異なる構造をもつことができます。クリティックが同じ構造をもつ場合、異なる初期パラメーター値をもたなければなりません。

各クリティック Q_k(S,A;ϕ_k) と対応するターゲット クリティック Q_tk(S,A;ϕ_tk) は、同じ構造とパラメーター表現をもたなければなりません。

関数近似のためのアクターとクリティックの作成の詳細については、Create Policies and Value Functionsを参照してください。

学習が実行されている間、エージェントは θ のパラメーター値を調整します。学習後、パラメーターは調整された値のままになり、学習済みのアクター関数近似器が π(A|S) に保存されます。

アクションの生成

SAC エージェントのアクターは、平均値と標準偏差の出力を生成します。アクションを選択するために、アクターはまずこれらのパラメーターをもつガウス分布から非有界アクションをランダムに選択します。学習が実行されている間、SAC エージェントは非有界の確率分布を使用して、特定の観測値に対する方策のエントロピーを計算します。

SAC エージェントが有界アクションを生成する必要がある場合、アクターは、ガウス分布からサンプリングされた非有界アクションに "tanh" 演算と "スケーリング" 演算を適用することにより、有界アクションを生成します。

エージェントの作成

SAC エージェントの作成および学習は、MATLAB^® コマンド ラインまたは強化学習デザイナー アプリを使用して行うことができます。強化学習デザイナーを使用したエージェントの作成の詳細については、Create Agents Using Reinforcement Learning Designerを参照してください。

コマンド ラインで、環境の観測仕様とアクション仕様に基づいて既定のアクターとクリティックを使用して、SAC エージェントを作成できます。これを行うには、次の手順を実行します。

1. 環境用の観測仕様を作成します。既に環境インターフェイス オブジェクトがある場合は、getObservationInfo を使用してこの仕様を取得できます。

2. 環境用のアクション仕様を作成します。既に環境インターフェイス オブジェクトがある場合は、getActionInfo を使用してこの仕様を取得できます。

3. 必要に応じて、それぞれの学習可能な層のニューロンの数、または再帰型ニューラル ネットワークを使用するかどうかを指定します。これを行うには、rlAgentInitializationOptions を使用してエージェント初期化オプション オブジェクトを作成します。

4. 必要に応じて、rlSACAgentOptions オブジェクトを使用してエージェント オプションを指定します。

5. rlSACAgent オブジェクトを使用してエージェントを作成します。

あるいは、アクターとクリティックを作成し、それらのオブジェクトを使用してエージェントを作成することもできます。この場合、アクターおよびクリティックの入力と出力の次元が、その環境の対応するアクション仕様および観測仕様と確実に一致するようにしてください。

1. rlContinuousGaussianActor オブジェクトを使用して確率的アクターを作成します。SAC エージェントの場合、平均値を目的のアクション範囲に適切にスケーリングするために、アクター ネットワークには、平均値の出力パスの最後の 2 つの層として tanhLayer と scalingLayer が含まれていてはなりません。ただし、標準偏差値が確実に非負となるように、アクター ネットワークには、標準偏差値の出力パスの最後の層として reluLayer を含めなければなりません。

2. rlQValueFunction オブジェクトを使用して 1 つまたは 2 つのクリティックを作成します。

3. rlSACAgentOptions オブジェクトを使用してエージェント オプションを指定します。

4. rlSACAgent オブジェクトを使用してエージェントを作成します。

関数近似のためのアクターとクリティックの作成の詳細については、Create Policies and Value Functionsを参照してください。

学習アルゴリズム

SAC エージェントは以下の学習アルゴリズムを使用し、アクターとクリティックのモデルとエントロピーの重みを定期的に更新します。学習アルゴリズムを構成するには、rlSACAgentOptions オブジェクトを使用してオプションを指定します。ここで、K = 2 はクリティックの数、k はクリティックのインデックスです。

• 各クリティック Q_k(S,A;ϕ_k) をランダムなパラメーター値 ϕ_k で初期化し、同じランダムなパラメーター値で各ターゲット クリティックを初期化する ($${\varphi }_{tk}={\varphi }_k$$)。

• アクター π(S;θ) をランダムなパラメーター値 θ で初期化する。

• π(S) の初期ランダム方策に従って一連のアクションを実行して、ウォーム スタートを実行する。アクションごとに、経験を経験バッファーに保存する。ウォームアップ アクションの数を指定するには、NumWarmStartSteps オプションを使用する。

• 各学習タイム ステップについて、次のようにする。

  1. 現在の観測値 S に対して、π(S;θ) の方策を使用してアクション A を選択します。

  2. アクション A を実行します。報酬 R と次の観測値 S' を観測します。

  3. 経験 (S,A,R,S') を経験バッファーに保存します。

  4. 経験バッファーから M 個の経験 (S_i,A_i,R_i,S'_i) のランダムなミニバッチをサンプリングします。M を指定するには、MiniBatchSize オプションを使用します。

  5. D_C タイム ステップごとに、すべてのサンプリングされた経験にわたる損失 L_k を最小化して、各クリティック パラメーターを更新します。D_C を指定するには、LearningFrequency オプションを使用します。

     $$L_k=\frac{1}{2M}{\displaystyle \sum _{i=1}^M{\left(y_i-Q_k\left(S_i,A_i;{\varphi }_k\right)\right)}^2}$$

     S'_i が終了状態の場合、価値関数のターゲット y_i は経験報酬 R_i に等しくなります。それ以外の場合、価値関数のターゲットは、R_i、クリティックからの将来の最小割引報酬、および重み付きエントロピーの合計になります。

     $$y_i=R_i+\gamma *\underset{k}{\min }\left(Q_{tk}\left(S_i\text{'},A_i\text{'};{\varphi }_{tk}\right)\right)-\alpha \ln \pi \left(S_i\text{'};\theta \right)$$

     ここで、以下となります。

     • A'_i は、アクター π(S'_i) の非有界出力から派生させた有界アクションです。

     • γ は割引率であり、DiscountFactor オプションを使用して指定します。

     • $$-\alpha \ln \pi \left(S;\theta \right)$$ は、状態 S にあるときのアクターの出力範囲の重み付き方策エントロピーです。α は、EntropyLossWeight オプションを使用して指定するエントロピー損失の重みです。

     NumStepsToLookAhead の値を N に等しいと指定した場合、N ステップの収益 (次の N ステップの報酬と、N 番目の報酬を発生させた状態の割引推定値を加算したもの) がターゲット y_i の計算に使用されます。

  6. D_A タイム ステップごとに、次の目的関数を最小化して、アクターのパラメーターを更新します。D_A を設定するには、LearningFrequency オプションと PolicyUpdateFrequency オプションの両方を使用します。

     $$J_{\pi }=\frac{1}{M}{\displaystyle \sum _{i=1}^M\left(-\underset{k}{\min }\left(Q_k\left(S_i,A_i;{\varphi }_k\right)\right)+\alpha \ln \pi \left(S_i;\theta \right)\right)}$$

  7. D_A タイム ステップごとに、次の損失関数を最小化して、エントロピーの重みも更新します。

     $$L_{\alpha }=\frac{1}{M}{\displaystyle \sum _{i=1}^M\left(-\alpha \ln \pi \left(S_i;\theta \right)-\alpha \mathscr{H}\right)}$$

     ここで、ℋ はターゲット エントロピーであり、EntropyWeightOptions.TargetEntropy オプションを使用して指定します。

  8. D_T ステップごとに、ターゲットの更新方法に応じてターゲット クリティックを更新します。D_T を指定するには、TargetUpdateFrequency オプションを使用します。詳細については、ターゲットの更新方法を参照してください。

  9. ステップ 4 ～ 8 を N_G 回繰り返します。ここで、N_G は勾配ステップの数であり、NumGradientStepsPerUpdate オプションを使用して指定します。

ターゲットの更新方法

SAC エージェントは、以下のターゲット更新方法のいずれかを使用して、ターゲットとなるクリティックのパラメーターを更新します。

• 平滑化 — 平滑化係数 τ を使用して、タイム ステップごとにターゲット クリティック パラメーターを更新します。平滑化係数を指定するには、TargetSmoothFactor オプションを使用します。

  $${\varphi }_{tk}=\tau {\varphi }_k+\left(1-\tau \right){\varphi }_{tk}$$

• 周期的 — 平滑化なしでターゲット クリティック パラメーターを周期的に更新します (TargetSmoothFactor = 1)。更新期間を指定するには、TargetUpdateFrequency パラメーターを使用します。

  $${\varphi }_{tk}={\varphi }_k$$

• 周期的平滑化 — 平滑化を使用してターゲット パラメーターを周期的に更新します。

ターゲットの更新方法を構成するには、rlSACAgentOptions オブジェクトを作成し、次の表に示すように TargetUpdateFrequency パラメーターと TargetSmoothFactor パラメーターを設定します。

更新方法	TargetUpdateFrequency	TargetSmoothFactor
平滑化 (既定)	1	1 未満
周期的	1 より大きい	1
周期的平滑化	1 より大きい	1 未満

参照

[1] Haarnoja, Tuomas, Aurick Zhou, Kristian Hartikainen, George Tucker, Sehoon Ha, Jie Tan, Vikash Kumar, et al. "Soft Actor-Critic Algorithms and Application." Preprint, submitted January 29, 2019. https://arxiv.org/abs/1812.05905.

参考

オブジェクト

• rlSACAgent | rlSACAgentOptions

関連する例

• Train Reinforcement Learning Agents

詳細

• 強化学習エージェント
• Create Policies and Value Functions