OAT

Ordered Action Tokenization

How to tokenize continuous actions into discrete tokens for input or generation? OAT represents actions as a sequence of discrete tokens instead of outputting a single vector at once. Since actions are typically trajectories over time, they are grouped into chunks.

Creating latents with Transformer encoder: action chunk → Transformer → latent register tokens, where register tokens are slots that summarize the actions. Each latent is converted to a codebook index using FSQ (Finite Scalar Quantization). Using nested dropout + causal attention, they are sorted by information hierarchy. This allows reconstruction even with just the prefix.