内容提要
SuperMimic提出一种人形机器人全身控制新方法:学习单一可微分世界模型刻画共享动力学,通过短时想象回传解析策略梯度,替代高成本无模型强化学习。单GPU训练3小时即超过SONIC已发布检查点,计算量减少超1000倍;在129,514个动作基准上仅25个失败,而SONIC失败179个。策略可迁移至真实Unitree G1机器人。
延伸解读
计算效率的突破点:共享动力学模型
文章指出,SONIC等现有方法训练需约21000个GPU小时,而SuperMimic仅用单GPU 3小时即超过其已发布检查点。关键在于它学习一个可微分世界模型来刻画所有人形动作共享的刚体动力学,从而用解析梯度替代大量无模型rollout。这提示读者:当任务共享底层动力学时,将学习成本摊销到单一模型上,可能比单纯堆算力更有效。
动作覆盖规模与失败率的对比
在包含129514个动作的基准上,SuperMimic单GPU训练3小时有25个动作失败,两块GPU训练9.8小时后全部成功;而SONIC已发布检查点有179个动作失败。这一对比说明,SuperMimic在扩大动作库时不仅计算量增长缓慢,失败率也显著更低。读者可关注其世界模型预测误差在数据量增长264倍时仍保持稳定的描述,这支撑了其可扩展性。
与可微分仿真及世界模型方法的区别
文章将SuperMimic与SHAC、RWM/RWM-U等方法区分:它不直接对模拟器求导,而是从模拟器状态转移中学习确定性世界模型,再通过短时想象rollout的BPTT获得解析策略梯度。因此它仍兼容具有非光滑接触和非可微实现的刚体模拟器。同时,世界模型在训练过程中用自身策略收集的数据拟合,而非依赖其他策略的轨迹预训练。
真机迁移与下游接口的初步验证
文章提到学习到的策略可迁移到真实Unitree G1机器人,并在下游接口初步研究中发现,用强化学习训练的速度编码器优于蒸馏基线。这表明SuperMimic不仅关注仿真中的跟踪性能,也尝试将共享动力学模型复用到速度等目标接口。不过,该部分被描述为初步研究,读者应注意其结论范围有限,尚不能推断所有下游任务均同等受益。
Q&A
SuperMimic 是什么?它主要解决人形机器人全身控制中的什么问题?
SuperMimic 是一种新的人形机器人全身控制框架。它主要解决传统无模型强化学习在扩展动作库时计算成本急剧增加的问题,通过一个可微分的世界模型来学习共享的人形动力学,从而高效训练全身控制器。
SuperMimic 相比 SONIC 在训练效率和成功率上有哪些具体优势?
SuperMimic 在单块 GPU 上仅用 3 小时就超过了 SONIC 已发布 checkpoint 的归一化回报,并将 GPU 计算小时数减少了超过 1000 倍。在包含 129,514 个动作的基准测试中,SuperMimic 单 GPU 运行 3 小时仅有 25 个动作失败,而 SONIC 已发布 checkpoint 有 179 个动作失败。
SuperMimic 的训练循环包含哪几个阶段?它们是如何交替进行的?
SuperMimic 的训练循环交替进行三个阶段:首先,当前策略从仿真器中收集状态转移数据并存入重放缓冲区;其次,将可微分世界模型拟合到来自重放缓冲区的短时间窗口数据上;第三,在冻结的世界模型中,通过反向传播虚拟回报来改进策略,每次回放都以重放缓冲区中的某个状态为初始状态。整个过程中缓冲区、模型和策略持续更新,动力学学习与控制能力同步提升。
SuperMimic 如何获得策略梯度?为什么它仍然兼容刚体模拟器?
SuperMimic 通过对学习到的可微分世界模型执行短时想象 rollout 并进行时间反向传播(BPTT),获得解析形式的策略梯度。由于这些梯度来源于学习得到的模型而非对模拟器本身求导,因此它依然兼容刚体模拟器,包括具有非光滑接触和非可微实现的模拟器。
SuperMimic 学习的世界模型有什么特点?它是如何构建的?
SuperMimic 学习一个确定性的可微分代理世界模型,它以每个回合的仿真上下文(摩擦系数、标定偏移以及躯干质心)为条件,并预测无噪声观测。该模型由已学习的动力学模型以及奖励、观测和终止头部组成,每个头部要么通过学习获得,要么通过解析方式计算。
SuperMimic 训练出的策略能否迁移到真实机器人?是否有实际部署验证?
是的,SuperMimic 学习得到的策略能够迁移到真实的 Unitree G1 机器人上。文章提到在对下游接口的初步研究中,用强化学习训练得到的速度编码器优于蒸馏基线方法,并展示了最终控制器的物理部署效果。