Commit ·
0a82e17
1
Parent(s): 389ea91
Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- world-model
|
| 5 |
+
- active-learning
|
| 6 |
+
- confidence-estimation
|
| 7 |
+
- robot-learning
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# ConfAL-WM · Model Checkpoints
|
| 11 |
+
|
| 12 |
+
Model weights for **ConfAL-WM: Confidence-Guided Active Learning for Action-Conditioned
|
| 13 |
+
World Models** (anonymous submission). These artifacts correspond to the
|
| 14 |
+
"07 · Models & Data" section of the project page.
|
| 15 |
+
|
| 16 |
+
All checkpoints were produced inside the release codebase; absolute paths and
|
| 17 |
+
machine-specific metadata have been scrubbed (`<DATA_ROOT>` / `<ANON…>` placeholders).
|
| 18 |
+
|
| 19 |
+
## Checkpoints
|
| 20 |
+
|
| 21 |
+
| File | Card | Description |
|
| 22 |
+
|---|---|---|
|
| 23 |
+
| `EVAC warmup v1.ckpt` | EVAC · Warmup v1 | RoboTwin2.0 domain-adapted warmup world model. Starting point for all active-learning rounds (v1 inference + confidence probe scoring). Lightning ckpt, epoch 10 / step 2000. |
|
| 24 |
+
| `EVAC-v2 weighting none.ckpt` | EVAC-v2 · Weighting None | Selection-only retrained checkpoint (mean-risk acquisition, seed 123). Lightning ckpt, epoch 8 / step 4000. |
|
| 25 |
+
| `EVAC-v2 weighting frame.ckpt` | EVAC-v2 · Frame | Confidence-guided **frame**-level weighting (mean-risk acquisition, seed 42). Lightning ckpt, epoch 8 / step 4000. |
|
| 26 |
+
| `EVAC-v2 weighting frame+patch.ckpt` | EVAC-v2 · Frame + Patch | Dense confidence-guided (frame + patch) weighting (persistent-risk acquisition, seed 42). Lightning ckpt, epoch 8 / step 4000. |
|
| 27 |
+
| `Confidence probe RoboTwin2.0.pt` | Confidence Probe · RoboTwin2.0 | Main C3 confidence probe used in the paper (probe step 6000). Takes EVAC decoder features (h_dec embeddings) and outputs per-frame/patch confidence. |
|
| 28 |
+
| `Confidence probe AgiBotWorld.pt` | Confidence Probe · AgiBot World | Additional confidence probe trained on AgiBot World (probe step 6000). |
|
| 29 |
+
| `YOLO RoboTwin2.0.pt` | YOLO · RoboTwin2.0 | Gripper/trajectory-metric detector (left/right gripper) for EWMBench-style evaluation. Ultralytics format; train args sanitized. |
|
| 30 |
+
|
| 31 |
+
## Usage notes
|
| 32 |
+
|
| 33 |
+
- The `EVAC*` checkpoints are PyTorch-Lightning archives; restore with
|
| 34 |
+
`LightningModule.load_from_checkpoint(...)` using the model definition in the code release.
|
| 35 |
+
- The probes are plain `torch.save` state dicts; load with `torch.load(..., map_location="cpu")`.
|
| 36 |
+
- The YOLO detector can be loaded directly with `ultralytics.YOLO(path)`.
|
| 37 |
+
- Companion precomputed data (v1 inference outputs, dense confidence maps, baseline
|
| 38 |
+
scoring artifacts, YOLO annotations, evaluation tables) is available in the dataset repo
|
| 39 |
+
`anonymous89793/ConfAL-WM-Dataset`.
|
| 40 |
+
|
| 41 |
+
## Anonymization
|
| 42 |
+
|
| 43 |
+
- All absolute filesystem paths inside metadata/pickles were replaced with placeholders
|
| 44 |
+
(`<DATA_ROOT>/`, `<ANON…>`); no usernames, hostnames, or machine paths remain.
|
| 45 |
+
- Checkpoint tensors were **not** modified — only metadata strings.
|