← 返回 2026-10-21

VLA-Rolling:基于自适应动作尾部复用,用更少去噪步数实现更好的机器人控制 VLA-Rolling: Fewer Denoising Steps, Better Control via Adaptive Action Reuse

Sicheng He, Yi Pan, Qiuting Yu, Jiaming Huang, Qi Lu, Shuyan Wu, Hongyan Feng, Xuanyu Liu, Jun Zhang, Jie Zhang, Wenqi Zhang, Xuhong Zhang, Jianwei Yin 📅 2026-10-18 👍 9 2026-11-11 18:30
VLA 动作分块 推理加速 机器人操作 流匹配 置信度估计

复用被丢弃的动作尾部,自适应分配去噪预算,让VLA更快更准

前置知识

流匹配 (Flow Matching)

一种生成式建模范式:训练速度场网络 $v_\\theta(x, \\tau)$,把高斯噪声(流时刻 $\\tau=0$)沿学到的向量场用欧拉法逐步积分到数据分布($\\tau=1$)。推理时通过 $N$ 步欧拉迭代完成去噪,步数越多越精确但越慢。π0、π0.5、SmolVLA 等机器人策略用流匹配生成连续动作,本文默认 $N_{full}=10$ 步。

本文的核心操作——把动作尾部分配到合适的流时刻 $\\tau$ 继续去噪、以及按置信度做'再噪化'插值——全部建立在流匹配的连续时间公式之上,不懂数值积分就无法理解方法。

VLA 与动作专家

视觉-语言-动作模型把视觉观测 $o_t$、本体状态 $q_t$ 和语言指令 $\\ell$ 输入 VLM 主干生成上下文(KV cache),再交给一个小得多的 Transformer '动作专家',用流匹配迭代去噪出连续动作。例如 π0.5 的 VLM 主干约 3B,而动作专家仅 30M 量级。动作专家的多步去噪占每次推理时延的主要部分。

VLA-Rolling 只加速动作专家的去噪阶段、完全冻结 VLA 主干,复用的正是动作专家已完成的部分去噪计算,必须理解主干与专家的分工才能明白节省从何而来。

动作分块与滚动时域执行

策略一次从单帧观测预测 $H$ 步动作块(本文 $H=50$),但只执行前 $K$ 步($K=10$)就重新观测再规划,以抑制开环执行累积的误差;未执行的后 $L=H-K$ 步称为'动作尾部',传统做法直接丢弃。这种以额外推理换取反馈闭环的范式是当前所有主流 VLA 的标准部署方式。

动作尾部正是本文要'抢救'的对象。理解滚动时域执行为什么丢弃尾部,才能明白论文发现的系统性计算浪费在哪里。

自适应去噪步数(逐样本预算)

不同样本所需去噪步数差异巨大:论文测得 π0.5 逐样本最优步数 $N^*=\\arg\\min_N E_N$ 从 1 到 10 全谱分布(均值 7.07、标准差 2.28)。固定步数要么浪费算力要么欠精化。本文用整块最小置信度线性映射预算:$N_{adapt}=\\mathrm{clip}(\\lceil(1-c_{min})N_{full}\\rceil, N_{min}, N_{full})$。

这是把'省算力'与'保精度'统一起来的关键机制,也是论文诊断实验(Figure 3)的直接推论:既然 $N^*$ 逐样本变化,预算就必须逐样本分配。

分位数归一化与 EMA 在线统计

动作尾与目标动作的误差 $E_t$ 跨多个数量级,直接回归难以学习。论文先取对数 $r_t=\\log E_t$ 压缩大误差、放大细微差异,再用 10%/90% 分位数做线性压缩到 $[0,1]$ 作为置信度标签 $c^*_t$;分位数用指数移动平均($\\eta=0.99$)在训练批间在线更新,避免对整个演示数据集重复统计。

置信度标签的构造决定了估计器学到什么;非对称 L1 损失(高估置信度罚 2 倍)与跨模型、跨环境的误差尺度统一都依赖这一标签设计。

研究动机

流匹配 VLA(π0.5、SmolVLA 等)一次从高斯噪声迭代去噪出 $H$ 步动作块,但在滚动时域执行下只运行前 $K$ 步(主流配置 $H=50, K=10$),剩余 $L=40$ 步的'动作尾部'连同其已完成的去噪计算被整体丢弃,下一轮规划从全新高斯噪声重新起步。以 π0.5 为例,每个动作块推理实测需 168-223 ms(LIBERO 168.09 ms、MetaWorld 222.69 ms),机器人每 10 个控制步就要为一次'从零去噪'买单,而其中大部分计算产物被扔掉。更关键的是,作者统计发现相邻环境步的变化极小:约 98% 的相邻步观测与本体状态高度相似(LIBERO 图像余弦相似度 0.953、本体状态余弦 0.998;MetaWorld 分别为 0.992 和 0.991,见 Table 9)。也就是说环境在两次规划之间几乎没有变,被丢弃的尾部仍然高度可用——这是一笔每 10 步发生一次、贯穿整个机器人运行过程的系统性浪费。

本文的目标是本文想回答的核心问题是:能否利用被丢弃的动作尾部,在不牺牲动作精度的前提下,用更少的去噪步数生成同等甚至更好的动作?具体拆成三问。其一,尾部是否保留了可迁移的去噪进度——诊断实验显示 π0.5 用尾部初始化只需 4 步去噪即可追平甚至超越 10 步高斯基线(误差比 1.039/1.007),SmolVLA 只需 6 步(1.018/1.006)。其二,每个新观测下应该'信任'尾部多少:逐样本最优步数 $N^*$ 在 1-10 间几乎全谱分布(π0.5 标准差 2.28,SmolVLA 2.88),且尾部去噪并非步数越多越好(π0.5 在 LIBERO 上误差比从 6 步的 1.073 降到 8 步的 1.037),说明固定预算行不通。其三,如何设计一个不改动、不重训冻结 VLA 的轻量推理框架,让单一置信度信号同时决定'保留多少尾部信息'(再噪化强度)与'花多少算力精化'(去噪步数预算),最终实现推理成本与任务成功率的帕累托改进。

与已有工作不同的是,已有两条相关路线各有盲区。动作块复用路线(时序一致性类工作)直接把上一块动作拿来用,但缺乏动作级的质量评估机制——低质量的块也会被无差别继承,导致错误在轨迹上传播;推理加速路线(蒸馏、少步采样、压缩动作解码器)通常要从头训练或牺牲动作精度,论文指出这类方法很少能在现有模型上通用并带来帕累托改进。本文的独特切入是把被丢弃的尾部视为流匹配轨迹上的'合法中间状态'而非废弃输出:既然 98% 的相邻环境步几乎不变,尾部就天然处于离目标不远的去噪中途,剩下的问题只是'逐样本决定还要走几步、退多少噪声'。为此引入一个仅 30M 参数、以演示轨迹内时间相邻动作对的误差为监督的置信度估计器,把质量评估下沉到动作级,并用置信度同时驱动再噪化与步数分配,完全冻结原 VLA——这在'复用但带评估'和'加速但不重训'两个维度上同时填补了空白。

核心方法

直觉上,上一轮没执行完的动作尾是对当前动作的'半成品草稿':环境变化不大时草稿比白纸(高斯噪声)更接近答案,但草稿也可能过时,需评估后再用。VLA-Rolling 分三阶段滚动进行。阶段一(动作尾继承):把上轮尾部 $A^{t-K}_{K:H}\in\mathbb{R}^{L\times d_a}$($L=H-K=40$)按时间对齐到当前规划窗口,并计算环境变化 $\Delta o_t=o_t-o_{t-K}$、$\Delta q_t=q_t-q_{t-K}$。阶段二(置信度估计):一个 30M 参数的 action-centric Transformer 读入尾部、当前观测/状态及其差分,为每个位置输出置信度 $c\in\mathbb{R}^{L\times 1}$,前向约 1.5 ms。阶段三(置信度引导精化):低置信位置注入更多高斯噪声(再噪化),整块按最小置信度分配自适应去噪步数 $N_{adapt}$,用与原 VLA 相同的欧拉更新沿剩余流轨迹继续积分,执行前 $K$ 步后新尾部滚入下一轮。全程冻结 VLA。

核心创新是'把被丢弃的动作尾当作流匹配去噪轨迹上的合法中间状态'。若尾部的最优剩余精化预算为 $N^*$,就把它的流时刻定在 $\tau_k=1-\frac{N^*-k}{N_{max}}$ 而非 $\tau=0$,再用原 VLA 的速度场 $v_\theta$ 与步长 $\delta=1/N_{max}$ 继续欧拉迭代——时间步嵌入与离散化完全兼容,无需改动模型。第二个创新是单一置信度同时控制两个自由度:再噪化按位置插值 $A_{init,i}=c_i A_{tail,i}+(1-c_i)\epsilon_i$,决定每个位置保留多少旧信息;预算 $N_{adapt}$ 决定整块花多少算力。置信度标签无需人工标注:取同一演示轨迹中相隔 $K$ 步的两时刻,用冻结 VLA 在旧状态生成的尾部与新时刻演示动作对齐,算 $E_t=\|A^{t-K}_{tail}-A^*_t\|_2^2$,经 $\log$ 与 EMA 分位归一化得标签,并以 $w^+{=}2>w^-{=}1$ 的非对称 L1 损失重罚高估。这与'整块无差别复用'或'固定少步采样'有本质区别。

方法步骤详情

流程分离线训练与在线推理。训练:在同一演示轨迹上取相隔 $K$ 步的 $t-K$ 与 $t$,用冻结 VLA 在旧状态生成动作块并截出尾部,与新时刻演示动作 $A^*_t$ 对齐算误差 $E_t$,取 $r_t=\log E_t$,用 EMA($\eta=0.99$)维护的 $q_{0.1},q_{0.9}$ 分位归一化得标签 $c^*_t$,以 $w^+{=}2,w^-{=}1$ 的加权 L1 损失训练(AdamW,lr $10^{-4}$,batch 64,4000 步,单卡 96GB)。估计器输入长度与 VLA 的 $H$ 对齐(尾部前填充高斯噪声),注意力掩码屏蔽填充,输出截取前 $L$ 个置信度。在线推理每轮四步:(1) 继承尾部并计算环境变化 $\Delta o_t,\Delta q_t$,估计器前向输出逐位置置信度 $c$;(2) 按公式 (9) 逐位置再噪化,低置信位置接近纯噪声;(3) 取 $c_{min}=\min_i c_i$,按 $N_{adapt}=\mathrm{clip}(\lceil(1-c_{min})N_{full}\rceil,N_{min},N_{full})$ 分配预算,把初始化点放到对应流时刻执行 $N_{adapt}$ 步欧拉去噪(校准参数 $\alpha=1.5$);(4) 执行前 $K$ 步,新尾部滚入下一轮。

技术新颖性

新颖性体现在四个层面。第一,问题定义新:以往工作把滚动时域丢弃的尾部当作计算废料,本文用诊断实验证明它是'已付成本的可回收中间态'(π0.5 用 4 步、SmolVLA 用 6 步追平 10 步高斯基线),属于对 VLA 推理范式的重新审视。第二,机制设计新:把尾部映射到流匹配中间时刻 $\tau$ 而非当作 $\tau=0$ 的起点,时间步嵌入与原模型一致——这是与 naive 复用的根本差异,消融显示只继承尾部会使 LIBERO-Plus 掉 4.89pp、LIBERO 掉 2.13pp。第三,监督构造新:不需要额外标注或在线试错,仅用演示轨迹内时间相邻动作对的对齐误差作监督,配合 EMA 分位归一化($\eta=0.99$)解决跨模型、跨环境的误差尺度漂移;非对称损失(高估罚 2 倍)防止'过度信任'造成不可恢复错误。第四,工程属性新:插件仅 30M(10M 不够好、100M 无增益,30M 成功率 98.00% 且开销仅 1.579 ms),框架与模型解耦,在 4 个 VLA、4 个基准上即插即用,理论上可与蒸馏、量化等其他加速手段叠加。

VLA-Rolling reuses rather than discards action tails. Confidence controls both how strongly inherited actions are renoised and how many denoising steps refine the next chunk.
Figure 1: VLA-Rolling reuses rather than discards action tails. Confidence controls both how strongly inherited actions are renoised and how many denoising steps refine the next chunk.
Overview of VLA-Rolling. VLA-Rolling reuses the unexecuted action tail from the previous inference round by estimating its confidence, which jointly determines tail preservation and the adaptive denoising budget while keeping the original VLA frozen.
Figure 4: Overview of VLA-Rolling. VLA-Rolling reuses the unexecuted action tail from the previous inference round by estimating its confidence, which jointly determines tail preservation and the adaptive denoising budget while keeping the original VLA frozen.

实验结果

主实验(Table 1):4 个 VLA × 4 个基准共 16 组合上推理成本无一例外下降(-5.00% 到 -32.52%,最高 1.48 倍吞吐),14 组成功率同时提升,最大 +15.08pp(SmolVLA 在 LIBERO-Plus 从 20.74% 升至 35.82%);仅 X-VLA 两组微降 0.13/0.28pp。消融(Table 2):只继承尾部(+A)在 LIBERO 掉 2.13pp、LIBERO-Plus 掉 4.89pp,naive 复用有害;+A+C 省 22.19% 成本但掉 1.48pp;完整模型(A+B+C)才同时 +1.45pp 与 -23.03% 成本。分析实验:环境剧变与置信度负相关($\rho(\Delta o,c_{min})$ 从 K=10 的 -0.257 衰减到 K=40 的 -0.100);MetaWorld 上成功轨迹比失败轨迹多用至多 2.33 步去噪(Very hard 8.34 vs 6.01);$N^*\pm1$ 扰动使成功率降 0.61-1.21pp,说明预算预测局部最优。敏感性消融(Table 5-7):校准 $\alpha=1.5$ 最优(98.50%、6.332 步);最小置信度转换误差最低(LIBERO 53.739 vs mean 54.420);估计器 30M 是甜点(98.00%、1.579 ms)。真机(Table 4):π0.5 换盘突变任务成功率 70%→75%(15/20),延迟 227.28→176.50 ms;Figure 6 显示目标突变处置信度跳变并自动触发更多去噪步。

Performance and inference efficiency comparison with VLA-Rolling. SR denotes the average success rate across tasks, and Cost denotes the inference time per action chunk.
Table 1: Performance and inference efficiency comparison with VLA-Rolling. SR denotes the average success rate across tasks, and Cost denotes the inference time per action chunk.
Ablation study of VLA-Rolling inference components with π0.5 on LIBERO, MetaWorld, and LIBERO-Plus. A: Action Tail Inheritance; B: Confidence-Guided Renoising; C: Adaptive-Step Denoising.
Table 2: Ablation study of VLA-Rolling inference components with π0.5 on LIBERO, MetaWorld, and LIBERO-Plus. A: Action Tail Inheritance; B: Confidence-Guided Renoising; C: Adaptive-Step Denoising.
Observation shift and confidence statistics under different execution horizon K. Correlation values ρ are Pearson correlations computed within each K group.
Table 3: Observation shift and confidence statistics under different execution horizon K. Correlation values ρ are Pearson correlations computed within each K group.
Model performance on real-world experiments.
Table 4: Model performance on real-world experiments.
Sensitivity of VLA-Rolling performance to the calibration parameter α on LIBERO. We test 40 tasks with 10 episodes per task.
Table 5: Sensitivity of VLA-Rolling performance to the calibration parameter α on LIBERO. We test 40 tasks with 10 episodes per task.
Offline trajectory-level action error under different denoising step convert strategies.
Table 6: Offline trajectory-level action error under different denoising step convert strategies.
Performance and inference cost of VLA-Rolling models on LIBERO.
Table 7: Performance and inference cost of VLA-Rolling models on LIBERO.
Observation and proprioceptive-state similarity across a 10-step execution horizon.
Table 9: Observation and proprioceptive-state similarity across a 10-step execution horizon.
Average denoising steps across task difficulty levels on MetaWorld. Successful trajectories consistently require more denoising steps than failed trajectories.
Figure 5: Average denoising steps across task difficulty levels on MetaWorld. Successful trajectories consistently require more denoising steps than failed trajectories.
Real-world experiment of context mutations revealed a significant decrease in confidence at the mutation site, leading to an increase in the number of denoising steps.
Figure 6: Real-world experiment of context mutations revealed a significant decrease in confidence at the mutation site, leading to an increase in the number of denoising steps.
查看结构化数据
任务指标本文基线提升
LIBERO(π0.5 主干,40 任务 × 100 回合) 平均成功率 / 每块推理耗时 97.83% / 129.60 ms 96.38% / 168.09 ms +1.45 pp,成本 -22.90%
MetaWorld(SmolVLA,50 任务 × 100 回合) 平均成功率 / 每块推理耗时 70.74% / 102.57 ms 65.84% / 122.88 ms +4.90 pp,成本 -16.53%
LIBERO-Plus(SmolVLA,扰动鲁棒性基准) 平均成功率 / 每块推理耗时 35.82% / 120.97 ms 20.74% / 130.90 ms +15.08 pp(最大提升),成本 -7.59%
LIBERO(MolmoAct2) 平均成功率 / 每块推理耗时 95.70% / 196.61 ms 95.50% / 291.34 ms +0.20 pp,成本 -32.52%(最大加速,1.48×)
MetaWorld(X-VLA) 平均成功率 / 每块推理耗时 61.46% / 90.05 ms 57.08% / 123.31 ms +4.38 pp,成本 -26.97%
真机换盘突变任务(π0.5 + AgileX PiPER 6-DoF) 成功率(20 次试验)/ 每块推理耗时 75%(15/20)/ 176.50 ms 70%(14/20)/ 227.28 ms +5 pp,成本 -22.4%

局限与改进

作者承认的局限:跨环境泛化探索有限,真机只有一个换盘突变任务、每种设置仅 20 次试验。我的观察:其一,置信度与环境变化的皮尔逊相关仅 -0.20 到 -0.32,K 增大到 40 时 $\rho(\Delta o,c_{min})$ 衰减到 -0.100,估计器对'该信任多少'的解释力有限且外推不足,更多是经验性校准;其二,LIBERO-Plus 每任务只跑 1 回合,SmolVLA +15.08pp 的亮眼数字可能受采样方差影响;其三,X-VLA 两组成功率微降(-0.13/-0.28pp),说明框架并非无条件帕累托——X-VLA 动作块更短($H=30,L=20$)时收益打折;其四,成本统计只含模型推理(KV cache + 去噪 + 估计器),不含感知与控制循环延迟,真机异步执行下执行时域 $K$ 随自适应步数波动(论文以上下界差不超过 100ms 论证可忽略),实际部署需要额外时序工程;其五,预算由整块最小置信度统一决定(Table 6 中 min 平均 6.33 步 vs mean 4.36 步),位置级差异只体现在再噪化上。

独立分析的弱点

弱点一:估计器条件信息较浅——视觉 token 只做均值池化,再与差分表示 LayerNorm 后拼接投影成 576 维上下文,丢失空间细节,这可能是其与真实环境变化相关性偏低($|\rho|\approx0.2\text{--}0.3$)的原因;改进方向是改用跨注意力或可学习空间池化,甚至接轻量世界模型预测'若执行尾部会到达什么状态'来直接构造复用价值信号。弱点二:估计器与具体 VLA 和 horizon 绑定,每个新模型都要重新采数据训练 4000 步、重建分位统计;可在多模型尾部分布上联合训练通用估计器或引入元学习适配。弱点三:$c_{min}$ 取整块最小值过于保守,Table 6 显示它误差最小但平均步数最多(LIBERO 6.33 步 vs mean 4.36 步),预算可下放到位置级或滑动窗口级。弱点四:训练目标是对齐演示动作的误差,而演示动作未必最优,置信度实际学的是'与演示一致'而非'与最优一致',在演示质量参差时可能系统性偏差。弱点五:每轮只复用上一轮尾部,多轮滚动后早期信息是否被反复再噪化稀释、长程任务是否存在'误差记忆',论文未分析,值得对照实验量化。

未来方向

作者提出的方向:把真机评估扩展到更复杂的任务设定与更多本体、交互场景,厘清除环境突变外还有哪些因素驱动置信度预测;将该推理范式与其他推理加速机制(蒸馏、量化、缓存复用等)结合,因为本方法不修改原模型、天然可组合,有望叠加出更大的加速比。基于成果可自然延伸的:一是把'尾部即流轨迹中间状态'的思想推广到流匹配/扩散策略的其他应用,如图像编辑的局部重绘、分子结构的局部修改——任何'滚动时域生成'场景都存在同样的被丢弃中间态;二是让执行时域 $K$ 本身自适应:既然置信度能感知环境变化幅度,$K$ 与去噪预算 $N_{adapt}$ 可以联合优化,环境稳定时多执行几步、突变时更频繁重规划;三是把置信度信号回传给数据采集端,构造面向困难状态的加权数据飞轮;四是在双臂协调(RoboTwin 已涉及)与多智能体场景验证逐位置再噪化对协调动作的价值;五是理论化工作:刻画尾部复用收益上界与环境漂移量、起始流时刻误差的解析关系,回答'何时该彻底弃用尾部'这一边界问题。

复现评估

复现条件中等偏友好。论文细节充分:估计器架构(30M action-centric Transformer、视觉 token 均值池化 + 差分表示拼接投影到 576 维、注意力掩码屏蔽高斯填充、sigmoid 输出)、全部训练超参($w^+{=}2,w^-{=}1$、EMA $\eta=0.99$、AdamW lr $10^{-4}$、batch 64、4000 步、校准 $\alpha=1.5$)、horizon 配置与真机协议(Lerobot pi05-base + AgileX PiPER,蓝/棕盘各采 20 条轨迹微调)都写在正文与附录。依赖的 4 个 VLA 与 4 个基准均为公开资源。算力门槛:训练估计器只需单张 96GB GPU,仿真评测用 4×RTX 4090。风险点:正文只给出项目主页未明确代码仓库链接,估计器实现与 EMA 分位统计需自行复刻;$N^*$ 诊断实验需大量 rollout 采样与动作对齐标注;真机需自备机械臂与遥操作数据。总体而言,仿真侧复现难度中等,真机侧门槛较高。

相似论文推荐