Unified Locomotion Transformer with Simultaneous Sim-to-Real Transfer for Quadrupeds

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liu, Dikai, Zhang, Tianwei, Yin, Jianxiong, See, Simon
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911087697330176
author Liu, Dikai
Zhang, Tianwei
Yin, Jianxiong
See, Simon
author_facet Liu, Dikai
Zhang, Tianwei
Yin, Jianxiong
See, Simon
contents Quadrupeds have gained rapid advancement in their capability of traversing across complex terrains. The adoption of deep Reinforcement Learning (RL), transformers and various knowledge transfer techniques can greatly reduce the sim-to-real gap. However, the classical teacher-student framework commonly used in existing locomotion policies requires a pre-trained teacher and leverages the privilege information to guide the student policy. With the implementation of large-scale models in robotics controllers, especially transformers-based ones, this knowledge distillation technique starts to show its weakness in efficiency, due to the requirement of multiple supervised stages. In this paper, we propose Unified Locomotion Transformer (ULT), a new transformer-based framework to unify the processes of knowledge transfer and policy optimization in a single network while still taking advantage of privilege information. The policies are optimized with reinforcement learning, next state-action prediction, and action imitation, all in just one training stage, to achieve zero-shot deployment. Evaluation results demonstrate that with ULT, optimal teacher and student policies can be obtained at the same time, greatly easing the difficulty in knowledge transfer, even with complex transformer-based models.
format Preprint
id arxiv_https___arxiv_org_abs_2503_08997
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Unified Locomotion Transformer with Simultaneous Sim-to-Real Transfer for Quadrupeds
Liu, Dikai
Zhang, Tianwei
Yin, Jianxiong
See, Simon
Robotics
Machine Learning
Quadrupeds have gained rapid advancement in their capability of traversing across complex terrains. The adoption of deep Reinforcement Learning (RL), transformers and various knowledge transfer techniques can greatly reduce the sim-to-real gap. However, the classical teacher-student framework commonly used in existing locomotion policies requires a pre-trained teacher and leverages the privilege information to guide the student policy. With the implementation of large-scale models in robotics controllers, especially transformers-based ones, this knowledge distillation technique starts to show its weakness in efficiency, due to the requirement of multiple supervised stages. In this paper, we propose Unified Locomotion Transformer (ULT), a new transformer-based framework to unify the processes of knowledge transfer and policy optimization in a single network while still taking advantage of privilege information. The policies are optimized with reinforcement learning, next state-action prediction, and action imitation, all in just one training stage, to achieve zero-shot deployment. Evaluation results demonstrate that with ULT, optimal teacher and student policies can be obtained at the same time, greatly easing the difficulty in knowledge transfer, even with complex transformer-based models.
title Unified Locomotion Transformer with Simultaneous Sim-to-Real Transfer for Quadrupeds
topic Robotics
Machine Learning
url https://arxiv.org/abs/2503.08997