Prediction-supervised sensorimotor representations connect physical history to action generation.
PSR learns representations from multimodal sensorimotor signals by jointly forecasting future force/torque sequences, future joint-effort sequences, and terminal joint-state displacement. PSR-VLA integrates the resulting hierarchy into selected Action Expert layers through gated cross-attention.
Real-world task suite
Six tasks across three levels of contact difficulty.
PSR-VLA rollouts span coupled assembly, precise insertion, and surface contact, with 20 consecutive evaluation trials per task.
These stress tests reuse the PSR-VLA checkpoints from the main experiments without expanding the training sets or performing additional training. They are reported separately and are not included in the six-task success-rate averages.
Main-experiment checkpoint · no added training
Water-Hose Insertion
442 / 450
442/450successful trials
98.2%success rate
9test positions
50trials per position
Main-experiment checkpoint · no added training
Three-Way Interlocking Joint Assembly
48 / 50
48/50successful trials
96.0%success rate
50consecutive trials
0retries or recovery
Additional evaluations
Two behaviors beyond the main benchmark.
These supplementary experiments broaden the evaluation beyond the fixed six-task comparison and are reported separately from the success-rate averages.
Additional experiment 01
Cable & LEGO Sorting
Mixed-object sorting with deformable cable and rigid LEGO components.Additional experiment 02
Long-Horizon Servo Packing
Extended-duration execution of the servo packing workflow.
Method
The PSR-VLA Model
PSR receives compressed multiview visual tokens together with force/torque, joint-state, and joint-effort histories. A six-layer predictive Transformer learns an ordered hierarchy of sensorimotor representations that conditions selected layers of the Action Expert.
Figure 2. PSR learns a sensorimotor representation hierarchy through multi-target forecasting in Stage 1. The shared decoder predicts future joint effort, joint displacement, and force/torque sequences.Figure 3. Depth-aligned integration of PSR representations into selected Action Expert layers through gated cross-attention.
01
Predictive pretraining
The first two Transformer layers preserve modality-specific cues, the middle three exchange information through bottleneck tokens, and the final layer refines the fused representation. A shared decoder reads all six retained layers and predicts future force/torque, future joint effort, and terminal joint-state displacement.
02
Sensorimotor representation integration
During action learning, the six PSR representations are aligned in depth order with Action Expert layers 3, 6, 9, 12, 15, and 18 through gated cross-attention. The prediction decoder is frozen and no prediction loss is applied in Stage 2; learnable gates regulate each residual sensorimotor update.
Quantitative evaluation
Success across all six tasks.
Success rates are measured over 20 consecutive trials per task. Tasks follow the hard-to-easy order in the experimental setup; Overall reports total successes over 120 trials.
Real-world task success rate (%)
π0.5ForceVLAForceVLA2PSR-VLA
Success rate (%)
100806040200
HardMediumEasyOverall
45606590
Interlocking Joint
65707595
Water- Hose
60608580
3-Hole Plug
65705590
LEGO
65758095
2-Hole Plug
708075100
Blackboard Wiping
61.769.272.591.7
110/ 120
Overall
Primary result91.7%
110 / 120 successful trials
Predictive sensorimotor representation learning improves performance across difficulty levels.
Under the same backbone and evaluation protocol, PSR-VLA reaches 110/120 successful trials, compared with 74/120 for π0.5, 83/120 for ForceVLA-π0.5, and 87/120 for ForceVLA2-π0.5. These correspond to gains of 30.0, 22.5, and 19.2 percentage points, respectively. PSR-VLA achieves the highest success rate on five of the six tasks.
Across the tabulated run and three additional PSR-VLA training runs, performance is 109.5 ± 2.5 successes out of 120, or 91.3 ± 2.1% (mean ± s.d.). This stability analysis is PSR-VLA-only and is not a seed-matched comparison with the baselines.
The gains remain consistent from hard tasks (92.5% mean) to medium (85.0%) and easy tasks (97.5%). Three-Hole Plug Insertion is the only task not led by PSR-VLA, where ForceVLA2-π0.5 reaches 85% compared with 80% for PSR-VLA.
Matched qualitative comparison
Maintaining the insertion axis under contact.
Both policies begin the Three-Hole Plug task with closely matched plug-to-socket alignment. As contact loads build, π0.5 develops lateral drift and ends off-axis. PSR-VLA preserves axial alignment and directs the applied force vertically downward, completing the insertion.
π0.5 baseline
Three-Hole Plug Insertion
Failure · Lateral drift
Initial alignment is comparable, but lateral displacement grows during contact and the plug moves away from the insertion axis.
PSR-VLA · Ours
Three-Hole Plug Insertion
Success · Vertical insertion
PSR-VLA maintains axial alignment, applies a stable downward force, and completes a clean insertion.
Representative paired rollouts with comparable initial alignment. This example illustrates contact behavior and is not used to estimate the overall success rate.
Predictive representation
What PSR learns before acting.
On held-out sequences, PSR reduces normalized MSE over a persistence baseline by 54.9% for future force/torque, 48.1% for future joint effort, and 87.5% for terminal joint-state displacement.
Ground TruthPrediction
Figure 4. Stage-1 predictions on held-out offline sequences; displayed channels are selected by event-associated ground-truth variation rather than prediction accuracy.
Ablation studies
Where the predictive hierarchy matters.
Three controlled questions test predictive pretraining, ordered depth-wise representations, and multi-depth integration on Water-Hose Insertion and Two-Hole Plug Insertion. Each cell reports real-robot success together with normalized offline action error.