数行のコードで強化学習を実装する

強化学習は過去 1 年間で大きな進歩を遂げ、最先端のテクノロジが 2 か月ごとにリリースされています。 AlphaGo が世界チャンピオンの囲碁プレイヤー、柯潔に勝利したり、マルチエージェントがかくれんぼをしたり、さらには AlphaStar が StarCraft で健闘したりするのを見てきました。

これらのアルゴリズムを実装するには、ディープラーニングと強化学習の両方を十分に理解する必要があるため、非常に困難になる可能性があります。この記事の目的は、強化学習の学習を簡単に開始できるように、いくつかの便利なパッケージをすぐに使い始めることができるようにすることです。

SOTA 深層強化学習アルゴリズムを実装する方法についての詳細なチュートリアルについては、こちらをご覧ください。ぜひ注意深く読んでみてください！

1. 環境

これらのアルゴリズムの実装を始める前に、まず作業環境、つまりゲームを作成する必要があります。アルゴリズムでは、アクション空間と観察空間が何であるかを理解することが重要です。この目的のために、興味深い環境を選択するために使用できるいくつかのソフトウェアパッケージを紹介します。

1. ジム

Gym は、強化学習アルゴリズムを開発および比較するためのツールキットです。環境を操作するための使いやすいインターフェースを提供するため、実験や研究の目的でよく使用されます。

次のコマンドを使用してパッケージをインストールするだけです。

 pip インストール ジム

その後、次のコードを使用して環境を作成できます。

ジムをインポート
env = gym.make ('CartPole-v0')

CartPole 環境では、カートに先端が取り付けられたポールが落ちないようにすることがタスクです。

env 変数には、環境 (ゲーム) に関する情報が含まれます。 CartPole のアクション空間が何であるかを調べるには、env.action_space を実行して Discrete(2) を生成します。これは、2 つの個別のアクションが可能であることを意味します。観測空間を表示するにはenv.observation_spaceを実行し、Box(4)を生成します。このボックスはn(4)個の閉区間の直積を表します。

ゲームをレンダリングするには、次のコードを実行します。

ジムをインポート
env = gym.make ('CartPole-v0') 
 
 obs = env .reset()
真の場合:
アクション= env .action_space.sample()
    obs、rewards、done、 info = env .step(action)
    env.render() 
 
    完了した場合:
        壊す

ランダムなアクションを選択した場合、カートは失敗し続けることがわかります。最終的な目標は、この問題を解決する方法を学習する強化学習アルゴリズムを実行することです。

Gym の環境の完全なリストについては、こちらをご覧ください。

注: Atari ゲームの実行中に問題が発生した場合は、https://github.com/openai/gym/issues/1726 を参照してください。

2. レトロ

興味深い環境を作成するためのもう 1 つのオプションは、Retro を使用することです。 OpenAI によって開発されたこのパッケージを使用すると、ROM を使用して Airstriker-Genesis などのゲームをエミュレートできます。

次のコマンドを使用してパッケージをインストールするだけです。

 pip インストール ジム レトロ

次に、以下を使用して環境を作成および表示できます。

レトロを輸入
env = retro .make(ゲーム= 'Airstriker-Genesis' )

同様に、ゲームをレンダリングするには、次のコードを実行します。

レトロを輸入
env = retro .make(ゲーム= 'Airstriker-Genesis' ) 
 
 obs = env .reset()
真の場合:
アクション= env .action_space.sample()
    obs、rewards、done、 info = env .step(action)
    env.render() 
 
    完了した場合:
        壊す

ROM をインストールするには、対応する .sha ファイルを見つけて、次のコマンドを実行する必要があります。

 python3 -m retro.import /path/to/your/ROMs/directory/

注: 簡単に利用できる環境の完全なリストについては、次のコマンドを実行してください。

レトロデータリスト()

3. プロジェクション

強化学習の典型的な問題は、結果として得られるアルゴリズムが特定の環境ではうまく機能することが多いものの、一般化可能なスキルを学習できないことです。たとえば、ゲームの見た目や敵の反応を変えたい場合はどうすればよいでしょうか?

この問題に対処するために、OpenAI は手続き的に生成された環境の作成を可能にする Procgen というパッケージを開発しました。このパッケージを使用すると、強化学習エージェントが一般的なスキルをどれだけ早く学習するかを測定できます。

ゲームのレンダリングは非常に簡単です:

ジムをインポート
パラメータ= {"num_levels": 1、"distribution_mode": "hard"}
 env = gym .make("procgen:procgen-leaper-v0", **param) 
 
 obs = env .reset()
真の場合:
アクション= env .action_space.sample()
    obs、rewards、done、 info = env .step(action)
    env.render() 
 
    完了した場合:
        壊す

これにより、アルゴリズムをトレーニングできる単一のレベルが生成されます。同じ環境のさまざまなバージョンを手順的に生成するには、いくつかのオプションがあります。

num_levels - 生成できる一意のレベル数
distribution_mode - 使用する変数のレベル。オプションは「easy」、「hard」、「extreme」、「memory」、「exploration」です。すべてのゲームは「イージー」と「ハード」をサポートしていますが、その他のオプションはゲームによって異なります。

2. 強化学習

さて、いよいよ実際の強化学習をやってみましょう。トレーニングアルゴリズムに使用できるパッケージは多数ありますが、ここでは実装が堅牢な Stable Baselines に焦点を当てます。

この投稿では、RL アルゴリズムが実際にどのように機能するかについては説明しません。そのためには、まったく新しい投稿が必要になります。 PPO、SAC、TD3 などの最先端のアルゴリズムの概要については、以下を参照してください。

https://github.com/dennybritz/reinforcement-learning.

1. 安定したベースライン

安定ベースライン (SB) は OpenAI ベンチマークに基づいており、研究コミュニティと業界が新しいアイデアをより簡単に複製、改善、特定できるように設計されています。彼らはベースラインを改良し、より安定したシンプルなツールにして、初心者が実装の詳細に煩わされることなく強化学習を試すことができるようにしました。

SB は、最先端の強化学習アルゴリズムを簡単かつ迅速に適用できるため、よく使用されます。さらに、RL モデルの作成とトレーニングには、数行のコードだけが必要です。

インストールは、pip install stable-baselines で簡単に実行できます。次に、RL モデル (例: PPO2) を作成して学習するには、次のコード行を実行します。

 stable_baselines から PPO2 をインポート
stable_baselines.common.policies から MlpPolicy をインポートします
モデル= PPO2 (MlpPolicy、env、 verbose = 1 )
モデル.learn(合計時間ステップ= 10_000 、ログ間隔= 10 )

いくつかの点については説明が必要かもしれません:

total_timesteps: トレーニングするサンプルの総数
MlpPolicy: アクター・クリティックを実装するポリシーオブジェクト。この場合、64 層の多層パーセプトロンが 2 つ使用されます。 CnnPolicyやCnnLstmPolicyなどの視覚情報ポリシーもあります。

このモデルを CartPole の例に適用するには、環境を Dummy でラップして SB で利用できるようにする必要があります。次に、CartPole 環境で PPO2 をトレーニングする完全な例を示します。

 stable_baselines.common.policies から MlpPolicy をインポートします
stable_baselines.common.vec_env から DummyVecEnv をインポートします
stable_baselines から PPO2 をインポート
ジムをインポート
 
 env = gym.make ('CartPole-v0')
 env =ダミーベクトルエンベロープ([lambda: env]) 
 
モデル= PPO2 (MlpPolicy、env、 verbose = 1 )
モデル.learn(合計時間ステップ= 50_000 、ログ間隔= 10 ) 
 
 obs = env .reset()
真の場合:
    アクション、 _states = model.predict (obs)
    obs、rewards、dones、 info = env .step(action)
    env.render()