Video wird geladen...

Video konnte nicht geladen werden

Zur Startseite

Announcing RoboFuME🤖💨, a system for autonomous & efficient real-world robot learning that 1. pre-trains a VLM reward model and a multi-task RL policy from diverse off-the-shelf demo data; 2. runs RL fine-tuning online with the VLM reward model. 🔗 🧵↓

88,312 Aufrufe • vor 3 Jahren •via X (Twitter)

6 Kommentare

Profilbild von Jingyun Yang
Jingyun Yangvor 3 Jahren

The pre-train + fine-tune paradigm in ML has shown promise in a lot of domains. RoboFuME is a reset-free fine-tuning system that trains with data from an existing robot dataset collected at a different institution and takes just 3 hours to fine-tune autonomously. [1/6]

Profilbild von Jingyun Yang
Jingyun Yangvor 3 Jahren

In pre-training, we use diverse prior demonstration data from the Bridge V2 dataset, which was collected at a different institution. For each task, we pick about 1,000 relevant trajectories in the dataset as prior data for the agent. [2/6]

Profilbild von Jingyun Yang
Jingyun Yangvor 3 Jahren

In addition to the prior data, we spend 30 minutes to collect 50 successful demos for both the forward and backward target tasks, as well as 10 minutes of failure demos for both tasks. [3/6]

Profilbild von Jingyun Yang
Jingyun Yangvor 3 Jahren

We pre-train an RL policy from both the prior data and the target demonstration data using an offline RL algorithm. Then, we rollout the policy to perform the forward and backward tasks alternatively while updating the policy with a VLM reward model as reward feedback. [4/6]

Profilbild von Jingyun Yang
Jingyun Yangvor 3 Jahren

We test our system on 5 real robot tasks. Our method significantly improves over both offline-only and BC performance after 30k steps of online interaction (2-4 hours). [5/6]

Profilbild von Jingyun Yang
Jingyun Yangvor 3 Jahren

This is joint work with an amazing team including @maxsobolmark, @brandontvu, @archit_sharma97, @leto__jean, @chelseabfinn Website: Paper: [6/6]

Ähnliche Videos