跳到正文
arXiv:cs.LG· Huixuan Cheng, Bike Zhang, Shankar Sastry, Koushil Sreenath·· 3 小时前

用扩散模型生成并理解人形机器人策略的神经网络权重

Modeling and Understanding Neural Weights of Humanoid Policies with Diffusion

AI 导读

研究采用线性自编码器加 Diffusion Transformer 的潜在扩散框架,在 Unitree G1 机器人强化学习策略网络上训练,直接生成权重即可部署的网络,能完成行走、跳舞和回旋踢等动作。研究还发现第一层权重存在一致反映机器人运动学先验的结构性规律,该先验在强化学习与模仿学习下均会出现。

正文

View PDF HTML (experimental)

Abstract:Recent advances in robot learning have driven rapid progress in humanoid control, enabling increasingly agile behaviors in the real world. However, humanoid control networks are typically obtained through conventional training procedures, while comparatively little attention has been given to directly modeling the distribution of their learned weights and generating functional networks in weight space. In this work, we investigate whether high-performing humanoid control networks can be generatively modeled directly through their weights. We adopt a latent diffusion framework that couples a linear autoencoder with a Diffusion Transformer. The autoencoder compresses actor network parameters into a compact latent space, and the Diffusion Transformer models the distribution of these latents. Trained on networks collected from reinforcement learning for the Unitree G1 robot, our approach generates deployable networks that perform various motions in the real world, including walking, dancing, and spin kick. Beyond generation, we further study the learned weight space to better understand the structure captured by these humanoid control networks. We identify a structural principle in the first-layer weights that consistently reflects the robot's kinematic prior. Notably, this kinematic prior emerges under both Reinforcement Learning (RL) and Imitation Learning (IL) training, suggesting that it is not specific to a particular optimization process but may instead reflect common knowledge associated with humanoid embodiment. Together, these results demonstrate the feasibility of generative modeling in humanoid network weight space, while providing initial insights into the information encoded by their learned weights. Additional results are available on our project website: this https URL
Comments: Project page: this https URL
Subjects: Robotics (cs.RO); Artificial Intelligence (cs.AI); Machine Learning (cs.LG)
Cite as: arXiv:2610.20606 [cs.RO]
  (or arXiv:2610.20606v1 [cs.RO] for this version)
  https://doi.org/10.48550/arXiv.2610.20606

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Huixuan Cheng [view email]
[v1] Wed, 14 Oct 2026 17:59:52 UTC (8,422 KB)

来源:arXiv:cs.LG · arxiv.org