日本フィジカルAI新聞

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

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

バランスの取れたデータダイエット:ロボット制御のための超大規模強化学習における探索ボトルネックの解消

A Balanced Data Diet: Addressing the Exploration Bottleneck in Mega-Scale RL for Robot Control

シェア:XThreadsFacebookLINEはてブBluesky

強化学習の大規模並列シミュレーションにおいて、ポリシーの能力の限界付近のタスク設定に学習を集中させる適応的サンプリング手法を提案し、四足歩行などの難タスクを効率的に学習できるようにした。

詳しい要約

1. どんなもの?

- 大規模並列シミュレーションを用いた強化学習(RL)によるロボット制御の研究。 - 四足歩行のlocomotionや接触の多いassemblyなど、精密・動的なタスクを対象。 - 課題: シミュレータリセットの分布を一様サンプリングすると、既に習得済みまたは未習得のタスク構成に学習経験が浪費される。 - 提案: Success Guided Sampling (SGS) という適応的サンプラー。 - ポリシーの能力のフロンティア付近のタスク構成に学習を集中させる。 - 最大2^20(100万以上)の並列環境で検証。 - 学習したmanipulationポリシーをRGBベースに蒸留し、実機でzero-shot転移を実証。

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

- 従来のRLパイプラインは、shaped rewardsやdemonstrationsなど、タスクごとの工学負担の大きい構造的priorに依存。 - 近年の研究では、多様なシミュレータリセットと大規模並列シミュレーションで工学負担を軽減できることが示された。 - しかし、そのパラダイムをより精密・動的な問題に単純にスケールするのは容易ではない。 - 一様サンプリングでは、ポリシーが既に習得した、またはまだ試せないタスク構成に学習経験が浪費される。 - そのため、並列環境のスケーリングの期待効果が得にくい。 - SGSは、ポリシーの能力のフロンティア付近に学習を集中させ、バッチ内の経験を最大限活用。 - 結果として、従来手法では解けなかった難しいmulti-terrain quadruped locomotionやcontact-rich assemblyを解決。

3. 技術・手法の肝は?

- Success Guided Sampling (SGS) というシンプルな適応的サンプラーを導入。 - ポリシーの能力のフロンティア付近のタスク構成にRL訓練を集中させる。 - これにより、大規模シミュレーションRLがバッチ内の経験を最大限活用できる。 - 大規模並列シミュレーションへの効果的なスケーリングを可能にする。 - 最大2^20(100万以上)の並列環境を使用。 - 学習したmanipulationポリシーをRGBベースのポリシーに蒸留。 - 実機でのzero-shot転移を実現。

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

- 最大2^20(100万以上)の並列環境を用いた実験を実施。 - 難しいmulti-terrain quadruped locomotionとcontact-rich assemblyタスクで、SGSがRLを解決に導くことを示した。 - 従来手法ではこれらのタスクを解けなかった。 - 学習したmanipulationポリシーをRGBベースに蒸留し、実機でいくつかの難しいassemblyタスクへのzero-shot転移を実証。 - プロジェクトウェブサイト: https://sgs-rl.github.io/

5. 議論はある?

- 要旨からは不明。 - ただし、一様サンプリングの非効率性と、SGSによるフロンティア集中の有効性が議論の中心。 - 大規模並列環境のスケーリングにおける学習信号の浪費が問題として提起されている。 - SGSの限界や適用範囲についての議論は要旨からは不明。

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

- 要旨で参照/比較されている研究: 多様なシミュレータリセットと大規模並列シミュレーションを組み合わせた先行研究(具体的な論文名は要旨に記載なし)。 - 関連手法: sim-to-real reinforcement learning (RL)、shaped rewards、demonstrations。 - 同分野の定番: domain randomization、curriculum learning、parallel simulation (例: Isaac Gym)。 - 具体的な次に読むべき論文は要旨からは不明。

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

著者: Octi Zhang, Mateo Guaman Castro, Patrick Yin, Ignacio Dagnino, Abhishek Gupta, Rosario Scalise, Byron Boots

分類: cs.RO, cs.LG

原文アブストラクト

General-purpose robots must perform a wide range of tasks from agile locomotion to dexterous manipulation. While sim-to-real reinforcement learning (RL) has proven to be a useful tool for this goal, current RL pipelines depend on engineering-heavy, per-task structural priors such as shaped rewards and demonstrations. Recent work has shown that diverse simulator resets, combined with massively parallel simulation, can alleviate much of this engineering burden on several manipulation problems. However, we find that naively scaling this paradigm to more precise or dynamic problems remains non-trivial. While simulator resets can help with exploration, uniformly sampling over this distribution wastes a growing fraction of learning experience on task configurations the policy has already mastered or cannot yet attempt. This makes it challenging to see the expected benefits of scaling parallel environments for RL, since much of the learning signal in a batch is wasted during learning. To mitigate this, we introduce Success Guided Sampling (SGS), a simple adaptive sampler that concentrates RL training on task configurations around the frontier of the policy's capabilities. Doing so allows large-scale simulated RL to make the most out of the experience in a batch, enabling much more effective scaling to large-scale parallel simulation. Across experiments using up to $2^{20}$ (over one million) parallel environments, SGS enables RL to solve challenging multi-terrain quadruped locomotion and contact-rich assembly tasks that prior methods fail to solve. Finally, we distill the learned manipulation policies into RGB-based policies and demonstrate zero-shot transfer to several challenging assembly tasks on real hardware. Project website: https://sgs-rl.github.io/.

関連論文

PR本紙発行元 EmplifAI