PaperOAT
OAT: Ordered Action Tokenization
Claim tested: OAT improves real-robot task success over QueST with full-token decoding.
Tested
Result
Improvement over baseline
Not replicated
- Reported
- +33 pts
- Ours
- −8 pts
- Gain vs baseline
- 95% CI, −18 pts to +2 pts
- Paper's reported gain
- No difference
Success rate
- Reported
- 80%
- Ours
- 9%
- Measured on our stations
- 95% CI, 5%–17%
- Reported in the paper
Across iterations
- Iteration 1vs QueST+3 pts
- Iteration 2vs QueST−8 pts
- Iteration 3vs QueST−16 pts
By task
| Task | BaselineQueST | MethodOAT | Improvement |
|---|---|---|---|
| e01Clamp electrical component | 0%0/30 | 0%0/30 | 0 pts−11 to +11 |
| e02Place loaded boxes in container (bimanual) | 17%5/30 | 0%0/30 | −17 pts−34 to −2 |
| e03Pour screws into the funnel | 33%10/30 | 27%8/30 | −7 pts−28 to +16 |
| All tasks | 17%15/90 | 9%8/90 | −8 pts−18 to +2 |
| Reported in the paper | 48% | 80% | +33 pts |
Rollouts
One clip per environment from this entry's evaluation, beside where each evaluation episode placed the environment's objects, coloured by its rating. Use the button beside each environment to open its evaluation dataset in the dataset viewer.
e03Pour screws into the funnel BaselineQueST10/3033%
Show in dataset viewer of QueST on Pour screws into the funnelMethodOAT8/3027%
Show in dataset viewer of OAT on Pour screws into the funnel- Held-out benchmark environment
e01Clamp electrical component BaselineQueST0/300%
Show in dataset viewer of QueST on Clamp electrical componentMethodOAT0/300%
Show in dataset viewer of OAT on Clamp electrical component - Held-out benchmark environment
e02Place loaded boxes in container (bimanual)
Implementation Details
What we changed
We implemented OAT and QueST on the paper’s shared autoregressive backbone, ported from the authors’ code. The two arms differ only in the action tokenizer, and both decode all 8 tokens over the paper’s 1,000-entry codebook. This round retrains both arms after the first round, in which both rarely succeeded. Differences from the paper:
- Pretrained visual encoder. Both arms use an ImageNet-pretrained ResNet-18 with BatchNorm and ImageNet input normalization, instead of the paper’s randomly initialised ResNet with GroupNorm.
- One observation frame. Both arms see one frame instead of two, because station serving sends one observation per request.
- Two arms and three cameras. Actions are 20-D end-effector poses for two arms at 30 Hz with three cameras, instead of a 7-D single arm with one webcam.
- Chunk-relative actions. Positions and rotations in each chunk are expressed relative to the current end-effector pose, while gripper widths stay absolute.
- Larger crop. Random crops keep 116 of 128 pixels instead of 76.
- Checkpoint selection. Both arms hold out every tenth demonstration, train for at most 8,000 steps and serve the checkpoint with the lowest held-out token loss, instead of selecting checkpoints by simulated success.
- Blend execution. Both arms run on the station’s blend runtime, which crossfades overlapping chunks, instead of executing 16 of 32 actions and replanning synchronously.
- QueST loss. QueST’s reconstruction loss is computed on normalized actions.
Prompt summary
Use ImageNet-pretrained ResNet weights in both arms and verify they load (that round was cancelled before boxes ran). Then retrain both arms with a held-out validation split, report validation loss and tokenizer reconstruction, select checkpoints by validation loss, check for overfitting and diagnose early episode endings, and evaluate on the blend runtime in a new project. (Kept 8 tokens and ImageNet encoders, switched to chunk-relative actions, capped training at 8k steps.)
Response summary
QueST vs OAT: pour 10/30 vs 8/30 (AP 46% vs 40%), clamp 0/30 vs 0/30, boxes 5/30 vs 0/30 (AP 53% vs 75%, one OAT boxes episode still unreviewed); OAT −7.8 pp success (p = 0.09).
Conclusion
QueST ahead or equal on every task (−7.8 pp for OAT). An independent review found both tokenizers under-capacity for 20-D chunks, motivating a larger token budget.