TL;DR
MPAIL2は、報酬・行動ラベル・シミュレーション・事前学習なしで、観測(例:動画)と実機インタラクションだけから逆強化学習を実機で行う手法です。潜在世界モデルとMPPI計画をオンラインで組み合わせ、論文では実機のピックアンドプレースを40分未満で82%成功と報告しています。
既存手法との違い
原論文では、同じIRLfO設定のAIRL、DAC、MAIRL、MPAILを、可能な範囲でMPAIL2の設計(計画・ダイナミクス・オフポリシー最適化)を適用して比較しています。また、手設計報酬と行動ラベル付きデモを使うRLPD、行動クローニングのDiffusion Policyとも比較しています。論文報告では、実機ピックアンドプレースでMPAIL2が40分未満・82%成功に対し、同一のインタラクション・デモ予算のRLとBCは0%と12%だったと述べています。AIRLやMPAILはシミュレーションでの成功率が無視できるため実機評価されていない、と本文にあります。
使える条件
入力は専門家の観測軌跡(例:動画)のみで、行動・報酬の教師信号は仮定しません。実機インタラクションで遷移を収集し、事前学習・シミュレーションなしでオンライン更新します。評価はFrankaとKinova Gen3を使った視覚マニピュレーション(Block Push、Mug-on-Plate、Pick-and-Place)で、ビデオのみの実験では外部固定カメラ観測を使用しています。他のロボット、長期タスク、多様な実環境への一般化は提示本文からは確認できません。
限界と未検証の範囲
評価は限られた実機タスクと2種類のロボットプラットフォームに留まります。ベストチェックポイントは成功ラベルを用いて事後選択されており、学習中には使わないものの、停止条件がない場合の性能はLastで別途報告されています。人間操作のデモやリセット手順に依存し、実機での探索・リセット負担も残ります。Q2/Q3の詳細な実験結果や、未評価のタスク・環境・長期運用への一般化は提示本文からは確認できません。
解説
ねえ智也くん、この「観測のみから実機ロボットが40分で学ぶオンライン世界モデル型逆強化学習」って論文、タイトルからしてすごいんだけど、何が問題意識なの?
実機ロボット学習が、手設計の報酬、シミュレーション、テレオペみたいな行動教師に依存しがちなのがボトルネック。この論文は観測だけ、例えば動画だけからIRLfOを実機でやる話。
観測だけって、行動ラベルも報酬もなしでしょ? それってかなり無理ゲーじゃん。
そう。IRLfOはデモ量を減らせる代わりに、報酬と方策を同時に最適化するから実機だと不安定でインタラクションが重い。だから実用化が難しかった。
MPAIL2はどうやってそれを可能にしたの?
潜在世界モデルをオンラインで学ぶ。観測を潜在状態にエンコードして、ダイナミクスで未来を予測。報酬と価値はオフポリシーで更新。計画にはMPPIを使って、複数ステップ方策で計画を温める。
ざっくり言うと、頭の中で未来を予測しながら、その場で計画を何個も試してる感じ?
近い。しかも報酬・行動ラベル・シミュレーション・事前学習を仮定しない。実機の観測、行動、次観測を貯めて全部オンラインで更新する。
評価は?
FrankaでBlock PushとMug-on-Plate、Kinova Gen3でPick-and-Place。ピックアンドプレースは40分未満で82%成功。同じインタラクションとデモ予算のRLは0%、BCは12%だった、と論文は報告してる。
RLとBCは報酬や行動教師があるのにそれってかなり衝撃。比較は他にもある?
同じIRLfO設定のAIRL、DAC、MAIRL、MPAILを、可能な範囲でMPAIL2の設計を入れて比較。手設計報酬と行動ラベル付きデモのRLPD、Diffusion Policyとも比較してる。AIRLとMPAILはシミュレーションで成功率がほぼ無いから実機評価してない、って本文にある。
なるほどね。でも条件は限られてる?
うん。入力は専門家の観測軌跡だけで、実機インタラクションで遷移を集める。評価は2種類のロボットの視覚マニピュレーション。ビデオのみの実験では外部固定カメラ観測を使ってる。他のロボットや長期タスクへの一般化は提示本文からは確認できない。
制限もちゃんとあるんだ。
ある。評価タスクとプラットフォームは限られる。人間操作のデモやリセット手順に依存する。Bestチェックポイントは成功ラベルで事後選択してるけど学習中は使わない。停止条件がない場合のLastも別に報告してる。
意義は?
実世界で視覚マニピュレーションをゼロから学ぶIRLの初例、って主張。実世界だけで正のオンライン転移を示したのも初、って書いてある。観測だけの動画デモへの拡張もやってる。
すごいね。ロボットが40分で覚えるなら、私も40分で部屋片付け覚えたいな。
君の部屋は観測だけじゃ足りないと思う。