ベルマン不等式を使い標的ネットワークなしで学習
Lifted Bellman Linear Programming for Offline Reinforcement Learning
この論文をやさしく読む
ひとことで言うと
記録済みの行動データだけから学ぶ強化学習で、価値を目標値に近づける回帰の代わりに、ベルマン不等式を満たすよう学習する方法です。
何に役立つ?
標的ネットワークやEMA更新を使わずにオフライン強化学習を行う選択肢になります。OGBenchでは比較対象中でパラメータ数とピークGPUメモリが最少でした。
この研究の面白いところ
複数ステップの軌道情報を、不等式の下界として使うことでオフポリシー補正なしに取り込めます。線形計画の理論からニューラルネットワークの学習法までつなげています。
どこまで分かった?
厳密な緩和解の一致は表形式の場合です。解の範囲や切り離した更新の停留点に関する結果には決定論的ダイナミクスなどの条件があります。OGBenchの比較結果を全課題の優位性とはしていません。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
オフライン強化学習では通常、指数移動平均(EMA)で更新する標的ネットワークによって安定化した、ブートストラップによる価値の標的に対し、回帰損失を最小化してクリティックを訓練する。複数ステップの標的には行動方策の行動が入るため、オフポリシー補正が必要になる。本研究では代わりに、不等式制約を通じてクリティックへ標本内のベルマン最適性を課す。 ベルマン最適性の線形計画による特徴付けを(Q,V)の共同空間へ持ち上げ、すべての制約がデータセット内の状態・行動対だけを含む、Lifted Bellman Linear Program(LBLP)を定式化する。その一意な最小化解は標本内で最適な対であり、データセットの軌道のKステップ区間に沿う制約を加えても、ロールアウト方策と期間が何であれ、この最小化解は変わらない。ダイナミクスが決定論的なら、この解はデータセット中の最良のリターンと最適価値の間にある。制約をヒンジ罰則に緩和すると、表形式の場合には有限の罰則係数を超えたところで同じ解が得られる。 Approximate Lifted Bellman Unconstrained Minimization(ALBUM)は、この緩和をニューラルネットワークで実装し、勾配停止によってKステップのロールアウト標的を切り離す。目的関数にはブートストラップ標的への二乗回帰がないため、標的ネットワークやEMA更新なしで訓練できる。決定論的ダイナミクスの下では、γとKに依存しない係数条件により、LBLPの解はこの切り離した更新の停留点となる。また、不等式制約によって、オフポリシー補正や行動のチャンク化なしに、データセットの軌道に沿う割引リターンを下界として使える。OGBenchでは、ALBUMはガウス方策と単一のクリティックを使い、FQLと同等の平均性能を達成し、近年の行動チャンク化手法にも匹敵した。同時に、比較した全手法の中でパラメータ数とGPUメモリのピーク使用量が最少だった。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-21(UTC)
- 最新改訂
- 2026-09-21 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-21 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Offline reinforcement learning (RL) typically trains a critic by minimizing a regression loss against bootstrapped value targets stabilized by target networks with exponential moving average (EMA) updates. Multi-step targets incorporate behavior-policy actions and therefore require off-policy correction. We instead impose in-sample Bellman optimality on the critic through inequality constraints. We formulate the Lifted Bellman Linear Program (LBLP), which lifts the linear programming characterization of Bellman optimality to the joint $(Q,V)$ space so that every constraint involves only state-action pairs in the dataset. Its unique minimizer is the in-sample optimal pair, and constraints along $K$-step segments of dataset trajectories leave this minimizer unchanged for any rollout policy and horizon. Under deterministic dynamics, this minimizer lies between the best dataset return and the optimal value. Relaxing the constraints into hinge penalties recovers the same solution above a finite penalty coefficient in the tabular case. Approximate Lifted Bellman Unconstrained Minimization (ALBUM) implements this relaxation with neural networks and detaches the $K$-step rollout targets by stop gradient. Its objective contains no squared regression onto bootstrapped targets, so it can be trained without target networks or EMA updates. Under deterministic dynamics, the LBLP solution is a stationary point of the detached update under a coefficient condition independent of $\gamma$ and $K$, and the inequality constraints allow discounted returns along dataset trajectories to serve as lower bounds without off-policy correction or action chunking. On OGBench, ALBUM uses a single critic with a Gaussian policy, matches the average performance of FQL, and is comparable to recent action-chunking methods, while using the fewest parameters and the least peak GPU memory among all compared methods.
arXiv ID: 2609.24489 / 要約の誤りについて