Structured Action-Space
Exploration for Flow-Based VLAs

Comparison of in-chain and action-space structured exploration for flow-based VLA reinforcement learning.

Exploration should preserve the structure of robot actions.

Flow-based vision-language-action models are powerful robot policies, but their usual reinforcement-learning exploration lives inside a multi-step denoising chain. That makes it difficult to control the actions that are ultimately executed.

We identify structured noise dilution: temporal correlations and action-group scales injected partway through the chain can be weakened by the remaining denoising steps. StructRL instead places exploration directly at the action output, where its intended structure reaches the robot intact.

Deterministic decoding.
Structured exploration.

StructRL decodes a clean action chunk with a deterministic ODE, then adds noise directly in action space. The noise is smooth over the chunk and learns separate scales for position, rotation, and gripper controls.

Last-step replay supplies a tractable policy-gradient signal to the flow decoder, without assigning likelihoods to intermediate denoising states.

StructRL framework: deterministic reverse flow, structured action-space noise, and last-step replay.
StructRL decodes an action deterministically, adds structured noise at the endpoint, and replays the final step for learning.
01

Action-space noise

Exploration is attached to the executed action, avoiding dilution.

02

AR(1) correlation

Smooth noise matches the temporal structure of action chunks.

03

Group-aware scales

Position, rotation, and gripper controls explore at appropriate magnitudes.

04

Last-step replay

An efficient learning signal for flow-decoder post-training.

Structured exploration improves performance across simulation benchmarks.

We evaluate StructRL on LIBERO and ManiSkill with three flow-based VLA backbones. Green marks StructRL; bold marks the best result within each backbone group.

Table 1

LIBERO success rate (%)

BackboneMethodSpatialObjectGoalLongAvg.
GR00T N1.5Few-shot SFT41.458.648.261.952.5
πRL (Flow-SDE + PPO)96.6100.093.895.696.5
Baseline96.299.491.895.695.8
StructRL99.2100.097.699.099.0
π0Few-shot SFT65.364.449.851.257.6
πRL (Flow-SDE + PPO)98.499.496.290.296.0
π-StepNFT93.598.083.786.790.5
Baseline98.899.097.290.296.3
StructRL98.899.898.491.897.2
π0.5Few-shot SFT84.695.484.643.977.1
πRL (Flow-SDE + PPO)99.6100.098.893.097.9
π-StepNFT97.8100.098.279.894.0
Baseline99.4100.093.886.094.8
StructRL99.699.898.890.297.2
Table 2

ManiSkill success rate (%)

BackboneMethodINDVisionSemanticExecutionOOD Avg.
π0SFT38.432.68.413.218.1
πRL (Flow-SDE + PPO)78.861.125.431.539.3
π-StepNFT79.269.149.133.150.4
Baseline84.179.463.359.167.3
StructRL87.880.064.256.266.8
π0.5SFT40.138.816.222.325.9
πRL (Flow-SDE + PPO)90.968.034.545.449.3
π-StepNFT85.476.956.645.159.5
Baseline85.976.262.259.465.9
StructRL90.978.368.464.270.3
Ablation of the action-space noise level.
Moderate noise enables stable improvement.
Ablation comparing noise injection schemes.
Structured action-space noise explores most efficiently.
Ablation of train-inference consistency.
Matching replay and inference improves learning.
Ablation of the number of denoising steps.
Last-step replay is robust across flow depths.

From ten demonstrations to reliable real-world manipulation.

We initialize each policy with SFT on 10 collected trajectories, then continue online reinforcement learning with StructRL + AWAC.

10 trajectoriesSFTStructRL + AWAC
5× speed
Pick Banana · online RL rollout · 5× speed

Task 01

Pick Banana

4%84%

Success rate after online RL improves by 80 percentage points over the SFT-initialized policy.

5× speed
Plug Charger In · online RL rollout · 5× speed

Task 02

Plug Charger In

0%100%

Online adaptation reaches perfect success on this precise insertion task.