Pith. sign in

REVIEW 3 major objections 6 minor 27 references

Generalized Locomotion in Out-of-distribution Conditions with Robust Transformer

T0 review · 3 major / 6 minor · reviewed 2026-08-06 · deepseek-v4-flash

Pith's one-line read A transformer policy trained on clean walking and one leg failure can keep a robot walking through unseen leg failures and sensor corruption.

desk verdict A strong hexapod damage result makes this worth reviewing, but the action-token mechanism is under-identified and several plots lack statistics. read the letter →

arxiv 2507.04039 v1 pith:IEJQ7NKD submitted 2025-07-05 cs.RO

classification cs.RO
keywords leggedlocomotionout-of-distributiongeneralizationtransformerpolicybodytokenizationconsistentdropoutfaulttolerancereinforcementlearningperceptualnoise
verification ladder T0 review T1 audit T2 compute T3 formal

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The reading

ROLT is a transformer policy for legged robots that is claimed to keep walking in situations that differ from training, without domain randomization or an online adaptation module. The controller is built around two ideas: body tokenization, in which each leg is a token that shares attention with the other legs and queries an action token instead of a learned position embedding, and consistent dropout, in which the same neuron mask and patch-position mask are applied during rollout and update. Trained only on healthy walking plus a single limping leg on a quadruped, or a handful of joint failures on a hexapod, the policy is reported to stay upright through unseen leg failures and to tolerate elevation-map corruption when trained on clean observations. The paper's claim is that these network-structure choices, rather than more elaborate training schemes, are what give the controller its out-of-distribution generalization.

What carries the argument

The load-bearing mechanism is the tokenization of the robot body into leg tokens, each containing ten steps of joint positions, joint velocities, previous actions, velocity command, and IMU history, projected by a shared MLP; a linear embedding of the previous action forms an action token. The action token replaces learnable position embeddings as the source of position information, so legs can query their roles through attention, allowing experience to transfer from one limb to another. Robustness to perceptual noise is carried by two consistent dropout operations in the two-layer encoder: feature dropout at rate 0.1 after attention and feed-forward layers, and patch position-embedding dropout at rate 0.75. Consistency means the same dropout mask is used at rollout and at update, which allows a high dropout rate to be used without destabilizing policy-gradient training.

What would settle it

A direct test would be to train ROLT exactly as described, with healthy walking plus one locked leg, and then evaluate it on failures outside the tested set: a hip joint locked, a leg with two joints disabled, partial motor weakening rather than full lock, or a change in payload. If the robot falls within seconds on any of these, the claim that one leg's fault experience transfers to arbitrary new limb failures would be falsified. Similarly, injecting noise into proprioception while keeping the elevation map clean would test whether the robustness mechanism really re-anchors on trusted body state or only on the specific perception corruption studied.

Watch

Extended reading notes

Core claim

On the paper's own terms, the central discovery is that a transformer can transfer fault-tolerance experience across limbs if position information is made data-dependent. Instead of adding fixed position embeddings to leg tokens, ROLT appends an action token encoding the previous action, and each leg token infers its spatial role through attention to this token. In the hexapod damage test, this design keeps the robot from falling in all nine unseen joint-failure conditions over 20 seconds, while the baselines fall in most of the same tests. The consistent-dropout half of the claim is that dropping features and patch position embeddings, with the same mask during rollout and update, produces a policy that handles high- and low-frequency perceptual noise even though it was trained with clean elevation maps. Attention visualizations are used to argue that position-embedding dropout makes the network shift its focus to trustworthy patches instead of anchoring to learned positions.

Load-bearing premise

The load-bearing premise is that the held-out failures tested (three other limping legs, nine joint disables, and three real-world perturbations) stand in for the entire class of out-of-distribution dynamic and perceptual changes; the paper does not formally characterize that class, so if the success is specific to these particular failures or to the simulation distribution, the broad robustness claim would not follow.

Editorial extensions

If this is right

  • If the central claim is right, a locomotion policy does not need to be trained in every failure mode; one or a few fault experiences can generalize to other limbs of the same body plan.
  • The same architecture, with the number of leg tokens changed, transfers from a four-legged to a six-legged robot, suggesting the controller is scalable to morphologies built from repeated limb modules.
  • Because noise robustness comes from a dropout strategy and attention rather than from adding Gaussian noise to training observations, policies can be trained on clean data and still deployed where sensors are corrupted or missing.
  • The hexapod's redundant joints can be exploited by the controller to keep walking when motors are disabled, so the method could be used for fault-tolerant operation on robots whose training time must be kept short.

Reading between the lines

Editorial extensions of the paper, not claims the author makes directly.

  • A natural extension the paper does not test is whether body tokenization transfers to any control problem with modular repeated structure, such as multi-finger hands, multi-robot teams, or configurable limbs, where combinatorial generalization across modules is desired.
  • The position-embedding dropout result suggests a general recipe for sensor-fusion robustness: make the network doubt the spatial anchoring of one modality so it re-anchors using a trusted modality; the paper demonstrates this with elevation maps, but it may extend to depth cameras, lidar, or tactile arrays.
  • A stress test the paper does not run is corruption of the proprioceptive channel itself; because the attention re-anchors on proprioception, noise injected into joint encoders or IMUs could be the first failure point.
  • The training set is framed as limited, but the experiments cover only discrete lock-or-disable joint failures; whether the mechanism extends to continuous parameter drifts, such as gradually weakening motors or changing payload, is left open.
Share X Bluesky LinkedIn Reddit HN

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, and a circularity audit.

Referee Report

3 major / 6 minor

Summary. The paper proposes ROLT, a transformer-based locomotion policy whose two main design elements are body tokenization (each leg is a token, plus an action token that replaces data-independent position embeddings) and consistent dropout (feature dropout and patch-PE dropout applied with the same mask during rollout and update). The policy is trained with PPO in IsaacGym on the Unitree A1 and a hexapod, using only a few damaged conditions and clean elevation observations. Evaluation covers simulated limb stiffness changes, unseen joint failures on the hexapod, low- and high-frequency elevation-map noise, and three real-world perturbations on a Unitree Go1. The paper claims that ROLT generalizes across limbs and to unseen perceptual noise better than MLP, Roll-Drop, RMA, and ROAM baselines.

Significance. If the reported results hold, this is a useful empirical contribution: it demonstrates that a transformer with body-level tokenization can transfer fault-tolerant behavior across limbs after training on only a handful of damage conditions, and that a consistent-dropout scheme can improve robustness to observation corruption without explicit noise injection during training. The hexapod damage results in Table II, which include means and standard deviations, are concrete evidence for the dynamics-generalization claim, and the real-world transfer experiments add practical credibility. The central architectural interpretation, namely that the action token acts as a data-dependent position encoding, is plausible but is not decisively supported by the current ablations; this should be fixed before the mechanism claim is accepted.

major comments (3)
  1. [Sec. III-A, Eq. (2) and Sec. IV-A, Module PE ablation] The Module PE ablation changes two variables at once: it removes the action token and it adds learnable per-leg position embeddings. The action token is not only a positional query; it is also the only token that carries the previous target angles of all joints, i.e., a global action-context signal. Therefore the large gap between ROLT and Module PE in the hexapod damage test (Table II) could be explained by the loss of this global context, or by the introduction of per-leg embeddings, rather than by the absence of a data-dependent position-query mechanism. I agree with the stress-test concern that this is a real confound. Please add ablations that separate these factors, such as ROLT with per-leg learnable PE added in addition to the action token, or Module PE with a non-positional global token carrying the same action vector, and/or attention analyses showing that leg tokens use the action token specifically for position information.
  2. [Sec. IV-A, Fig. 3 and Sec. IV-B, Figs. 6-7] The A1 stiffness test and both perceptual-noise tests are presented only as plots, with no numeric tables, no number of seeds, no standard deviations, and no test-episode counts. These experiments are load-bearing for the paper's claims of dynamics generalization and perceptual robustness, so without variance information the reader cannot assess whether ROLT's advantage is statistically meaningful or whether the curves reflect one or a few favorable runs. Please report mean/std tables and seed counts for the stiffness and noise experiments, as is already done for the hexapod damage test.
  3. [Sec. IV-C, Table III] The real-world experiments report means and standard deviations but do not state the number of trials per condition, and they compare against only RMA. Given the small distances involved (0.0-4.8 m over 10 s) and the strength of the claim that ROLT generalizes in real-world deployment, this table needs the number of runs per condition and ideally per-trial results. Without these details, the real-world conclusion is difficult to evaluate.
minor comments (6)
  1. [Sec. IV-B, Figs. 6-7] The figure captions appear to be swapped: the text says low-frequency noise results are shown in Figure 6 and high-frequency noise results in Figure 7, but the captions read "Fig. 6. High-frequency noise test results" and "Fig. 7. Low-frequency noise test results."
  2. [Table II] The fall-time entry for HRC is truncated as "6.74±3."; please provide the full value.
  3. [Throughout] There are several typos and grammar issues, including "prioprocetion," "hexpodal," "contary," "wired gait" (should be "weird gait"), "Gussian," "a extremely," and "both ROLT maintain." A careful proofreading pass is needed.
  4. [Sec. IV-D, Fig. 8] The attention visualization is described as averaging "the attention score of the map patches on each leg token," but the text does not specify which attention heads or layers are averaged; please specify the exact aggregation to make the analysis reproducible.
  5. [Table I] The foot airtime reward term is listed without a coefficient, while all other reward terms have one; please clarify whether it has a coefficient or is intentionally unweighted.
  6. [Sec. III-C] The text says that consistent dropout applies the same mask during rollout and update, but it is not stated whether the same mask is shared across parallel environments or across time steps; please clarify the implementation.

Circularity Check

0 steps flagged · score 0.0 of 10

No significant circularity: ROLT's generalization and robustness claims are empirical outcomes evaluated against external baselines and ablations, not reduced to fitted inputs or self-citations.

full rationale

The paper's central claims are empirical: training a transformer policy with body tokenization and consistent dropout in limited dynamic and clean-observation settings, then evaluating under held-out leg failures and sensor noise. Nothing in the method fits a parameter to the test conditions. The training distribution is explicitly disjoint from test failures: for the stiffness test, the paper trains in healthy conditions and front-right limping and tests when two other legs are limping; for the hexapod damage test, training disables only FRT, MRT and MLC while the nine test conditions are all the remaining calf and thigh joints. The perceptual-noise tests train with clean elevation maps and inject noise only at test time. Dropout rates are fixed hyperparameters (feature dropout 0.1, PE dropout 0.75), not tuned on test outcomes. The comparisons against MLP, Roll-Drop, RMA, ROAM, and the Module PE / No PE Drop / No Feature Drop ablations are external benchmarks, so the results are falsifiable rather than definitionally guaranteed. The action-token position-query explanation is potentially under-identified because the Module PE ablation changes two factors at once, but this is a confound in an ablation, not circularity: the empirical generalization result does not depend on that explanation being uniquely identified. References to consistent dropout [23] and Roll-Drop [24] are prior external work, not self-citations, and no uniqueness theorem is imported from the authors' own papers. I find no step where an equation or claim reduces, by construction, to its own input.

Assumptions & free parameters 4 free parameters · 4 assumptions · 0 invented entities

The empirical method introduces no new physical entities. The central claims rest on domain assumptions about body symmetry, proprioception reliability, and the representativeness of the small set of training failures. Hand-chosen architecture hyperparameters are listed as free parameters. No circular derivation is present; the evaluation is against external baselines.

free parameters (4)
  • feature dropout rate = 0.1
    Hand-set; the 'No Feature Drop' ablation shows it contributes to high-frequency noise robustness (Sec. IV-B).
  • PE dropout rate = 0.75
    Hand-set; intended to prevent the transformer from relying on patch position embeddings when exteroception is noisy (Sec. III-C).
  • hidden dimension = 160
    Chosen to keep the transformer lightweight for training on a single GPU (Sec. III training details).
  • number of transformer encoder layers = 2
    Chosen as a lightweight design; no systematic sweep is reported.
assumptions (4)
  • domain assumption Legs of a legged robot have similar structure and can share control knowledge through tokenization
    Justifies body tokenization; stated in Sec. III-A.
  • domain assumption The action token supplies sufficient data-dependent positional information to leg tokens
    The design replaces learnable per-leg position embeddings with an action token; experiments show it helps, but the exact mechanism is not proven (Sec. III-A).
  • domain assumption Proprioception is more precise than exteroception under noise
    Motivates PE dropout; stated in Sec. III-C.
  • domain assumption Training with a limited set of failures plus healthy walking is representative for generalization to other failures
    Underlies the OOD generalization claims; the paper provides no formal coverage argument (Sec. IV-A).

how reviews work

0 comments
Cite this review

Pith. "Pith review of Generalized Locomotion in Out-of-distribution Conditions with Robust Transformer." pith.science (2026). https://pith.science/paper/IEJQ7NKD

@misc{pith2026250704039,
  author       = {Pith},
  title        = {Pith review of: Generalized Locomotion in Out-of-distribution Conditions with Robust Transformer},
  year         = {2026},
  howpublished = {\url{https://pith.science/paper/IEJQ7NKD}},
  note         = {Machine review of arXiv:2507.04039}
}
read the original abstract

To succeed in the real world, robots must deal with situations that differ from those seen during training. Those out-of-distribution situations for legged robot mainly include challenging dynamic gaps and perceptual gaps. Here we study the problem of robust locomotion in such novel situations. While previous methods usually rely on designing elaborate training and adaptation techniques, we approach the problem from a network model perspective. Our approach, RObust Locomotion Transformer(ROLT),a variation of transformer,could achieve robustness in a variety of unseen conditions. ROLT introduces two key designs: body tokenization and consistent dropout. Body tokenization supports knowledge share across different limbs, which boosts generalization ability of the network. Meanwhile, a novel dropout strategy enhances the policy's robustness to unseen perceptual noise. We conduct extensive experiments both on quadruped and hexapod robots. Results demonstrate that ROLT is more robust than existing methods. Although trained in only a few dynamic settings, the learned policy generalizes well to multiple unseen dynamic conditions. Additionally, despite training with clean observations, the model handles challenging corruption noise during testing.

Figures

Figures reproduced from arXiv: 2507.04039 by the authors.

Figure 1
Figure 1. We present a locomotion controller to generalize under various out-of-distribution conditions. The pictures with green edges indicate the training distribution while the red ones indicate the test OOD distributions. While training under limited dynamic situations and clean observation, the controller can generalize to various unseen dynamics and sensor noise. Abstract—To succeed in the real world, robots must deal w… view at source ↗
Figure 2
Figure 2. Model overview. We propose a multi-modal transformer to fuse exteroception and prioproception. The prioproception is tokenized into body [PITH_FULL_IMAGE:figures/full_fig_p003_2.png] view at source ↗
Figure 3
Figure 3. ROLT has demonstrated surprising generalization [PITH_FULL_IMAGE:figures/full_fig_p004_3.png] view at source ↗
Figures from the paper (5 more)
Figure 3
Figure 3. Figure 3: Stiffness test results TABLE II DAMAGE TEST RESULTS ON THE HEXAPODAL ROBOT Damage joints FLC(calf) FLT(thigh) FRC MRC MLT HLT HLC HRT HRC Method Distance /m↑ (distance that the robot walked in 20s given a velocity command at 0.75m/s) RMA 1.83±0.77 2.56±2.11 5.52±2.95 1…
Figure 4
Figure 4. Figure 4: Real World Experiments also note that ROAM fails to walk in all three test failures. This is mainly because each behavior module of ROAM is trained independently within its respective scenario. So the generalization of each module is limited. While ROAM relies on choos…
Figure 5
Figure 5. Figure 5: Unstructured gaits shown by ROLT when the hexapodal robot faces unseen motor damages. [PITH_FULL_IMAGE:figures/full_fig_p006_5.png]
Figure 7
Figure 7. Figure 7: Low-frequency noise test results (a) (b) (c) (d) (e) (f) [PITH_FULL_IMAGE:figures/full_fig_p006_7.png]
Figure 8
Figure 8. Figure 8: Elevation maps and attention maps in different situations. [PITH_FULL_IMAGE:figures/full_fig_p006_8.png]

Discussion (0). Sign in to comment.

Reference graph

Works this paper leans on

27 extracted references · 12 canonical work pages

  1. [1]

    Learning agile and dynamic motor skills for legged robots,

    J. Hwangbo, J. Lee, A. Dosovitskiy, D. Bellicoso, V . Tsounis, V . Koltun, and M. Hutter, “Learning agile and dynamic motor skills for legged robots,” Science Robotics, vol. 4, no. 26, p. eaau5872, 2019

  2. [2]

    Learning robust perceptive locomotion for quadrupedal robots in the wild,

    T. Miki, J. Lee, J. Hwangbo, L. Wellhausen, V . Koltun, and M. Hutter, “Learning robust perceptive locomotion for quadrupedal robots in the wild,” Science Robotics , vol. 7, no. 62, p. eabk2822, 2022

  3. [3]

    Rma: Rapid motor adaptation for legged robots,

    A. Kumar, Z. Fu, D. Pathak, and J. Malik, “Rma: Rapid motor adaptation for legged robots,” arXiv preprint arXiv:2107.04034 , 2021

  4. [4]

    Legged locomotion in challenging terrains using egocentric vision,

    A. Agarwal, A. Kumar, J. Malik, and D. Pathak, “Legged locomotion in challenging terrains using egocentric vision,” in Conference on Robot Learning. PMLR, 2023, pp. 403–415

  5. [5]

    Learn- ing quadrupedal locomotion over challenging terrain,

    J. Lee, J. Hwangbo, L. Wellhausen, V . Koltun, and M. Hutter, “Learn- ing quadrupedal locomotion over challenging terrain,”Science robotics, vol. 5, no. 47, p. eabc5986, 2020

  6. [6]

    Dreamwaq: Learning robust quadrupedal locomotion with implicit terrain imagination via deep reinforcement learning,

    I. M. A. Nahrendra, B. Yu, and H. Myung, “Dreamwaq: Learning robust quadrupedal locomotion with implicit terrain imagination via deep reinforcement learning,” in 2023 IEEE International Conference on Robotics and Automation (ICRA) . IEEE, 2023, pp. 5078–5084

  7. [7]

    Hybrid internal model: A simple and efficient learner for agile legged locomotion,

    J. Long, Z. Wang, Q. Li, J. Gao, L. Cao, and J. Pang, “Hybrid internal model: A simple and efficient learner for agile legged locomotion,” arXiv preprint arXiv:2312.11460 , 2023

  8. [8]

    You only live once: Single-life reinforcement learning,

    A. Chen, A. Sharma, S. Levine, and C. Finn, “You only live once: Single-life reinforcement learning,” Advances in Neural Information Processing Systems, vol. 35, pp. 14 784–14 797, 2022

Show all 27 references
  1. [9]

    Legged robots that keep on learning: Fine-tuning locomotion policies in the real world,

    L. Smith, J. C. Kew, X. B. Peng, S. Ha, J. Tan, and S. Levine, “Legged robots that keep on learning: Fine-tuning locomotion policies in the real world,” in 2022 International Conference on Robotics and Automation (ICRA). IEEE, 2022, pp. 1593–1599

  2. [10]

    Adapt on-the-go: Behavior modulation for single-life robot deployment,

    A. S. Chen, G. Chada, L. Smith, A. Sharma, Z. Fu, S. Levine, and C. Finn, “Adapt on-the-go: Behavior modulation for single-life robot deployment,” arXiv preprint arXiv:2311.01059 , 2023

  3. [11]

    Attention is all you need,

    A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, Ł. Kaiser, and I. Polosukhin, “Attention is all you need,” Advances in neural information processing systems , vol. 30, 2017

  4. [12]

    Sim- to-real transfer for quadrupedal locomotion via terrain transformer,

    H. Lai, W. Zhang, X. He, C. Yu, Z. Tian, Y . Yu, and J. Wang, “Sim- to-real transfer for quadrupedal locomotion via terrain transformer,” in 2023 IEEE International Conference on Robotics and Automation (ICRA). IEEE, 2023, pp. 5141–5147

  5. [13]

    Barkour: Bench- marking animal-level agility with quadruped robots,

    K. Caluwaerts, A. Iscen, J. C. Kew, W. Yu, T. Zhang, D. Freeman, K.-H. Lee, L. Lee, S. Saliceti, V . Zhuang et al. , “Barkour: Bench- marking animal-level agility with quadruped robots,” arXiv preprint arXiv:2305.14654, 2023

  6. [14]

    Learning humanoid locomotion with transformers,

    I. Radosavovic, T. Xiao, B. Zhang, T. Darrell, J. Malik, and K. Sreenath, “Learning humanoid locomotion with transformers,” arXiv preprint arXiv:2303.03381 , 2023

  7. [15]

    Learning vision-guided quadrupedal locomotion end-to-end with cross-modal transformers,

    R. Yang, M. Zhang, N. Hansen, H. Xu, and X. Wang, “Learning vision-guided quadrupedal locomotion end-to-end with cross-modal transformers,” arXiv preprint arXiv:2107.03996 , 2021

  8. [16]

    Intriguing properties of vision transformers,

    M. M. Naseer, K. Ranasinghe, S. H. Khan, M. Hayat, F. Shahbaz Khan, and M.-H. Yang, “Intriguing properties of vision transformers,” Ad- vances in Neural Information Processing Systems , vol. 34, pp. 23 296– 23 308, 2021

  9. [17]

    An image is worth 16x16 words: Transformers for image recognition at scale,

    A. Dosovitskiy, L. Beyer, A. Kolesnikov, D. Weissenborn, X. Zhai, T. Unterthiner, M. Dehghani, M. Minderer, G. Heigold, S. Gelly et al., “An image is worth 16x16 words: Transformers for image recognition at scale,” arXiv preprint arXiv:2010.11929 , 2020

  10. [18]

    Making transformers solve compositional tasks,

    S. Ontanón, J. Ainslie, V . Cvicek, and Z. Fisher, “Making transformers solve compositional tasks,” arXiv preprint arXiv:2108.04378 , 2021

  11. [19]

    Metamorph: Learning universal controllers with transformers,

    A. Gupta, L. Fan, S. Ganguli, and L. Fei-Fei, “Metamorph: Learning universal controllers with transformers,” arXiv preprint arXiv:2203.11931, 2022

  12. [20]

    Dropout: a simple way to prevent neural networks from overfitting,

    N. Srivastava, G. Hinton, A. Krizhevsky, I. Sutskever, and R. Salakhut- dinov, “Dropout: a simple way to prevent neural networks from overfitting,” The journal of machine learning research , vol. 15, no. 1, pp. 1929–1958, 2014

  13. [21]

    Unidrop: A simple yet effective technique to improve transformer without extra cost,

    Z. Wu, L. Wu, Q. Meng, Y . Xia, S. Xie, T. Qin, X. Dai, and T.-Y . Liu, “Unidrop: A simple yet effective technique to improve transformer without extra cost,” arXiv preprint arXiv:2104.04946 , 2021

  14. [22]

    Informative dropout for robust representation learning: A shape-bias perspective,

    B. Shi, D. Zhang, Q. Dai, Z. Zhu, Y . Mu, and J. Wang, “Informative dropout for robust representation learning: A shape-bias perspective,” in International Conference on Machine Learning . PMLR, 2020, pp. 8828–8839

  15. [23]

    Consistent dropout for policy gradi- ent reinforcement learning,

    M. Hausknecht and N. Wagener, “Consistent dropout for policy gradi- ent reinforcement learning,” arXiv preprint arXiv:2202.11818 , 2022

  16. [24]

    Roll-drop: accounting for observation noise with a single parameter,

    L. Campanaro, D. De Martini, S. Gangapurwala, W. Merkt, and I. Havoutis, “Roll-drop: accounting for observation noise with a single parameter,” in Learning for Dynamics and Control Conference . PMLR, 2023, pp. 718–730

  17. [25]

    Learning modular robot control policies,

    J. Whitman, M. Travers, and H. Choset, “Learning modular robot control policies,” IEEE Transactions on Robotics , 2023

  18. [26]

    Proximal policy optimization algorithms,

    J. Schulman, F. Wolski, P. Dhariwal, A. Radford, and O. Klimov, “Proximal policy optimization algorithms,” arXiv preprint arXiv:1707.06347, 2017

  19. [27]

    Learning to walk in minutes using massively parallel deep reinforcement learning,

    N. Rudin, D. Hoeller, P. Reist, and M. Hutter, “Learning to walk in minutes using massively parallel deep reinforcement learning,” in Conference on Robot Learning . PMLR, 2022, pp. 91–100

Pith tools

Reviewed August 6, 2026 · model on record in the stance chip above.