Research project · 2026

PSR-VLA: Predictive Sensorimotor Representation Learning for Contact-Rich Manipulation

Anonymous submission

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.

01

Hard

Coupled alignment and deformable insertion

Three-Way Interlocking Joint Assembly300 demonstrations
Water-Hose Insertion300 demonstrations
02

Medium

Precise multi-contact alignment

Three-Hole Plug Insertion200 demonstrations
LEGO Assembly200 demonstrations
03

Easy

Broader feasible contact regions

Two-Hole Plug Insertion150 demonstrations
Blackboard Wiping60 demonstrations

Evaluation protocol

One consistent real-robot protocol.

6
contact-rich tasks
1,210
expert demonstrations
20
trials per task
120
trials per method

Long-horizon stability tests

Continuous execution beyond the main evaluation.

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.

PSR predictive pretraining pipeline
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.
Depth-aligned PSR-VLA integration into Action Expert layers
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.5 ForceVLA ForceVLA2 PSR-VLA
Success rate (%)
45606590

Interlocking
Joint

65707595

Water-
Hose

60608580

3-Hole
Plug

65705590

LEGO

65758095

2-Hole
Plug

708075100

Blackboard
Wiping

61.769.272.591.7

Overall

Primary result 91.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
Predicted and measured right-arm and left-arm force, torque, and joint-effort trajectories during interlocking-joint assembly Predicted and measured right-arm and left-arm force, torque, and joint-effort trajectories during plug insertion
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.

20 robot trials · 100 offline episodes per task

Q1

Predictive Pretraining

VariantWater-Hose
Insertion
Two-Hole Plug
Insertion
Average
w/o Stage-1 PretrainingNo predictive initialization 14/2070% successMSE .005190L1 .026432 11/2055% successMSE .004378L1 .033470 62.5%mean success
PSR-VLAPrediction-pretrained model 19/2095% successMSE .003378L1 .021413 19/2095% successMSE .003816L1 .035424 95.0%mean success
Q2

Ordered Depth-Wise Representations

VariantWater-Hose
Insertion
Two-Hole Plug
Insertion
Average
Shared Deepest RepresentationReuse z6 at all six injection sites 14/2070% successMSE .004378L1 .031470 15/2075% successMSE .004066L1 .036335 72.5%mean success
Reversed AlignmentReverse representation-to-layer pairing 13/2065% successMSE .004696L1 .029665 12/2060% successMSE .004807L1 .041436 62.5%mean success
PSR-VLAOrdered layer-to-depth pairing 19/2095% successMSE .003378L1 .021413 19/2095% successMSE .003816L1 .035424 95.0%mean success
Q3

Multi-Depth Integration

VariantWater-Hose
Insertion
Two-Hole Plug
Insertion
Average
Output-Level ConditioningCondition only at the Action Expert output 10/2050% successMSE .005969L1 .035745 11/2055% successMSE .004438L1 .041181 52.5%mean success
PSR-VLASix depth-aligned injection sites 19/2095% successMSE .003378L1 .021413 19/2095% successMSE .003816L1 .035424 95.0%mean success
Ablation success counts for Water-Hose and Two-Hole Plug Insertion
Figure 5. Ablation success counts on Water-Hose Insertion and Two-Hole Plug Insertion over 20 real-robot trials per task.

Resources

Citation

Venue, paper, code, and model links will be updated when the public release information is confirmed.

@article{anonymous2026psrvla,
  title   = {PSR-VLA: Predictive Sensorimotor Representation Learning
             for Contact-Rich Manipulation},
  journal = {Under Review},
  year    = {2026}
}