首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >NVIDIA推出RoboTTT训练框架,实现机器人超长时序感知新突破

NVIDIA推出RoboTTT训练框架,实现机器人超长时序感知新突破

作者头像
GPUS Lady
发布2026-07-16 21:28:17
发布2026-07-16 21:28:17
1390
举报
文章被收录于专栏:GPUS开发者GPUS开发者

一、行业痛点:传统机器人短时记忆瓶颈

作为 NVIDIA 联合斯坦福大学、德州大学奥斯汀分校的机器人算法研发团队,在通用机器人基础模型研发中发现,现有主流视觉运动策略普遍存在致命短板:绝大多数机器人模型仅支持单步或极短时序上下文,有效记忆窗口不足 0.1 秒,只能处理零散帧数据,执行多阶段、长时序装配任务时会瞬间遗忘此前动作与环境交互信息,难以完成连贯、复杂的连续操作。

传统循环神经网络、注意力缓存方案均存在固有缺陷:自注意力机制会随上下文拉长线性提升推理开销;循环模型依靠固定维度向量存储历史,信息压缩能力有限;主流 GDN 等门控时序网络在时序拉长后性能增长迅速停滞。行业现有最优方案仅能承载数百步时序,距离人类长达数分钟的连续动作记忆存在三个数量级的差距,这也是机器人无法自主完成 5 分钟多工序精密装配、无法单次模仿全新作业流程、失误后难以自主修正的核心根源。

针对这一行业难题,该团队推出RoboTTT(Test-Time-Training Robot Policies)机器人策略框架,原生将视觉运动上下文拓展至 8000 时序步,对应 5 分钟连续运动记忆,全程推理成本恒定不随上下文长度增加而上升,综合时序承载能力较现有 SOTA 方案提升三个数量级,首次证明上下文长度可作为机器人基础模型全新的性能缩放维度。

详细信息:https://research.nvidia.com/labs/gear/robottt/

二、RoboTTT 核心架构:内嵌微型模型的测试时训练机制

RoboTTT 的底层创新是将测试时训练(TTT)深度融合至视觉 - 语言 - 动作(VLA)基础模型,团队基于开源通用人形机器人基础模型 GR00T N1.7 完成架构改造,核心设计为内置微型子模型的 TTT 层,以快速权重(Fast Weights)替代传统固定维度隐藏状态存储全部历史交互信息。

2.1 TTT 基础运行逻辑

每一组传感器、视觉、本体感知输入 Token 抵达 TTT 层时,会触发两套串行运算:

权重更新阶段:基于键值绑定自监督损失 TTT-KVB 执行单次梯度下降,将当前帧的环境、动作信息压缩写入微型模型的快速权重;

推理输出阶段:使用更新完成的权重生成当前动作输出。

区别于传统缓存机制,TTT 不会留存海量历史帧,而是把全部时序信息持续压缩进固定尺寸的神经网络权重,无论上下文拓展至数千步,模型隐状态占用空间始终恒定,从根源解决长时序推理算力爆炸问题。测试阶段模型持续迭代更新权重,部署后无需重训即可无限积累作业经验。

2.2 兼容预训练模型的门控机制

为避免新增 TTT 层覆盖 GR00T 原有通用操作能力,团队设计可学习 tanh 门控模块。训练初期门控输出趋近于 0,模型完全沿用预训练基础策略;随着训练推进,门控自动适配放大 TTT 分支输出权重,在保留原有通用技能的前提下,逐步激活长时序记忆能力,实现新旧模型能力平滑融合,大幅降低微调成本。

三、长时序配套训练方案:解决超长序列训练显存难题

单纯架构创新不足以支撑 8K 时序训练,我们配套研发两套专属训练技术,在固定 GPU 显存预算下实现无限长序列训练:

3.1 分段时序反向传播(TBPTT)

完整 8000 步序列直接反向传播会存储全部中间激活值,显存占用与时序长度成正比。TBPTT 将完整轨迹切分为固定长度片段,梯度仅在片段内部回流,片段边界切断梯度传播,但快速权重状态完整跨段传递。显存占用仅由分段长度决定,不受总时序步影响,让 8K、乃至未来百万级时序训练成为可行方案。

3.2 序列动作强制降噪(Sequence Action Forcing)

模型采用流匹配损失训练动作头,传统全局统一噪声等级会导致整条序列训练难度失衡。我们为每一段动作块独立随机采样噪声系数,均衡不同时序段的训练难度,稳定长序列训练收敛效果,有效解决超长轨迹训练震荡、难以拟合的问题。

3.3 元学习快速权重初始化

快速权重初始值 W₀通过双层梯度元学习优化,模型在海量机器人轨迹中学习通用时序信息压缩规则,开机即可拥有高效的历史存储基底,大幅缩短新任务的在线适配周期。

四、上下文学习范式:两种核心自适应能力

依托损失掩码机制,RoboTTT 区分 “纯上下文信息” 与 “监督动作样本”,衍生两大工业实用能力:

4.1 单视频样本上下文模仿

电路板装配场景中,仅输入统一文字指令 “组装电路”,搭配一段人类全新装配演示视频作为上下文,训练阶段屏蔽演示帧动作损失,仅用视频更新快速权重;测试时机器人仅凭单段演示视频,就能精准复刻从未见过的元器件排布与装配顺序,完成小样本一次性模仿,突破传统机器人需要数十组示范数据的限制。

4.2 故障 - 修正算法蒸馏(DAgger Distillation)

团队将机器人失误动作设为上下文、人工修正动作设为监督目标,训练模型自主学习 “失误→修复” 通用映射逻辑。部署后机器人作业中途出现抓取偏移、零件掉落、外力扰动等问题时,无需人工干预即可实时自主纠错;每一次自主修复都会存入快速权重,持续优化后续动作决策,复刻人类下意识纠错的运动反射能力。实验中人为中途取走已安装零件,RoboTTT 可识别环境变化并返回重新完成装配,抗扰动能力显著优于基线模型。

五、实测量化评估:全面超越主流基线模型

团队在双手机械臂 YAM 平台开展三类 5 分钟长时序精密装配测试:玩具车组装、遥控齿轮机器人装配、未知电路模块装配,选取三类行业主流方案作为对比基线:单步上下文 GR00T N1.7、增加单帧历史的 GR00T Hist、固定状态循环模型 GDN。

5.1 整体任务完成度

RoboTTT 平均任务完成率达 79%,较单步基线提升 87%;其中长时序玩具车装配任务完成率 78%,基线最高仅 54%;未知电路单样本模仿任务完成率 71%,大幅领先竞品。市面上所有基线模型均无法完整跑完十工序、5 分钟连续装配任务,仅 RoboTTT 可稳定全程自主完成。

5.2 上下文缩放核心结论

团队测试 128 步至 8000 步多档预训练上下文长度,得到机器人领域首个明确的上下文缩放曲线:随着时序步增加,闭环作业性能持续稳定上升,无任何性能饱和迹象。8K 时序预训练模型任务完成度,比 1K 时序版本高出 62%;反观 GDN 等循环模型,时序拉长后性能增长停滞甚至下降。这一结果证明,如同大语言模型依靠长上下文实现能力跃升,时序长度是机器人基础模型全新、可持续的性能提升路径,未来拓展至百万级时序上下文具备充分可行性。

六、落地价值与未来展望

从开发者视角来看,RoboTTT 解决了通用机器人落地的两大核心痛点:一是恒定推理开销下的超长时序记忆,适配流水线多工序连续作业;二是零微调在线自进化,依靠单段人类演示完成全新任务、自主修复操作失误,大幅降低机器人现场部署的数据与人工成本。

当前团队已验证 8000 步上下文的工程可行性,依托 TBPTT 显存优化方案,团队下一阶段目标将突破百万时序上下文,进一步拉近机器人与人类连续运动记忆的差距。长时序上下文技术将重构通用机器人的能力边界,让机器人真正理解完整作业流程,实现无需重复编程、持续自我优化的柔性智能生产。

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2026-07-16,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 GPUS开发者 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一、行业痛点:传统机器人短时记忆瓶颈
  • 二、RoboTTT 核心架构:内嵌微型模型的测试时训练机制
    • 2.1 TTT 基础运行逻辑
    • 2.2 兼容预训练模型的门控机制
  • 三、长时序配套训练方案:解决超长序列训练显存难题
    • 3.1 分段时序反向传播(TBPTT)
    • 3.2 序列动作强制降噪(Sequence Action Forcing)
    • 3.3 元学习快速权重初始化
  • 四、上下文学习范式:两种核心自适应能力
    • 4.1 单视频样本上下文模仿
    • 4.2 故障 - 修正算法蒸馏(DAgger Distillation)
  • 五、实测量化评估:全面超越主流基线模型
    • 5.1 整体任务完成度
    • 5.2 上下文缩放核心结论
  • 六、落地价值与未来展望
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档