日本フィジカルAI新聞

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

模倣学習/強化学習arXiv:2505.10442

IN-RIL: 模倣学習と強化学習を交互に組み合わせた方策ファインチューニング

IN-RIL: Interleaved Reinforcement and Imitation Learning for Policy Fine-Tuning

シェア:XThreadsFacebookLINEはてブBluesky

強化学習の更新の合間に模倣学習の更新を定期的に挿入し、勾配を直交部分空間に分離することで、ファインチューニングの安定性とサンプル効率を向上させる手法を提案した。

詳しい要約

1. どんなもの?

- ロボティクス方策の fine-tuning 手法 IN-RIL(INterleaved Reinforcement learning and Imitation Learning)の提案 - IL ベースの pre-training 後に RL で fine-tuning する二段階パラダイムの不安定性・低サンプル効率を改善することを狙う - RL 更新の間に定期的に IL 更新を挿入する interleaved な学習スキーム - 様々な state-of-the-art な RL アルゴリズムに組み込める汎用 plug-in として位置づけ

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

- 従来の IL pre-training → RL fine-tuning の二段階手法は RL fine-tuning 段階で不安定・サンプル効率が悪い - IN-RIL は fine-tuning 全体を通じて IL の安定性と expert data の guidance を活用 - IL と RL の目的関数が異なることに起因する干渉を gradient separation で防ぐ点が新しい - 14 タスク・3 benchmarks でサンプル効率改善と性能崩壊の緩和を示す - Robomimic Transport で成功率 12%→88%、6.3x 改善という具体的成果

3. 技術・手法の肝は?

- RL 更新を複数回行った後に IL 更新を定期的に注入する interleaved 更新 - IL と RL は最適化目的が異なるため、勾配分離機構(gradient separation mechanisms)を導入 - 競合しうる勾配更新を直交部分空間(orthogonal subspaces)に分離して破壊的干渉を防止 - interleaving が学習を安定化しサンプル効率を高める理由について厳密な解析を実施

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

- FurnitureBench、OpenAI Gym、Robomimic の 3 benchmarks にわたる 14 の robot manipulation / locomotion タスクで実験 - long-horizon と short-horizon、sparse reward と dense reward の両設定で評価 - オンライン fine-tuning におけるサンプル効率改善と性能崩壊の緩和を確認 - Robomimic Transport で成功率が 12% から 88% へ、6.3x の改善 - 各種 state-of-the-art RL アルゴリズムに plug-in として適用可能なことを示す

5. 議論はある?

- interleaving IL と RL がなぜ学習を安定化しサンプル効率を改善するかについて厳密な解析を行い知見を示す - IL と RL の目的関数の違いによる破壊的干渉を gradient separation で扱う設計上の議論 - 具体的な限界や失敗ケース、計算コスト等の詳細は要旨からは不明 - 汎用 plug-in 性の範囲や適用条件の詳細は要旨からは不明

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

- 比較対象である IL pre-training + RL fine-tuning の二段階パラダイムの代表的研究 - 提案手法が plug-in 可能とされる各種 state-of-the-art RL アルゴリズム(例:model-free off-policy RL 手法) - 評価に用いられた Robomimic、FurnitureBench、OpenAI Gym の関連ベンチマーク論文 - IL と RL の統合や勾配干渉・直交部分空間分離に関する関連手法

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

著者: Dechen Gao, Hang Wang, Hanchu Zhou, Nejib Ammar, Shatadal Mishra, Ahmadreza Moradipari, Iman Soltani, Junshan Zhang

分類: cs.RO, cs.AI

原文アブストラクト

Imitation learning (IL) and reinforcement learning (RL) each offer distinct advantages for robotics policy learning: IL provides stable learning from demonstrations, and RL promotes generalization through exploration. While existing robot learning approaches using IL-based pre-training followed by RL-based fine-tuning are promising, this two-step learning paradigm often suffers from instability and poor sample efficiency during the RL fine-tuning phase. In this work, we introduce IN-RIL, INterleaved Reinforcement learning and Imitation Learning, for policy fine-tuning, which periodically injects IL updates after multiple RL updates and hence can benefit from the stability of IL and the guidance of expert data for more efficient exploration throughout the entire fine-tuning process. Since IL and RL involve different optimization objectives, we develop gradient separation mechanisms to prevent destructive interference during \ABBR fine-tuning, by separating possibly conflicting gradient updates in orthogonal subspaces. Furthermore, we conduct rigorous analysis, and our findings shed light on why interleaving IL with RL stabilizes learning and improves sample-efficiency. Extensive experiments on 14 robot manipulation and locomotion tasks across 3 benchmarks, including FurnitureBench, OpenAI Gym, and Robomimic, demonstrate that \ABBR can significantly improve sample efficiency and mitigate performance collapse during online finetuning in both long- and short-horizon tasks with either sparse or dense rewards. IN-RIL, as a general plug-in compatible with various state-of-the-art RL algorithms, can significantly improve RL fine-tuning, e.g., from 12\% to 88\% with 6.3x improvement in the success rate on Robomimic Transport. Project page: https://github.com/ucd-dare/IN-RIL.

関連論文

  1. 1Boston Dynamics、車輪付きAtlasの研究試作機を紹介研究・技術
  2. 2GXOとApptronik、Apolloの物流現場試験を拡大 最大1,000台の展開を構想量産・展開
  3. 3newmoとKDDIが資本業務提携 2028年に堺市で自動運転タクシー商用化目指す提携・買収
  4. 4NVIDIA、CoRLでGR00T 2とIsaac Lab 3.0を発表新製品・発表
  5. 5UC BerkeleyとNVIDIAの研究者、AIエージェントでロボット制御コードを仮想環境で改善研究・技術

PR・アフィリエイト広告

MINISFORUM(ミニPC)

MINISFORUM(ミニPC)の公式広告バナーMINISFORUM公式ストアを見る ↗
PR本紙発行元 EmplifAI