Stoch Lab Logo

BiRoDiff: Diffusion policies for bipedal robot locomotion on unseen terrains (ICC 2024)

Why limit yourself to just generating images when you could also generate trajectories?
Robert Bosch Centre for Cyber Physical Systems,
Indian Institute of Science, Bangalore
exploded
cad_rendered

CAD Assembly of Stoch BiRo

Note

All experiments were done on Our Custom made Bipedal Robot (Stoch Biro). Refer to my old work, published at ICCAR 2024

Abstract

Locomotion on unknown terrains is essential for bipedal robots to handle novel real-world challenges, thus expanding their utility in disaster response and exploration. In this work, we introduce a lightweight framework that learns a single walking controller that yields locomotion on multiple terrains. We have designed a real-time robot controller based on diffusion models, which not only captures multiple behaviours with different velocities in a single policy but also generalizes well for unseen terrains. Our controller learns with offline data, which is better than online learning in aspects like scalability, simplicity in training scheme etc. We have designed and implemented a diffusion model-based policy controller in simulation on our custom-made Bipedal Robot model named Stoch BiRo. We have demonstrated its generalization capability and high frequency control step generation relative to typical generative models, which require huge onboarding compute.

Architecture of Diffusion Policy

teaser
teaser

Algorithms

teaser
teaser

Simulations & Results


Template is inspired by this Repository.