When Instructions Retrieve Trajectories: Diagnosing and Mitigating Generalization Failures in VLA Models
Organizations: National Tsing Hua University · National Taiwan University
Abstract
Vision-language-action (VLA) models can exceed 90% success on in-distribution tasks and withstand nuisance changes that preserve the required action, yet fail under counterfactual changes that demand a different action. Aggregate robustness scores can therefore conceal a more specific failure, in which a policy responds to both language and vision yet does not combine them to select the action the task requires. We call this failure instruction-action binding. Instructions cue familiar trajectory families, and visual feedback adjusts their execution. Behavioral analyses of fine-tuned and GR00T-N1.7 policies reveal that failed rollouts often retain the source behavior or switch to another demonstrated task. These switches show that language is not simply ignored. Readouts and interventions connect these choices to task-conditioned internal states. Our analysis of the imitation objective shows how narrow conditional action support can leave grounded and instruction-keyed solutions indistinguishable on the demonstrations. This motivates Equivariant Counterfactual Training (ECT), which acts at two levels. ECT data supply valid demonstrations in which the same instruction requires different actions in distinguishable scenes, while the ECT loss trains each demonstration with its counterpart in the same update. In a controlled LIBERO-PRO comparison, full ECT raises 's mean position-swap success from 36% to 59%. On CALVIN, where counterparts already occur in the original data, the ECT loss improves five-task completion without new demonstrations. On a real UR5e under a fixed demonstration budget, full ECT raises unseen-position success from 8% to 88%.
Figures & tables
| Cell | Type | Binding pattern | ||
|---|---|---|---|---|
| Semantic (Sem.) | 0 | 0 | Nuisance | familiar action remains valid |
| Object (Obj.) | 0 | 0 | Nuisance | familiar action can remain valid |
| Swap | 0 | 1 | Counterfactual | same key, different required action |
| Task | 0 or 1 | 1 | Counterfactual | source retention or other-task retrieval |
| Spatial | Object | Goal | Long | |||||||||||||
| Model | Swap | Task | Swap | Task | Swap | Task | Swap | Task | ||||||||
| Frozen LM (vision encoder and action expert trained) | ||||||||||||||||
| Standard | ||||||||||||||||
| ECT data + ECT loss | (+21.7) | ( 0.2) | (+32.5) | (+8.3) | (+14.2) | (+15.1) | (+22.0) | (+3.0) | ||||||||
| Full FT (all parameters trained) | ||||||||||||||||
| Official | ||||||||||||||||
| Spatial | Object | Goal | Long | |
| Standard | 52.7 | 38.3 | 41.3 | 12.7 |
| ECT data | 72.1 | 71.1 | 51.8 | 29.5 |
| ECT data + ECT loss | 74.4 | 70.8 | 55.5 | 34.7 |
| (ECT data Standard) | ||||
| (ECT data + ECT loss ECT data) |
Appendix figures & tables33 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning |
|---|---|
| Instruction-action binding framework | |
| Demonstration dataset. | |
| Observation, instruction, and expert trajectory from task . | |
| Task identity. | |
| Conditional entropy: remaining uncertainty about task identity after observing cue , such as instruction or observation . | |
| Task-level action requirement, abstracting from incidental execution variation. | |
| Experiment | Evaluation unit | Measurement |
|---|---|---|
| LIBERO-PRO diagnosis | Perturbation rollout | Terminal-state simulator predicate |
| Human rollout labels | Rollout video | Behavioral category |
| Prefix-KV retrieval | same_to_other rollout | Nearest-neighbor executed-task recovery |
| Persistent intervention | Rollout | End-effector proximity and simulator predicate |
| GR00T-N1.7 | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Suite | ID | Sem. | Obj. | Swap | Task | ID | Sem. | Obj. | Swap | Task |
| Spatial | 98.4 | 98.0 | 98.0 | 46.6 | 53.0 | 93.6 | 88.8 | 93.0 | 1.2 | 51.4 |
| Object | 98.6 | 99.0 | 94.4 | 18.2 | 11.0 | 95.6 | 97.6 | 86.0 | 0.0 | 9.0 |
| Goal | 96.4 | 94.6 | 88.8 | 34.2 | 21.2 | 94.2 | 93.8 | 76.0 | 2.2 | 10.0 |
| Long | 91.8 | 90.8 | 66.8 | 9.4 | 17.0 | 89.0 | 88.4 | 54.6 | 0.2 | 10.2 |
| Fine-grained label | Coarse group | Definition |
|---|---|---|
| same_to_origin | Source retrieval | Clearly executes the original source behavior |
| close_to_origin | Source retrieval | Approximately follows the source behavior |
| same_to_other | Other-task retrieval | Matches another familiar demonstrated task |
| same_to_change | Correct or near-correct | Executes the modified instruction correctly |
| close_to_change | Correct or near-correct | Approximately follows the modified instruction |
| mix | Hybrid retrieval | Combines source, changed, or neighboring sub-trajectories |
| Granularity | Agreement | Cohen’s | Macro F1 | |
|---|---|---|---|---|
| Fine labels | 117 | 73.5 [65.0, 81.2] | 0.611 [0.504, 0.712] | 45.9 [35.5, 54.0] |
| Coarse 5-class | 117 | 83.8 [76.9, 90.6] | 0.731 [0.617, 0.833] | 58.5 [47.0, 67.6] |
| Retrieval-like binary | 117 | 90.6 [84.6, 95.7] | 0.778 [0.639, 0.894] | 88.9 [81.9, 94.7] |
| Suite | Rule | Source | Other-task | Hybrid | Correct | Collapse | |
|---|---|---|---|---|---|---|---|
| Goal | 33 | 27 | 2 | 0 | 0 | 4 | |
| 33 | 12 | 12 | 0 | 9 | 0 | ||
| 21 | 6 | 9 | 0 | 6 | 0 | ||
| 30 | 7 | 21 | 0 | 0 | 2 | ||
| 3 | 3 | 0 | 0 | 0 | 0 | ||
| total | 120 | 55 | 44 | 0 | 15 | 6 |
| Trajectory metric | Usable rows | Label-consistent NN |
|---|---|---|
| Object+EEF DTW | 99 | 87/99 (87.9%) |
| EEF+gripper resampled distance | 99 | 83/99 (83.8%) |
| Condition | Observed behavior | |
| Same instruction, swapped object + 10 cm shift | 9 | 9 / 9 follow the swapped-and-shifted source-bound object |
| Suite | Rule | Description |
| Spatial | R1 | Replace the target object |
| R2 | Alter the spatial description of the source location (object repositioned to match), with two variants | |
| R3 | Replace the placement destination | |
| Object | R1 | Swap the target object’s position with another in-scene object |
| R2 | Slightly shift the target object’s position | |
| stress | Swap and shift simultaneously |
| Source completion | NN source | |||||
| Suite | Rule | Official | ECT | Official | ECT | |
| Goal | 33 | 84.8 | 66.7 | 90.9 | 78.8 | |
| 33 | 39.4 | 9.1 | 45.5 | 36.4 | ||
| 21 | 38.1 | 19.0 | 38.1 | 19.0 | ||
| 30 | 16.7 | 0.0 | 20.0 | 3.3 | ||
| total | 117 | 46.2 | 24.8 | 50.4 | 36.8 | |
| Official | Official GR00T-N1.7 | + ECT | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Suite | Follow | Retr. | Coll. | Follow | Retr. | Coll. | Follow | Retr. | Coll. | |
| Spatial | 120 | 42.5 | 51.7 | 5.8 | 33.3 | 54.2 | 12.5 | 64.2 | 31.7 | 4.2 |
| Object | 69 | 40.6 | 58.0 | 1.4 | 15.9 | 84.1 | 0.0 | 75.4 | 24.6 | 0.0 |
| Goal | 120 | 12.5 | 82.5 | 5.0 | 0.8 | 71.7 | 27.5 | 26.7 | 60.8 | 12.5 |
| Long | 72 | 9.7 | 87.5 | 2.8 | 0.0 | 97.2 | 2.8 | 27.8 | 72.2 | 0.0 |
| All | 381 | 26.5 | 69.3 | 4.2 | 13.6 | 73.2 | 13.1 | 47.5 | 47.2 | 5.2 |
| Patch | Target | First L2 impr. | Pos. frac. | Chunk L2 impr. | Pos. frac. |
|---|---|---|---|---|---|
| Source final | Source | 0.69 | 0.87 | ||
| Executed final | Executed | 0.62 | 0.62 | ||
| Source all | Source | 0.42 | 0.84 | ||
| Executed all | Executed | 0.42 | 0.62 |
| Analysis | Own target | Off-target | Specificity |
|---|---|---|---|
| Cosine alignment | |||
| Normalized projection | |||
| Raw projection |
| Model | Injected closer | Control closer | Pairs, larger shift | Pairs 4/4 | Pairs 0/4 | Control completes source | Injected completes target |
|---|---|---|---|---|---|---|---|
| (prefix-KV) | 109/200 | 0/200 | 50/50 | 25 | 21 | 200/200 | 0/200 |
| GR00T-N1.7 (backbone features) | 121/200 | 0/200 | 50/50 | 28 | 19 | 200/200 | 0/200 |
| Construction | ID | Sem. | Obj. | Swap |
|---|---|---|---|---|
| No ECT baseline | 94.2 | 94.8 | 84.4 | 0.0 |
| X-mirror, paired | 23.8 | 22.6 | 25.4 | 3.0 |
| Y-mirror, paired | 93.2 | 96.4 | 85.0 | 1.0 |
| All mirrors, paired | 29.0 | 21.8 | 20.2 | 12.0 |
| X-mirror + X-shift, paired | 97.4 | 99.2 | 95.8 | 1.6 |
| XY-mirror + XY-shift, paired | 95.6 | 98.6 | 91.4 | 7.6 |
| Suite | Transform family | Action transform | Role in ECT |
|---|---|---|---|
| Spatial | mirror | reflected, then replayed | changes placement direction under fixed instruction |
| Object | Y-mirror / mirror+shift | reflected (x and xy also shifted), then replayed | preserves separability while changing source-object branch |
| Goal | mirror | reflected, then replayed | changes goal-relative spatial branch |
| Long | shift-augmented mix | shifted, then replayed | preserves long-horizon executability |
| Median distance to | Minimum distance | Within ID scale | |||
|---|---|---|---|---|---|
| Suite | ID scale | original | transformed | to transformed | of transformed |
| Spatial | 1.0 | 6.6 | 28.7 | 13.3 | 0% |
| Object | 0.3 | 6.4 | 28.5 | 25.2 | 0% |
| Goal | 0.9 | 11.3 | 24.9 | 19.6 | 0% |
| Long | 1.7 | 13.9 | 16.2 | 6.4 | 0% |
| Recipe | Configuration | LM | ViT | AE | Trainable |
|---|---|---|---|---|---|
| Frozen LM | _expert_only | frozen | trained | trained | 844.9M |
| Full FT | pi05_libero (official), _full_sft (ECT) | trained | trained | trained | 3,353.4M |
| Suite | Model | Recipe | ID | Sem. | Obj. | Swap | Task |
|---|---|---|---|---|---|---|---|
| Spatial | Official | Full FT | 98.4 | 98.0 | 98.0 | 46.6 | 53.0 |
| ECT data + ECT loss | Full FT | 95.8 | 94.0 | 95.6 | 72.8 | 53.2 | |
| Standard † | Frozen LM | 95.9 | 94.5 | 95.4 | 52.7 | 22.3 | |
| ECT data † | Frozen LM | 96.5 | 93.0 | 95.2 | 72.1 | 18.5 | |
| ECT data + ECT loss † | Frozen LM | 98.4 | 96.1 | 97.1 | 74.4 | 22.1 | |
| Object | Official | Full FT | 98.6 | 99.0 | 94.4 | 18.2 | 11.0 |
| Suite | Model | Training scope | ID | Sem. | Obj. | Swap | Task |
|---|---|---|---|---|---|---|---|
| Spatial | Official | Per suite | 93.6 | 88.8 | 93.0 | 1.2 | 51.4 |
| ECT data | Per suite | 87.0 | 91.4 | 83.2 | 17.8 | 53.0 | |
| ECT data | Four suites | 90.8 | 88.0 | 89.8 | 38.2 | 64.0 | |
| Object | Official | Per suite | 95.6 | 97.6 | 86.0 | 0.0 | 9.0 |
| ECT data | Per suite | 97.6 | 98.2 | 88.6 | 0.0 | 10.0 | |
| ECT data | Four suites | 97.2 | 97.2 | 79.8 | 7.0 | 10.2 |
| ID | Swap | Task | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Suite | 25% | 50% | 100% | 25% | 50% | 100% | 25% | 50% | 100% |
| Spatial | 95.2 | 95.8 | 90.0 | 53.6 | 57.4 | 55.4 | 22.6 | 26.8 | 21.2 |
| Object | 98.2 | 98.6 | 97.8 | 39.6 | 35.8 | 23.8 | 20.2 | 36.4 | 16.6 |
| Goal | 90.6 | 93.4 | 93.6 | 36.0 | 40.8 | 40.2 | 16.2 | 20.6 | 19.4 |
| Long | 94.0 | 90.4 | 94.4 | 14.4 | 15.2 | 15.2 | 16.2 | 24.0 | 15.0 |
| Suite | Model | ID | Sem. | Obj. | Swap | Task |
|---|---|---|---|---|---|---|
| Spatial | Standard, then | 96.0 | 92.0 | 93.0 | 55.4 | 20.2 |
| Spatial | ECT data, then | 95.0 | 91.4 | 94.4 | 69.4 | 28.2 |
| Spatial | ECT data + ECT loss, then | 96.4 | 94.0 | 95.0 | 75.0 | 15.2 |
| Object | Standard, then | 98.0 | 98.0 | 83.0 | 22.4 | 16.8 |
| Object | ECT data, then | 99.2 | 99.4 | 89.0 | 68.6 | 39.0 |
| Object | ECT data + ECT loss, then | 99.4 | 99.0 | 81.0 | 68.4 | 30.6 |
| Suite | Cell | Seed 0 | Seed 1 | Seed 42 | Mean | Std. |
|---|---|---|---|---|---|---|
| Spatial | ID | 98.2 | 99.0 | 98.0 | 98.4 | 0.5 |
| Spatial | Sem. | 97.0 | 96.0 | 95.2 | 96.1 | 0.9 |
| Spatial | Obj. | 97.2 | 96.6 | 97.6 | 97.1 | 0.5 |
| Spatial | Swap | 75.4 | 75.0 | 72.8 | 74.4 | 1.4 |
| Spatial | Task | 21.8 | 28.8 | 15.6 | 22.1 | 6.6 |
| Object | ID | 99.0 | 99.6 | 98.8 | 99.1 | 0.4 |
| Model | Suite | Comparison | Swap [95% CI] | ID [95% CI] |
|---|---|---|---|---|
| Spatial | ECT data + ECT loss (Frozen LM) vs. official | [+22.7, +32.7] | [-1.3, +1.4] | |
| Object | ECT data + ECT loss (Frozen LM) vs. official | [+47.3, +57.9] | [-0.6, +1.8] | |
| Goal | ECT data + ECT loss (Frozen LM) vs. official | [+16.5, +26.2] | [-2.9, +1.4] | |
| Long | ECT data + ECT loss (Frozen LM) vs. official | [+20.0, +31.3] | [-1.7, +3.9] | |
| GR00T-N1.7 | Spatial | ECT data, per suite vs. official | [+13.2, +20.3] | [-10.3, -2.9] |
| GR00T-N1.7 | Object | ECT data, per suite vs. official | [-0.8, +0.8] | [-0.3, +4.4] |
| Scope | Retrieval-like | Correct/near-correct | Collapse |
|---|---|---|---|
| All probes | 264/381 ( [64.5, 73.7]) | 101/381 ( [22.3, 31.2]) | 16/381 ( [2.6, 6.7]) |
| Goal | 99/120 ( [74.7, 88.3]) | 15/120 ( [7.7, 19.6]) | 6/120 ( [2.3, 10.5]) |
| Long | 63/72 ( [77.9, 93.3]) | 7/72 ( [4.8, 18.7]) | 2/72 ( [0.8, 9.6]) |
| Object | 40/69 ( [46.2, 68.9]) | 28/69 ( [29.8, 52.4]) | 1/69 ( [0.3, 7.8]) |
| Spatial | 62/120 ( [42.8, 60.4]) | 51/120 ( [34.0, 51.4]) | 7/120 ( [2.9, 11.6]) |