anonymous89793 commited on
Commit
0a82e17
·
1 Parent(s): 389ea91

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +45 -0
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.