atlas

Research map / Reinforcement learning / Visual and representation RL

PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Control

ICML 2024 · Reinforcement learning › Visual and representation RL · approach: Skill learning and diffusion policies · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes PRISE, which quantizes continuous actions into discrete codes with a state-conditioned VQ module and applies BPE to learn variable-horizon skill tokens that improve behavior cloning on MetaWorld and LIBERO.

Problem

Learning temporal action abstractions (variable-timespan skills) from multitask robotic demonstrations to improve downstream behavior cloning in continuous control with pixel observations.

Approach

Treat temporal action abstraction as sequence compression: discretize continuous actions via a state-conditioned vector-quantized action encoder trained with a latent forward-dynamics (BYOL-style) objective, then apply byte pair encoding over the code sequences to induce variable-length skill tokens used by a high-level BC policy.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use PRISE skill tokens instead of removing BPE temporal skill tokens (ablation)PRISE on MetaWorld 5-shot IL · MetaWorld 5-shot IL, 5 unseen taskssuccess rate: 60.1± 1.5 -> 66.8± 1.8Helped
Use PRISE skill tokens instead of removing BPE temporal skill tokens (ablation)PRISE on LIBERO 5-shot IL · LIBERO 5-shot IL, 8 unseen taskssuccess rate: 43.5± 0.6 -> 52.7± 0.9Helped
Add the latent forward-dynamics objective (ablation)PRISE action quantization · LIBERO action-code separationζ: 14.53 -> 72.42Helped
Use the PRISE-pretrained encoder for BCBC on MetaWorld · MetaWorld 5-shot IL, 5 unseen taskssuccess rate: 17.3± 4.2 -> 57.9± 1.4Helped
Use the PRISE-pretrained encoder for ACTACT on MetaWorld · MetaWorld 5-shot IL, 5 unseen taskssuccess rate: 55.0± 1.8 -> 59.3± 2.0Helped
Use the PRISE-pretrained encoder for BCBC on LIBERO · LIBERO 5-shot IL, 8 unseen taskssuccess rate: 40.4± 1.2 -> 39.0± 4.1Hurt
Use the PRISE-pretrained encoder for ACTACT on LIBERO · LIBERO 5-shot IL, 8 unseen taskssuccess rate: 41.5± 2.1 -> 42.0± 3.2No clear effect
Pretrain PRISE skills with more trajectories per task (ablation)PRISE downstream adaptation · LIBERO 5-shot ILsuccess rateHelped
Vary BPE vocabulary size across the reported range (ablation)PRISE downstream adaptation · LIBERO 5-shot IL, 8 unseen taskssuccess rateMixed
Increase the number of action codes to 40 or 50 (ablation)PRISE temporal abstraction · LIBERO 5-shot ILsuccess rateHurt

Surprising findings

Limitations

Experimental setup

Datasets: LIBERO-90 LIBERO-LONG MetaWorld

Benchmarks: LIBERO-90 LIBERO-LONG (8 unseen tasks, 5-shot) MetaWorld (5 unseen tasks, 5-shot)

Metrics: success rate

Baselines: BC (ResNet-RNN) BC (ResNet-T) BC (ViT-T) ACT PRISE w/o BPE BC from scratch

Builds on: ACT VQ-VAE Byte Pair Encoding BYOL ResNet-T

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy