arXiv reaDer
XIRL:クロスボディメント逆強化学習
XIRL: Cross-embodiment Inverse Reinforcement Learning
エージェントが同じタスクを示す他のエージェント(人間など)のビデオからポリシーを学習するが、その実施形態(形状、アクション、エンドエフェクタのダイナミクスなど)に大きな違いがある視覚的なクロスボディミテーション設定を調査します。この作品では、これらの違いにロバストなクロスボディのデモンストレーションビデオからビジョンベースの報酬機能を自動的に発見して学習することが可能であることを示しています。具体的には、時間的サイクル一貫性制約を活用して、それぞれが同じタスクを実行する複数のエキスパートエージェントにわたるデモンストレーションのオフラインビデオからタスクの進行をキャプチャする深い視覚的埋め込みを学習する、クロスボディメント逆強化学習(XIRL)の自己教師あり方法を提示します。実施形態の違いにより異なる。私たちの仕事の前に、自己監視埋め込みから報酬を生成するには、通常、参照軌道との位置合わせが必要でした。これは、具体的な実施形態の大きな違いの下では取得が難しい場合があります。埋め込みがタスクの進行状況を認識している場合、学習された埋め込みスペースで現在の状態と目標の状態の間の負の距離を取るだけで、強化学習を使用したトレーニングポリシーの報酬として役立つことを経験的に示します。学習した報酬関数は、トレーニング中に見られる実施形態で機能するだけでなく、まったく新しい実施形態に一般化することもわかります。さらに、実際の人間のデモンストレーションをシミュレートされたロボットに転送すると、XIRLは現在の最良の方法よりもサンプル効率が高いことがわかります。定性的な結果、コード、およびデータセットは、https://x-irl.github.ioで入手できます。
We investigate the visual cross-embodiment imitation setting, in which agents learn policies from videos of other agents (such as humans) demonstrating the same task, but with stark differences in their embodiments -- shape, actions, end-effector dynamics, etc. In this work, we demonstrate that it is possible to automatically discover and learn vision-based reward functions from cross-embodiment demonstration videos that are robust to these differences. Specifically, we present a self-supervised method for Cross-embodiment Inverse Reinforcement Learning (XIRL) that leverages temporal cycle-consistency constraints to learn deep visual embeddings that capture task progression from offline videos of demonstrations across multiple expert agents, each performing the same task differently due to embodiment differences. Prior to our work, producing rewards from self-supervised embeddings typically required alignment with a reference trajectory, which may be difficult to acquire under stark embodiment differences. We show empirically that if the embeddings are aware of task progress, simply taking the negative distance between the current state and goal state in the learned embedding space is useful as a reward for training policies with reinforcement learning. We find our learned reward function not only works for embodiments seen during training, but also generalizes to entirely new embodiments. Additionally, when transferring real-world human demonstrations to a simulated robot, we find that XIRL is more sample efficient than current best methods. Qualitative results, code, and datasets are available at https://x-irl.github.io
updated: Mon Sep 20 2021 17:53:26 GMT+0000 (UTC)
published: Mon Jun 07 2021 18:45:07 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト