Rewrite model card: full AWBC/LR-soup lineage, dataset & sibling links (team INHA-UNITED)
Browse files
README.md
CHANGED
|
@@ -1,19 +1,46 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
| 3 |
tags:
|
| 4 |
- robotics
|
| 5 |
- vla
|
| 6 |
- pi0.5
|
|
|
|
| 7 |
- rby1
|
|
|
|
|
|
|
|
|
|
| 8 |
- robocup
|
| 9 |
---
|
| 10 |
|
| 11 |
# Doing Laundry — RoboCup 2026 Incheon
|
| 12 |
|
| 13 |
-
**Team
|
| 14 |
|
| 15 |
-
|
| 16 |
|
| 17 |
-
-
|
| 18 |
-
-
|
| 19 |
-
- Prompt: `
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
library_name: openpi
|
| 4 |
+
pipeline_tag: robotics
|
| 5 |
tags:
|
| 6 |
- robotics
|
| 7 |
- vla
|
| 8 |
- pi0.5
|
| 9 |
+
- openpi
|
| 10 |
- rby1
|
| 11 |
+
- fold
|
| 12 |
+
- awbc
|
| 13 |
+
- model-soup
|
| 14 |
- robocup
|
| 15 |
---
|
| 16 |
|
| 17 |
# Doing Laundry — RoboCup 2026 Incheon
|
| 18 |
|
| 19 |
+
> **Team INHA-UNITED** · RB-Y1 T-shirt folding policy for the RoboCup 2026 @Home league (Incheon).
|
| 20 |
|
| 21 |
+
A π0.5 vision-language-action policy for Rainbow Robotics RB-Y1 T-shirt folding, deployed as the competition model. This is a **greedy model soup** (`M0_lr_soup`) — a weight average of a learning-rate sweep over the AWBC fold family, picked to maximize robustness on the real robot.
|
| 22 |
|
| 23 |
+
- **Method:** Model Arithmetic **greedy weight soup** of a 3-way LR sweep, all warm-started from the [v11 AWBC](https://huggingface.co/RB3159/pi05-rby1-fold-awbc) checkpoint (step 69999) and continued to step 99999. Members and soup weights: `lr1` (2.5e-6) ×2, `lr2` (5e-6) ×1, `lr4` (1e-5) ×1 — `lr1` was selected twice by greedy souping.
|
| 24 |
+
- **Training objective:** Advantage-Weighted Behavior Cloning (AWBC) following χ₀ (kai0) — per-frame `awbc_weight` from the [advantage estimator](https://huggingface.co/RB3159/rby1-fold-advantage-estimator)'s relative advantage `V(s+H) − V(s)`, thresholded at a fixed global p70 cutoff (~30% positive). See [arXiv:2602.09021](https://arxiv.org/abs/2602.09021), [OpenDriveLab/kai0](https://github.com/OpenDriveLab/kai0).
|
| 25 |
+
- **Prompt:** single task, `fold the t-shirt, Advantage: positive` (hold at inference).
|
| 26 |
+
- **Architecture:** π0.5 (`pi05`, PaliGemma `gemma_2b` + action expert `gemma_300m`, action horizon 50).
|
| 27 |
+
- **Lineage:** [pi05-rby1-fold-base](https://huggingface.co/RB3159/pi05-rby1-fold-base) (mir280) → [3stage](https://huggingface.co/RB3159/pi05-rby1-fold-3stage) → [v11 AWBC](https://huggingface.co/RB3159/pi05-rby1-fold-awbc) → LR sweep → **this soup**.
|
| 28 |
+
- **Training data:** 432 episodes — [rby1-fold-v1.1](https://huggingface.co/datasets/RB3159/rby1-fold-v1.1) master + left/right mirror + weak time-scale augmentation + DAgger interventions, with a per-frame `awbc_weight` column, img224.
|
| 29 |
+
- **Action space:** 16-dim — `[R_arm(7), R_gripper(1), L_arm(7), L_gripper(1)]`, delta on the 7-DoF arms, absolute on the grippers.
|
| 30 |
+
- **Robot:** Rainbow Robotics RB-Y1 (dual-arm).
|
| 31 |
+
|
| 32 |
+
## Why a soup
|
| 33 |
+
|
| 34 |
+
The three LR-sweep runs share an init and a data distribution but converge to different basins. Averaging their weights (greedy soup) keeps the shared skill while smoothing over run-specific overfitting — no extra inference cost, and empirically the safest single model to deploy under competition conditions.
|
| 35 |
+
|
| 36 |
+
## Files
|
| 37 |
+
|
| 38 |
+
`params/` (orbax inference weights) + `assets/.../norm_stats.json` + `_CHECKPOINT_METADATA`. Trained/merged with [openpi](https://github.com/Physical-Intelligence/openpi).
|
| 39 |
+
|
| 40 |
+
## Usage
|
| 41 |
+
|
| 42 |
+
```bash
|
| 43 |
+
python scripts/serve_policy.py --config pi05_rby1_fold_v11_1prompt_f2r03_awbc --checkpoint RB3159/Doing-Laundry-Robocup2026-Incheon
|
| 44 |
+
```
|
| 45 |
+
|
| 46 |
+
Hold the prompt to `fold the t-shirt, Advantage: positive`.
|