Poke &Wiggle
Papers
Paper

OAT

OAT: Ordered Action Tokenization

Claim tested: OAT improves real-robot task success over QueST with full-token decoding.

Tested

Paper
arXiv⁠ (opens in a new tab)
Code
GitHub⁠ (opens in a new tab)
Venue
RSS 2026
Authors
Chaoqi Liu et al.
Farm tasks
3
Evaluation
540 episodes (270 for baseline, 270 for method)

Result

Improvement over baseline

Not replicated

Reported
+33 pts
Ours
−8 pts
−200+20+40+60
  • Gain vs baseline
  • 95% CI, −18 pts to +2 pts
  • Paper's reported gain
  • No difference

Success rate

Reported
80%
Ours
9%
0%20%40%60%80%100%
  • Measured on our stations
  • 95% CI, 5%–17%
  • Reported in the paper

Across iterations

  1. Iteration 1vs QueST+3 pts
  2. Iteration 2vs QueST−8 pts
  3. Iteration 3vs QueST−16 pts
−200+20+40+60

By task

TaskBaselineQueSTMethodOATImprovement
e01Clamp electrical component0%0/300%0/300 pts−11 to +11
e02Place loaded boxes in container (bimanual)17%5/300%0/30−17 pts−34 to −2
e03Pour screws into the funnel33%10/3027%8/30−7 pts−28 to +16
All tasks17%15/909%8/90−8 pts−18 to +2
Reported in the paper48%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.

  1. e03Pour screws into the funnel
  2. Held-out benchmark environment
    e01Clamp electrical component
  3. 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.

See the code changes on GitHub⁠ (opens in a new tab)

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.