1. 项目概述:VLA与Psi0的革新意义
在通用人形机器人研究领域,VLA(Vision-Language-Action)模型正掀起一场认知革命。Psi0作为这一技术路线的代表性成果,首次实现了从人类视频数据到机器人全身协调控制的端到端学习。这个开源基础模型包含三个突破性创新:基于MM-DiT架构的多模态动作专家系统、实时分块控制技术(RTC)、以及专为移动操作设计的遥操作框架。
传统机器人控制面临的核心矛盾在于:精细操作需要高维度动作空间(如48-DoF的手部控制),而长时程任务又要求稳定的全身协调。Psi0通过分层训练策略巧妙解决了这一难题——先在829小时的人类自我中心视频(EgoDex数据集)上预训练视觉语言理解能力,再在31小时的真实机器人数据(Humanoid Everyday数据集)上微调动作生成模块。这种"先理解后执行"的两阶段方法,使得模型仅需少量领域数据就能适应新任务。
2. 核心技术解析
2.1 MM-DiT架构设计
Psi0的核心创新在于改进的MM-DiT(Multi-Modal Diffusion Transformer)架构。与原始DiT相比,其关键差异体现在:
- 双流调制机制:时间步长τ分别调制视觉语言(VL)分支和动作(A)分支的特征
- 联合注意力层:VL token与A token在每个Transformer块中进行全局注意力交互
- 残差连接设计:保留原始视觉特征的同时融合语言指令信息
这种结构在8个真实世界任务测试中,比标准DiT架构的平均成功率提升23%。特别是在需要双臂协调的"搬运箱子"任务中,成功率达到90%,而基线模型最高仅60%。
2.2 实时动作分块技术
面对2.5B参数模型160ms的推理延迟,Psi0引入了训练时实时分块(RTC)方案:
# 伪代码示例:RTC训练逻辑 def train_step(obs, action_chunk): max_delay = random.randint(0, H-s) # 模拟推理延迟 noisy_actions = add_noise(action_chunk[:max_delay]) masked_loss = compute_loss(noisy_actions) # 仅计算未延迟部分的损失 return masked_loss该技术通过随机掩码部分动作token,强制模型学习生成与历史动作连贯的新动作块。实测表明,这使30Hz控制周期下的动作抖动降低72%,碰撞率下降58%。
3. 硬件实现细节
3.1 机器人平台配置
Psi0在Unitree G1人形机器人上验证,关键硬件参数:
| 组件 | 规格 |
|---|---|
| 自由度 | 29 (全身) |
| 手部 | Dex3-1灵巧手 (7-DoF/手) |
| 视觉 | Intel RealSense D435i (640x480@30Hz) |
| 计算单元 | NVIDIA Jetson AGX Orin (32GB) |
3.2 遥操作套件设计
为实现高质量示范数据采集,开发了多模态遥操作系统:
- 手部追踪:Manus手套直接控制所有手指关节
- 上身姿态:PICO VR头显+腕部追踪器驱动逆运动学解算
- 移动控制:腰部追踪器生成速度指令,足部追踪器提供转向信号
这种设计相比传统手柄控制,在折叠衣物等精细任务中采集的数据成功率提升41%。
4. 训练流程剖析
4.1 预训练阶段
使用FAST动作分词器处理EgoDex数据集:
- 将48维连续动作离散为20个token
- 重建误差L1=0.005 (比开源基线提升5倍)
- 在64张A100上训练10天,batch size=1024
关键发现:仅预测单步动作(而非多步序列)就足以学习有效的视觉表征,这使训练效率提升3倍。
4.2 后训练阶段
冻结视觉语言主干,专注优化动作专家:
- 输入分辨率降为320×240
- 扩散时间步长τ∈[0,1]均匀采样
- 32张A100训练30小时,batch size=2048
此时模型已能完成基础抓取、推动等操作,但长时程任务成功率仅65%。
4.3 微调阶段
每个任务仅需80条遥操作轨迹:
- 学习率1e-4 (cosine衰减)
- 40k训练步数
- 引入RTC后成功率提升至89%
特别在"开水龙头"任务中,手指关节的精确控制误差<0.8mm。
5. 实战问题排查
5.1 常见故障模式
- 动作抖动:检查RTC的max_delay参数是否超过H-s
- 抓取失败:调整FAST tokenizer的词汇量(建议2048)
- 移动不稳:确保腰部追踪器采样率≥100Hz
5.2 性能优化技巧
- 视觉预处理:将输入图像从1920x1080降采样到360x240,推理速度提升4倍
- 内存优化:使用DeepSpeed的Zero-3阶段减少显存占用
- 实时性保障:设置动作块长度s=16,执行窗口H=32
6. 领域应用展望
Psi0的技术路线为通用人形机器人指明了发展方向:
- 医疗辅助:通过手术视频学习精细操作
- 家庭服务:利用日常视频理解家务语义
- 工业检测:结合AR眼镜实现远程维修
当前限制主要在硬件负载能力(G1仅可承载5kg),但随着本体技术进步,这套算法框架将释放更大潜力。建议后续研究关注多任务联合微调,以及从人类视频到机器人控制的域适应理论。