拡散ポリシーで誘導する拡散プランナーによるオフライン強化学習
DPDP: Diffusion Policy-Steered Diffusion Planner for Offline Reinforcement Learning
拡散ポリシー(価値志向だが近視眼的)と拡散プランナー(長期志向だが振る舞い依存)をプランナーの実行段階で統合し、逆動力学の事前分布を価値最適化ポリシーで置き換えるDPDPを提案。D4RLで性能向上を実証。
著者: Ruyi An, Mingyuan Zhou
分類: cs.LG
原文アブストラクト
In offline reinforcement learning, diffusion models fill two roles with complementary strengths. Typically guided by a critic, a diffusion policy is value-aware and favors high-return, behavior-supported actions, but it is short-sighted: it never lays out the states it intends to visit. A diffusion planner is far-sighted, synthesizing such a long-horizon trajectory, but it is only behavior-driven at execution, realizing each planned transition using only offline behavior.Each is strong exactly where the other is weak. We introduce DPDP to bring them together at the planner's execution step, where a planned transition becomes an action. Reading the planner's inverse-dynamics executor as a posterior over actions: the environment's transition likelihood relates an action to the planned next state, while the behavioral prior determines its weight under the offline data. Replacing this prior with a value-optimized diffusion policy preserves the planned transition likelihood and favors valuable actions that can serve the plan. We target the resulting clean-action distribution in two ways. A training-free sampler composes policy and inverse-dynamics scores at matched action noise levels and applies Langevin correction; this route uses a shared action noise schedule and an approximately common terminal prior. A preference objective instead amortizes the transfer into a single diffusion executor. Across D4RL, DPDP improves the planner it builds on, staying competitive with reactive diffusion policies on locomotion and leading on long-horizon navigation.
関連論文
- MAC-v2: ピクセル入力からの長期的モデルベース価値展開オフライン強化学習
- タスク認識表現を備えたDecision Transformerによるオフライン・メタ強化学習オフライン強化学習
- 効率的な拡散オフライン強化学習のための合成的平均速度モデリングオフライン強化学習
- 対話制約付きオフライン強化学習による自動運転オフライン強化学習
- LASER: 潜在空間随伴マッチングによるサポート制約付きエントロピー正則化オフライン強化学習オフライン強化学習
- 役割適応型方策最適化によるオフライン強化学習オフライン強化学習