arXiv reaDer
XIRL: クロス具現化逆強化学習
XIRL: Cross-embodiment Inverse Reinforcement Learning
エージェントは、同じタスクを示す他のエージェント (人間など) のビデオからポリシーを学習しますが、形、アクション、エンドエフェクターのダイナミクスなど、具体化に大きな違いがあります。この作業では、これらの違いに対して堅牢なクロス具現化デモンストレーション ビデオから、視覚ベースの報酬関数を自動的に発見して学習できることを示します。具体的には、時間的サイクルの一貫性制約を活用して、複数のエキスパート エージェントが同じタスクを実行するデモンストレーションのオフライン ビデオからタスクの進行をキャプチャする深い視覚的埋め込みを学習する、クロス具体化逆強化学習 (XIRL) の自己監視方法を紹介します。実施形態の違いにより異なる。私たちの仕事の前に、自己監視埋め込みから報酬を生成するには、通常、取得が難しい場合がある参照軌道との調整が必要でした。埋め込みがタスクの進行状況を認識している場合、学習された埋め込みスペースで現在の状態と目標状態の間の負の距離を単純に取ることが、強化学習によるトレーニング ポリシーの報酬として役立つことを経験的に示しています。学習した報酬関数は、トレーニング中に見られる具体化に対して機能するだけでなく、まったく新しい具体化にも一般化されることがわかりました。また、XIRL ポリシーはベースラインよりもサンプル効率が高く、場合によっては、グラウンド トゥルース スパース報酬でトレーニングされた同じエージェントのサンプル効率を超えることもわかりました。
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 has typically required alignment with a reference trajectory, which may be difficult to acquire. 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. We also find that XIRL policies are more sample efficient than baselines, and in some cases exceed the sample efficiency of the same agent trained with ground truth sparse rewards.
updated: Mon Jun 07 2021 18:45:07 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アソシエイト