Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,86 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Robo-ValueRL
|
| 2 |
+
|
| 3 |
+
[[Project Page](https://gewu-lab.github.io/robo-valuerl/)] [[GitHub](https://github.com/Open-X-Humanoid/Robo-ValueRL)] [[Paper](#)] [[Data](#)]
|
| 4 |
+
|
| 5 |
+
This repository contains the **models and data** for **Robo-ValueRL: Reliable Value Estimation for Offline-to-Online Reinforcement Learning**.
|
| 6 |
+
|
| 7 |
+
Robo-ValueRL studies how reliable value estimation can guide robotic policy learning from heterogeneous offline demonstrations and online rollout trajectories. The framework learns a history-conditioned value estimator, converts value differences into action-quality conditions for offline policy pretraining, and uses value-guided rollout filtering for stable online residual adaptation.
|
| 8 |
+
|
| 9 |
+
## π€ Model Description
|
| 10 |
+
|
| 11 |
+
**Robo-ValueRL** is an offline-to-online robotic reinforcement learning framework centered on reliable value estimation. Instead of only reporting final task success, Robo-ValueRL explicitly diagnoses whether learned values capture global task progress and local action-level preference, then propagates these value signals into downstream policy learning.
|
| 12 |
+
|
| 13 |
+
The released model suite includes:
|
| 14 |
+
|
| 15 |
+
1. **History-Conditioned Value Estimator**
|
| 16 |
+
Predicts normalized task progress from multi-view robot observations, language instructions, and visual history. The temporal context helps reduce ambiguity caused by occlusions, repeated motions, and visually similar task stages.
|
| 17 |
+
|
| 18 |
+
2. **Quality-Conditioned VLA Policy**
|
| 19 |
+
Uses value differences to derive action-quality conditions. These conditions guide a Vision-Language-Action policy during offline pretraining, allowing the policy to prioritize useful behaviors from mixed-quality demonstrations.
|
| 20 |
+
|
| 21 |
+
3. **Online Residual Adaptation Module**
|
| 22 |
+
Learns lightweight corrections from value-filtered online rollouts while keeping the pretrained base policy frozen. This enables targeted failure recovery and self-correction without overwriting the offline prior.
|
| 23 |
+
|
| 24 |
+
## π¦ Data Description
|
| 25 |
+
|
| 26 |
+
The Robo-ValueRL dataset contains heterogeneous real-robot experience collected on two long-horizon manipulation tasks:
|
| 27 |
+
|
| 28 |
+
- **Chip Insertion**: millimeter-level precision manipulation requiring PCB grasping, pose adjustment, chip grasping, and insertion.
|
| 29 |
+
- **Block Disassembly**: generalizable object disassembly requiring robust grasping, separation, and classification behaviors.
|
| 30 |
+
|
| 31 |
+
The dataset includes:
|
| 32 |
+
|
| 33 |
+
- **240 hours** of offline demonstrations
|
| 34 |
+
- **3,000+** online rollout trajectories
|
| 35 |
+
- Multi-view robot observations
|
| 36 |
+
- Language task instructions
|
| 37 |
+
- Robot states and action chunks
|
| 38 |
+
- Mixed-quality trajectories, including successful demonstrations, corrections, suboptimal behaviors, and failure cases
|
| 39 |
+
- Value-derived action-quality labels / conditions for policy learning
|
| 40 |
+
- Online rollout segments for residual adaptation
|
| 41 |
+
|
| 42 |
+
## π§± Model Hierarchy
|
| 43 |
+
|
| 44 |
+
1. **Value Estimation**
|
| 45 |
+
- Learns history-conditioned value functions from heterogeneous robot data.
|
| 46 |
+
- Evaluates value reliability with global-progress and local-preference metrics.
|
| 47 |
+
|
| 48 |
+
2. **Offline Policy Pretraining**
|
| 49 |
+
- Converts value differences into action-quality conditions.
|
| 50 |
+
- Trains a quality-conditioned VLA policy on mixed-quality demonstrations.
|
| 51 |
+
|
| 52 |
+
3. **Online Policy Improvement**
|
| 53 |
+
- Uses reliable value estimates to filter online rollout data.
|
| 54 |
+
- Trains a lightweight residual adapter for targeted real-world improvement.
|
| 55 |
+
|
| 56 |
+
## β¨ Key Features
|
| 57 |
+
|
| 58 |
+
* **Reliable Value Estimation**: Uses visual history to produce smoother progress estimates and sharper error responses.
|
| 59 |
+
* **Value-Guided Data Utilization**: Prioritizes useful demonstrations and rollout segments from heterogeneous robot experience.
|
| 60 |
+
* **Quality-Conditioned Policy Learning**: Conditions the VLA policy on value-derived action quality.
|
| 61 |
+
* **Stable Offline-to-Online Improvement**: Improves real-world performance through residual adaptation while preserving the pretrained base policy.
|
| 62 |
+
* **Real-Robot Evaluation**: Evaluated on precision chip insertion and generalizable block disassembly.
|
| 63 |
+
|
| 64 |
+
## π Highlights
|
| 65 |
+
|
| 66 |
+
- 86% final success on chip insertion
|
| 67 |
+
- 84% final success on block disassembly
|
| 68 |
+
- +26% offline gain on chip insertion
|
| 69 |
+
- +34% offline gain on block disassembly
|
| 70 |
+
- 240h offline demonstrations
|
| 71 |
+
- 3,000+ online rollout trajectories
|
| 72 |
+
|
| 73 |
+
## π Usage
|
| 74 |
+
|
| 75 |
+
The released assets are organized for reproducing Robo-ValueRL's model and data pipeline.
|
| 76 |
+
|
| 77 |
+
Please refer to the [GitHub repository](https://github.com/Open-X-Humanoid/Robo-ValueRL) for setup instructions, inference scripts, training code, and data-processing utilities.
|
| 78 |
+
|
| 79 |
+
## π Citation
|
| 80 |
+
If you use the XR-1 model in your research, please cite our work:(Citation will be updated after the arXiv release.)
|
| 81 |
+
|
| 82 |
+
## π License
|
| 83 |
+
Please refer to the license file in the [GitHub repository](https://github.com/Open-X-Humanoid/Robo-ValueRL).
|
| 84 |
+
|
| 85 |
+
## π¬ Contact
|
| 86 |
+
For questions, please open an issue on our [GitHub repository](https://github.com/Open-X-Humanoid/Robo-ValueRL).
|