日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

週刊ニュースレター購読
強化学習arXiv:2611.05295

幾何学的サポートによる行動正則化でオンライン強化学習を改善

GSBR: Geometric Support Behavior Regularization for Online RL

シェア:XThreadsFacebookLINEはてブBluesky

事前学習済み方策のサポートからの距離に基づいて罰則を与えるGSBRを提案し、サブ最適行動の排除と改善の両立を実現、4つのロボットベンチマーク19タスクで成功率を33%から95%に向上させた。

詳しい要約

1. どんなもの?

- オンライン強化学習(online RL)において、事前学習済みのbase policy(拡散モデルやflow-matchingモデルなど)を正則化に用いる手法。 - base policyのsupport(分布が確率質量を持つ領域)からの距離に基づいてactionを罰するGeometric Support Behavior Regularization(GSBR)を提案。 - support内のactionは無罰、support外は距離に応じて滑らかに罰が増大する設計。 - 4つのrobotic benchmark、19タスクで評価。

2. 先行研究と比べてどこがすごい?

- 既存のbehavioral cloning系正則化はactorをbase policy分布全体へ引き寄せ、suboptimalなmodeの棄却を罰してしまう(過度に制限的)。 - hard support constraint系はbase policyのsupport直外の改善を禁じてしまう(過度に制限的)。 - GSBRはsupport内では罰を与えず高価値modeへ集中でき、support外でも価値利得が罰を上回れば改善を許容する点で両者の制限を緩和。 - 平均success rateをbase policyの33%から95%へ向上させ、比較手法中最高の平均success rateとthroughputを達成。次善のRL手法よりnormalized throughputが約23%高い。

3. 技術・手法の肝は?

- actionのbase policy supportからの距離を罰する正則化を導入。 - support内: 罰なし。support外: 距離に対し滑らかに増大する罰。 - 価値推定の信頼性が低いsupport外では罰を強めつつ、価値利得が罰を上回る改善は許容。 - 距離はbase policyからの有限個のサンプル集合から推定するため、base policyへのsample accessのみでよい。

4. どうやって有効だと検証した?

- 4つのrobotic benchmark、計19タスクで評価。 - base policyの平均success rate 33%に対しGSBRは95%を達成。 - 比較した全手法中で最高の平均success rateとthroughputを記録。 - 次善のRL手法と比べnormalized throughputが約23%高い。

5. 議論はある?

- 要旨からは不明(サンプル数依存性や距離推定の精度、support外改善の許容度に関する議論は明示されていない)。 - 提案手法はsample accessのみを要求する点が実用上の利点として示唆される。

6. 次に読むべき論文は?

- diffusion modelやflow-matching modelをbase policyとするimitation learning研究。 - behavioral cloning正則化を用いるonline RL手法。 - hard support constraintを用いるonline RL手法。 - 比較対象となる次善のRL手法(具体名は要旨からは不明)。

※ AIが要旨から生成した要約です。正確性は原文をご確認ください。

著者: Charles Xu, Jacob Berg, Liyiming Ke, Sergey Levine

分類: cs.RO, cs.LG

原文アブストラクト

Regularizing online reinforcement learning (RL) toward a pretrained base policy can improve sample efficiency and learning stability, but regularization that is too restrictive limits how much the policy can improve. Base policies -- typically diffusion or flow-matching models trained by imitation learning on large, diverse datasets -- capture many behaviors that are reasonable but suboptimal. Existing regularizers using such base policies are too restrictive in one of two ways: behavioral cloning regularizers pull the actor toward the full base-policy distribution, penalizing it for discarding suboptimal modes, while hard support constraints forbid improvements that lie just outside the base policy's support. We propose Geometric Support Behavior Regularization (GSBR), which penalizes actions by their distance from the base policy's support. Actions within the support incur no penalty, so the actor can concentrate on high-value modes; outside the support, where value estimates are less reliable, the penalty grows smoothly with distance while still permitting improvement when the value gained outweighs the penalty. GSBR estimates this distance from a finite set of base-policy samples, so it requires only sample access to the base policy. Across 19 tasks on four robotic benchmarks, GSBR raises the average success rate from 33% for the base policy to 95%, achieving the highest average success rate and throughput among all compared methods, with about 23% higher normalized throughput than the next-best RL method. Project website and open-source code at https://charlesxu0124.github.io/gsbr/ .

関連論文

PR本紙発行元 EmplifAI