Symmetry-Guided Memory Augmentation for Efficient Locomotion Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bao, Kaixi, Li, Chenhao, As, Yarden, Krause, Andreas, Hutter, Marco
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911542035873792
author Bao, Kaixi
Li, Chenhao
As, Yarden
Krause, Andreas
Hutter, Marco
author_facet Bao, Kaixi
Li, Chenhao
As, Yarden
Krause, Andreas
Hutter, Marco
contents Training reinforcement learning (RL) policies for legged locomotion often requires extensive environment interactions, which are costly and time-consuming. We propose Symmetry-Guided Memory Augmentation (SGMA), a framework that improves training efficiency by combining structured experience augmentation with memory-based context inference. Our method leverages robot and task symmetries to generate additional, physically consistent training experiences without requiring extra interactions. To avoid the pitfalls of naive augmentation, we extend these transformations to the policy's memory states, enabling the agent to retain task-relevant context and adapt its behavior accordingly. We evaluate the approach on quadruped and humanoid robots in simulation, as well as on a real quadruped platform. Across diverse locomotion tasks involving joint failures and payload variations, our method achieves efficient policy training while maintaining robust performance, demonstrating a practical route toward data-efficient RL for legged robots.
format Preprint
id arxiv_https___arxiv_org_abs_2502_01521
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Symmetry-Guided Memory Augmentation for Efficient Locomotion Learning
Bao, Kaixi
Li, Chenhao
As, Yarden
Krause, Andreas
Hutter, Marco
Machine Learning
Artificial Intelligence
Robotics
Training reinforcement learning (RL) policies for legged locomotion often requires extensive environment interactions, which are costly and time-consuming. We propose Symmetry-Guided Memory Augmentation (SGMA), a framework that improves training efficiency by combining structured experience augmentation with memory-based context inference. Our method leverages robot and task symmetries to generate additional, physically consistent training experiences without requiring extra interactions. To avoid the pitfalls of naive augmentation, we extend these transformations to the policy's memory states, enabling the agent to retain task-relevant context and adapt its behavior accordingly. We evaluate the approach on quadruped and humanoid robots in simulation, as well as on a real quadruped platform. Across diverse locomotion tasks involving joint failures and payload variations, our method achieves efficient policy training while maintaining robust performance, demonstrating a practical route toward data-efficient RL for legged robots.
title Symmetry-Guided Memory Augmentation for Efficient Locomotion Learning
topic Machine Learning
Artificial Intelligence
Robotics
url https://arxiv.org/abs/2502.01521