该项目在Isaac Lab中复现了DreamWaQ的非对称Actor-Critic和CENet估计器,使仅依赖本体感觉的策略在Unitree Go2上达到与使用特权信息基线相近的性能。

Stars

5

7 天增长

暂无数据

Fork 数

0

开放 Issue

0

开源协议

MIT

最近更新

2026-07-28

AI 仓库情报摘要
FR-AI / ANALYSIS

为什么值得关注

它展示了一个简洁的单阶段训练流程,通过从本体感觉历史中学习潜在上下文,替代了特权地形和速度输入,在真实硬件约束下实现了接近特权信息的运动能力。

适合谁使用

  • 研究Sim-to-Real迁移的机器人学者
  • 从事四足机器人运动控制的强化学习实践者
  • 在腿式机器人上部署学习控制器的工程师
  • 希望学习模块化实现ICRA 2023已知方法的学生

典型使用场景

  • 无需外部传感器即可在Unitree Go2上部署鲁棒运动策略
  • 为腿式机器人评估本体感觉速度估计性能
  • 在不同地形上基于课程学习训练四足步态
  • 在NVIDIA Isaac Lab中复现并扩展DreamWaQ结果

项目优势

  • 仅使用64维策略输入(对比235维特权输入)即达到相近运动性能
  • 提供了严格的验证:单元测试、梯度检查以及逐组件通过/失败检查
  • 包含详细的训练曲线、速度估计图以及评估滚动回放
  • 代码作为独立任务包,无需修改Isaac Lab框架文件即可使用

使用前须知

  • 仅使用单一随机种子训练,结果缺乏统计显著性
  • 仅使用2048个并行环境(原论文为4096个),受限于8 GB显存
  • 未实现原论文中的AdaBoost自适应引导门
  • β超参数固定为0.5且未进行扫描,因为β=1时导致后验坍塌

README 快速开始

DreamWaQ in Isaac Lab - Unitree Go2

A reproduction of the two core contributions of DreamWaQ (Nahrendra, Yu & Myung, ICRA 2023) in NVIDIA Isaac Lab on a Unitree Go2: an asymmetric actor-critic and CENet, a context-aided estimator that lets the deployed policy run on proprioception alone.

Result: the proprioception-only policy matched a privileged baseline on terrain difficulty and episode length, trailing it by ~3 reward and ~0.1 m/s tracking error. CENet recovered true body velocity to MSE ≈ 0.006 (m/s)² from proprioceptive history alone.

https://github.com/user-attachments/assets/daabeb17-dc1e-4bb3-8c0d-82a0601c9b6a

Trained Go2 robots crossing discrete-obstacle terrain in Isaac Sim, each tracking its own velocity command.


The problem

In simulation an RL policy can see privileged information - the exact terrain under the robot, its true base velocity. A real quadruped has neither. It has joint encoders and an IMU. So a policy trained on privileged inputs cannot be deployed.

DreamWaQ's answer is to have the robot infer that hidden state from its own recent proprioceptive history, and to train the policy, the value function, and the estimator together in a single phase - no teacher-student distillation.

Method

Asymmetric actor-critic. The actor sees only what hardware can measure (45-D proprioception) plus CENet's outputs. The critic, used only during training, additionally sees a 187-D terrain height scan and the true base velocity.

GroupDimContentsSeen by
policy45ang. velocity, gravity, command, joint pos/vel, last actionactor + critic
height_scan18717×11 terrain height grid (privileged)critic only
privileged_vel3true base linear velocitycritic + CENet target

Actor input 64 (45 + 3 velocity + 16 latent) · Critic input 235 (45 + 187 + 3)

CENet. A shared encoder over the last 5 proprioceptive frames (225-D) splits into two heads: a velocity head regressing 3-D body velocity, and a β-VAE head producing a 16-D latent context. Trained with a hybrid loss:

L = MSE(v̂, v) + MSE(ô, o) + β · KL( z ‖ N(0, I) )

CENet's outputs are detached before reaching the actor, so the policy's gradient cannot reshape the estimator - it learns only from its own prediction error.

Results

Both policies: 2000 iterations, se

项目描述

Reproduction of DreamWaQ (ICRA 2023) in NVIDIA Isaac Lab - asymmetric actor-critic + CENet for proprioception-only quadrupedal locomotion on a Unitree Go2

相关仓库与替代方案

根据分类、Topic 和编程语言匹配的相似项目。

lopopolo
精选
lopopolo GitHub avatar

harness-engineering

Harness Engineering is a methodology for improving coding agent outputs by carefully crafting the environment around them—providing curated context, tools, and executable constraints that encode an organization’s nonfunctional requirements and cumulative lessons.

AI 与机器学习AI 智能体
2,390
slvDev
精选
slvDev GitHub avatar

esp32-ai

A 28.9 million parameter language model runs on an $8 ESP32-S3 microcontroller entirely on-device, generating simple stories at about 9.5 tokens per second.

AI 与机器学习大语言模型
1,960
littledivy
精选
littledivy GitHub avatar

mimic

mimic captures traffic from any iOS or web app and automatically generates a Python client library that lets you call the app's API like a regular library.

AI 与机器学习
1,482