遷移の“先読み”がある強化学習:正確な計画はNP困難だが近似と学習は効く
研究の主題は、行動を選ぶ前に「これからℓステップ分、どの状態に行くか」を観察できる強化学習(RL)です。こうした先読み情報は性能を大きく改善できますが、計画問題(最良の行動を決める問題)が計算上どれだけ難しいかは不明な点が残っていました。本論文は二つの問いに答えます。第一に、割引率(将来報酬をどれだけ重視するかを示す数)を任意の固定有理数にしても、正確な最適計画はNP困難であることを示します。第二に、先読みの深さℓが固定であれば、ランダム化された多項式時間の近似スキーム(RPTAS)で高い確率で一様に近似最適な方策を得られることを示します。さらに未知の遷移や確率的な報酬の下でも、楽観主義と分散に応じた確信域を組み合わせて学習アルゴリズムを拡張し、累積後悔(最適方策との差の合計)が古典的なタブラ型割引RLと同位相の主要項で一致することを示しています。要するに、正確解は計算困難でも、現実的な条件下では効率的に近似・学習できる、という結論です。
設定をもう少し平易に説明します。対象は有限の状態と有限の行動を持つマルコフ意思決定過程(MDP)で、報酬を割引して合計する標準的な目標を考えます。先読みとは、現在の時点で「次のℓステップにおける可能な遷移の結果(完全な深さℓの探索木)」を観察できるというものです。実世界の例としては、リアルタイム交通情報が得られるナビゲーション、必要に応じて高精度シミュレータで将来を試せる場面、配送予測が渡されるサプライチェーンなどが挙げられます。先読み情報は有用ですが、観測する情報を状態に組み入れると状態空間が大きく膨らみ、それが計算困難の原因になります。
これまでの研究では、最悪の場合に先読み付き計画がNP困難であることが示されていましたが、その証明は割引率が1に非常に近い場合を使っていました。本稿はその点を解消します。著者らは任意の固定有理の割引率(0より大きく1未満)に対しても、正確な最適計画がNP困難であると証明しました。つまり、割引の度合いを現実的な値に固定しても、一般には全体最適を求めるのは計算上難しい、という堅い負の結果です。ただしこれは「正確に解く」場合の主張であり、近似可能性とは別の話です。
そこで著者らは建設的な正の結果を示します。先読み深さℓが固定であれば、ランダム化多項式時間近似スキーム(RPTAS)を構成できます。これは事前に確率的にサンプルした遷移表を使って近似問題を解き、その解が任意の観測ウィンドウに対して一様に良い性能を持つことを保証する方式です。未知の遷移確率や確率的報酬を扱う場合には、楽観主義(未確定な部分を有利に扱う手法)と分散に応じた信頼区間を組み合わせています。この拡張版アルゴリズムは累積後悔の主要な項が古典的なタブラ型割引RLと一致し、対数因子の差にとどまると報告されています。要するに、正確解の困難性は残るものの、現実的な前提のもとで効率的に「ほぼ最適に行動する」ことは可能です。
本研究の意義は二点あります。まず、追加情報(先読み)があることで計画が容易になるとは限らないことを厳密に示した点です。次に、そのように計算困難な問題でも、先読みの深さを固定すれば実用的な多項式時間アルゴリズムで良好な近似と学習を行えることを示した点です。これは先読み情報を持つ実システム(交通予測やシミュレータ活用など)で、全探索を避けつつ高性能な方策を得る方針を裏付けます。
重要な注意点もあります。結果は有限の表形式MDP(状態と行動が有限)での定常かつ割引尺度の設定に基づきます。RPTASや学習アルゴリズムの多項式時間性は先読み深さℓを固定した場合の話であり、ℓが大きくなると計算は急速に難しくなる可能性があります。また、近似保証は高確率での結果であり、また「正確に最適化することはNP困難」という負の結果は依然として残ります。最後に、ここで参照したのは論文抜粋の範囲内の記述であり、詳細な定理の条件や証明の細部は原論文を参照してください。