Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ayalew, Tewodros, Zhang, Xiao, Wu, Kevin Yuanbo, Jiang, Tianchong, Maire, Michael, Walter, Matthew R.
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:https://arxiv.org/abs/2411.17764
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917849637847040
author Ayalew, Tewodros
Zhang, Xiao
Wu, Kevin Yuanbo
Jiang, Tianchong
Maire, Michael
Walter, Matthew R.
author_facet Ayalew, Tewodros
Zhang, Xiao
Wu, Kevin Yuanbo
Jiang, Tianchong
Maire, Michael
Walter, Matthew R.
contents We present PROGRESSOR, a novel framework that learns a task-agnostic reward function from videos, enabling policy training through goal-conditioned reinforcement learning (RL) without manual supervision. Underlying this reward is an estimate of the distribution over task progress as a function of the current, initial, and goal observations that is learned in a self-supervised fashion. Crucially, PROGRESSOR refines rewards adversarially during online RL training by pushing back predictions for out-of-distribution observations, to mitigate distribution shift inherent in non-expert observations. Utilizing this progress prediction as a dense reward together with an adversarial push-back, we show that PROGRESSOR enables robots to learn complex behaviors without any external supervision. Pretrained on large-scale egocentric human video from EPIC-KITCHENS, PROGRESSOR requires no fine-tuning on in-domain task-specific data for generalization to real-robot offline RL under noisy demonstrations, outperforming contemporary methods that provide dense visual reward for robotic learning. Our findings highlight the potential of PROGRESSOR for scalable robotic applications where direct action labels and task-specific rewards are not readily available.
format Preprint
id arxiv_https___arxiv_org_abs_2411_17764
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PROGRESSOR: A Perceptually Guided Reward Estimator with Self-Supervised Online Refinement
Ayalew, Tewodros
Zhang, Xiao
Wu, Kevin Yuanbo
Jiang, Tianchong
Maire, Michael
Walter, Matthew R.
Robotics
Artificial Intelligence
We present PROGRESSOR, a novel framework that learns a task-agnostic reward function from videos, enabling policy training through goal-conditioned reinforcement learning (RL) without manual supervision. Underlying this reward is an estimate of the distribution over task progress as a function of the current, initial, and goal observations that is learned in a self-supervised fashion. Crucially, PROGRESSOR refines rewards adversarially during online RL training by pushing back predictions for out-of-distribution observations, to mitigate distribution shift inherent in non-expert observations. Utilizing this progress prediction as a dense reward together with an adversarial push-back, we show that PROGRESSOR enables robots to learn complex behaviors without any external supervision. Pretrained on large-scale egocentric human video from EPIC-KITCHENS, PROGRESSOR requires no fine-tuning on in-domain task-specific data for generalization to real-robot offline RL under noisy demonstrations, outperforming contemporary methods that provide dense visual reward for robotic learning. Our findings highlight the potential of PROGRESSOR for scalable robotic applications where direct action labels and task-specific rewards are not readily available.
title PROGRESSOR: A Perceptually Guided Reward Estimator with Self-Supervised Online Refinement
topic Robotics
Artificial Intelligence
url https://arxiv.org/abs/2411.17764