【与身俱能】机械臂训练(学习笔记)

一共两个项目,均有开源代码,各有各的好处

项目一:SimpleVLA-RL,1条轨迹教会机器;

项目二:SmolVLA,RTX 3080级别显卡的训练

项目一:SimpleVLA-RL

由清华大学、上海人工智能实验室、上海交通大学、北京大学和香港大学研究团队共同完成的突破性研究发表于2025年1月,论文题为"SimpleVLA-RL: Scaling VLA Training via Reinforcement Learning"。

源码地址:https://github.com/PRIME-RL/SimpleVLA-RL

论文连接:https://arxiv.org/pdf/2509.09674

视频介绍地址:https://www.bilibili.com/video/BV1DsJdzGEUG/?vd_source=1b067419e0d22b3a5c182de38a58b079

强化学习训练需要一个SFT(监督微调) VLA模型(大约15G)

可用模型包括:

  • libero-10 traj1/trajall SFT
  • libero-goal/object/spatial traj1 SFT
  • Robotwin2.0 tasks traj1000 SFT

模型库下载地址:https://huggingface.co/collections/Haozhan72/simplevla-rl

这篇文章对这个项目介绍的比较详细

清华重磅!SimpleVLA-RL炸场:用RL让VLA模型少样本也能强,泛化能力飙升!入门到精通全解析,收藏这篇就够了!-CSDN博客


项目二:SmolVLA

是由Hugging Face团队于2025年6月推出的开源机器人AI模型,属于视觉-语言-行动(VLA)架构,旨在降低机器人智能体的研发门槛并推动通用化进程

项目地址

https://ai.gitcode.com/hf_mirrors/merve/smol-vision

SmolVLA的主要功能

  • 多模态输入处理:SmolVLA 能处理多种输入,包括多幅图像、语言指令以及机器人的状态信息。通过视觉编码器提取图像特征,将语言指令标记化后输入解码器,将传感运动状态通过线性层投影到一个标记上,与语言模型的标记维度对齐。

  • 动作序列生成:模型包含一个动作专家模块,是一个轻量级的 Transformer,能基于视觉-语言模型(VLM)的输出,生成未来机器人的动作序列块。采用流匹配技术进行训练,通过引导噪声样本回归真实数据分布来学习动作生成,实现高精度的实时控制。

  • 高效推理与异步执行:SmolVLA 引入了异步推理堆栈,将动作执行与感知和预测分离,实现更快、更灵敏的控制,使机器人可以在快速变化的环境中更快速地响应,提高了响应速度和任务吞吐量。

SmolVLA的技术原理

  • 视觉-语言模型(VLM):SmolVLA 使用 SmolVLM2 作为其 VLM 主干,模型经过优化,能处理多图像输入。包含一个 SigLIP 视觉编码器和一个 SmolLM2 语言解码器。图像标记通过视觉编码器提取,语言指令被标记化后直接输入解码器,传感运动状态则通过线性层投影到一个标记上,与语言模型的标记维度对齐。解码器层处理连接的图像、语言和状态标记,得到的特征随后传递给动作专家。

  • 动作专家:动作专家是一个轻量级的 Transformer(约1亿参数),基于 VLM 的输出,生成未来机器人的动作序列块。动作专家采用流匹配技术进行训练,通过引导噪声样本回归真实数据分布来学习动作生成,实现高精度的实时控制。

  • 视觉 Token 减少:为了提高效率,SmolVLA 限制每帧图像的视觉 Token 数量为64个,大大减少了处理成本。

  • 层跳跃加速推理:SmolVLA 跳过 VLM 中的一半层进行计算,有效地将计算成本减半,同时保持了良好的性能。

  • 交错注意力层:与传统的 VLA 架构不同,SmolVLA 交替使用交叉注意力(CA)和自注意力(SA)层。提高了多模态信息整合的效率,加快推理速度。

  • 异步推理:SmolVLA 引入了异步推理策略,让机器人的"手"和"眼"能独立工作。在这种策略下,机器人可以一边执行当前动作,一边已经开始处理新的观察并预测下一组动作,消除推理延迟,提高控制频率。

由于smolVLA对显卡要求不高,所以用自己的电脑就可以训练,网上很多项目的搭建教程

这两个地址介绍的不错

轻量化视觉语言动作模型SmolVLA微调实战指南:从环境搭建到部署落地-CSDN博客

smolVLA初体验_smolvla部署-CSDN博客

相关推荐
十月的皮皮1 小时前
C语言学习笔记20260706-栈的压入、弹出序列验证
c语言·笔记·学习
渣渣灰飞2 小时前
MySQL 系统学习 第二阶段 第三章 DQL(Data Query Language)第二节:WHERE 条件查询
sql·学习·mysql
维他奶糖612 小时前
基于 DrissionPage 实现小红书搜索笔记批量采集
笔记
心中有国也有家3 小时前
AtomGit Flutter 鸿蒙客户端:E-Brufen 架构设计
学习·flutter·华为·harmonyos
hssfscv3 小时前
QT的学习笔记4——QMainWindow、资源文件以及主要控件的介绍
笔记·qt·学习
克里斯蒂亚诺更新3 小时前
电脑关闭密码的操作
笔记
小小的代码里面挖呀挖呀挖4 小时前
杰理蓝牙耳机开发--LE Audio与游戏耳机应用
笔记·单片机·物联网·学习
xiaomici4 小时前
SAC中的Dataset/Model/Planning model
笔记
一楼的猫5 小时前
AI写作检测机制技术深度解析:200+维度检测、叙事指纹与网文AI辅助怎么过审
人工智能·学习·安全·chatgpt·ai写作
学数学的口心贝5 小时前
从零搭建并维护个人主页
经验分享·笔记