跳转至

Generative Control as Optimization: Time Unconditional Flow Matching for Adaptive and Robust Robotic Control

会议: NeurIPS 2026
arXiv: 2603.17834
OpenReview: https://openreview.net/forum?id=tIN1NuFf6l
代码: https://hrh6666.github.io/GeCO/
领域: 机器人
关键词: 机器人控制, 流匹配, 无时间条件生成, 自适应推理, OOD检测

一句话总结

将生成式机器人动作生成从固定步数的时间积分范式重构为时不变静态速度场中的迭代优化(GeCO),实现按任务复杂度自适应分配算力与零额外开销的分布外异常检测。

研究背景与动机

基于扩散模型与流匹配(Flow Matching)的生成策略已成为现代机器人模仿学习与具身决策的核心基石,在多模态动作轨迹合成与高自由度操控任务中展现出卓越的表达能力。当前主流策略(如 Diffusion Policy、\(\pi_0\) 系列 VLA)普遍采用依赖显式时间变量的时间条件动力系统,通过将生成过程离散化为固定的去噪或常微分方程积分步数,逐步将高斯先验变换为可行控制动作序列。

然而,这种对时间条件的依赖导致了具身控制中显著的结构性低效:预设的固定积分步长完全独立于机器人当前的实际物理状态复杂度。无论机械臂是在空旷工作空间中执行低难度的直线回正,还是在精密装配与接触密集的瓶颈阶段操作,策略都被迫消耗完全相同的计算预算。这一“盲目积分”现象不仅浪费了硬件上极其紧缺的实时推理算力,而且由于时间变化矢场缺乏时不变的能量曲面,策略无法对生成动作的有效性提供内在置信度度量,更难以实时感知测试时的分布外(OOD)扰动。

本文的切入角度是解耦生成轨迹与显式时间步进,通过构建时不变的稳态速度场将动作采样重塑为无约束梯度优化。核心 idea:学习将专家示范动作作为稳定吸引子的静态速度场,通过速度重缩放衰减机制确保目标流形处梯度归零,使动作生成转化为按收敛残差自适应早退的优化过程,并以内在残差范数实现免训练的零开销分布外检测。

方法详解

整体框架

GeCO 抛弃了传统流匹配对时间标量 \(\gamma \in [0, 1]\) 的条件依赖,将动作生成任务从沿着虚拟时间轴的数值积分转化为在时不变稳态速度场 \(f_\theta(x, s_t)\) 上的极小化迭代优化。在每个闭环控制步 \(t\),智能体接收多模态观测 \(s_t\)(多视角 RGB 图像、本体感知状态及语言指令),从各向同性高斯噪声先验出发,通过多步梯度下降迭代将动作轨迹推向专家行为所处的吸引子盆地。当速度场的范数降至指定收敛容限或达到最大迭代步数时自适应早退,并直接以最终优化点的残差梯度范数作为内在环境异常度评估信号。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["多模态环境观测 s_t<br/>RGB 图像 + 本体感知 + 语言指令"] --> B["静态速度场构建<br/>时间无关矢场 f_θ(x, s_t)"]
    B --> C["自适应迭代优化推理<br/>a^(k+1) = a^(k) - η_k f_θ(a^(k), s_t)"]
    C -->|梯度范数 ||f_θ|| < τ| D["自适应收敛早退<br/>输出动作块 a^ 并下发执行"]
    C -->|持续大残差 ||f_θ|| > τ_OOD| E["内在几何分布外检测<br/>在线报警并提前终止交互"]

关键设计

1. 静态速度场构建:消除时间条件并引入速度衰减吸引子机制 传统流匹配学习时间依赖的速度场 \(v_\theta(x_\gamma, \gamma, s_t)\),积分步数若偏离 \([0, 1]\) 训练区间系统将发散。GeCO 在模型输入端直接抹除时间标量 \(\gamma\),仅在训练数据插值采样阶段将 \(\gamma \sim \mathcal{U}(0, 1)\) 作为隐式插值系数构造混合动作 \(x_\gamma = \gamma a + (1-\gamma)\varepsilon\)(其中 \(a\) 为专家动作真值,\(\varepsilon \sim \mathcal{N}(0, I)\) 为高斯白噪声)。为了使专家动作 \(a\) 成为静态场的自然稳定吸引子并允许优化算法在无时间引导下自主收敛,GeCO 引入速度重缩放标量函数 \(c(\gamma)\) 构造监督目标: $\(g^\star(a, \varepsilon, \gamma) = (\varepsilon - a) \, c(\gamma), \quad c(\gamma) = \begin{cases} \lambda, & \gamma \le \alpha_0 \\ \lambda \frac{1-\gamma}{1-\alpha_0}, & \gamma > \alpha_0 \end{cases}\)$ 其中 \(\lambda\) 为尺度常数,\(\alpha_0\) 为线性衰减转折阈值。当插值逼近专家动作(\(\gamma \to 1\))时,\(c(1) = 0\) 强制目标速度矢量严格衰减归零。模型通过均方误差损失 \(\mathbb{E}_{(s_t, a), \varepsilon, \gamma} \|f_\theta(x_\gamma, s_t) - g^\star(a, \varepsilon, \gamma)\|^2\) 学习统一的静态矢场,使专家示范在动作空间中形成势阱极小值。

2. 自适应迭代优化推理:按任务难度动态调配算力 在推理部署时,动作合成脱离固定积分轨迹,转化为标准的梯度优化流程。控制策略从高斯先验 \(a^{(0)} \sim \mathcal{N}(0, I)\) 初始化,采用逐步递减的学习率调度序列 \(\{\eta_k\}\) 执行迭代更新: $\(a^{(k+1)} = a^{(k)} - \eta_k f_\theta(a^{(k)}, s_t)\)$ 推理循环实时监控场梯度范数 \(\|f_\theta(a^{(k)}, s_t)\|_2\)。一旦范数低于收敛容限 \(\tau_{\text{opt}}\),即表明动作已落入目标极值区域,优化器立即早退截断;若面对高精度对齐或接触丰富的狭窄操作空间,优化过程则持续迭代至最大步数上限 \(K_{\max}\)。这种机制完全打破了积分总步长必须等于 1 的硬性约束,实现了简单状态(如自由空间转移)极速输出、复杂操作精细打磨的计算弹性。

3. 内在几何分布外检测:免训练的稳态梯度残差安全信号 分布内(ID)环境观测 \(s_t \sim p_{\text{data}}\) 所诱导的条件速度场在专家动作周围具有明确收敛的稳态势阱,经过迭代优化后动作收敛点的残差范数必定接近于零;而遭遇非分布(OOD)的环境几何、未知障碍物或受扰动指令时,模型无法诱导出具有一致收敛性的势能流形,优化终点的矢量范数将呈现持久的大梯度残留。GeCO 定义每个动作规划调用的异常分数为 \(\text{Score}(s_t) = \|f_\theta(\hat{a}(s_t), s_t)\|_2\)。通过结合移动平均滤波或漏桶累加器(Leaky Bucket),机器人能在执行不可恢复的失败动作前,零额外开销地在线触发报警并提前截断交互,大幅节省异常环境中的无效动作消耗。

损失函数 / 训练策略

训练采用端到端参数优化,直接监督时不变速度头输出。对于基于 Diffusion Transformer (DiT) 的架构,采用 DINOv2 冻结视觉特征与 T5 文本嵌入通过交叉注意力注入;对于 VLA 大模型(如 \(\pi_0\) 系列),由于其原本包含时间嵌入通道,GeCO 采取直接将时间输入置零(\(t=0\))的即插即用策略进行微调,冻结时间维度并重用预训练权重。超参数设置 \(\lambda = 4\)、\(\alpha_0 = 0.1\),优化最大步数设为 \(K_{\max} = 30\)(早退容限 \(\tau = 0.4\)),学习率采用分段衰减策略(从 0.1 逐渐降低至 0.01)。

实验关键数据

主实验

在 LIBERO 标准机械臂模拟操作基准评测中,GeCO 与固定步数扩散策略(Diffusion Policy)及流匹配基线(Rectified Flow)进行了详尽对比,验证了自适应算力分配机制在成功率与推理效率之间的双重优势。

方法 最大步数 Goal (%) Spatial (%) Object (%) Long (%) 平均成功率 (%) 平均 NFE (步)
Diffusion Policy 100 (固定) 82.9 91.4 88.9 82.7 86.5 100.0
Rectified Flow 20 (固定) 92.4 94.6 97.0 76.0 90.0 20.0
GeCO (Ours) 5 91.6 95.4 98.2 82.4 91.9 5.0
GeCO (Ours) 10 93.0 95.8 99.0 81.8 92.4 8.7
GeCO (Ours) 20 95.2 95.8 99.0 83.8 93.5 11.6
GeCO (Ours) 30 93.6 95.2 98.8 84.8 93.1 12.8

在更具挑战性的长程推理与复杂具身基准 VLABench 上,将 \(\pi_0\) 的动作生成头无缝替换为 GeCO,显著提升了具身大模型的操作鲁棒性。

方法 Track 1 (基准操作) Track 2 (跨类泛化) Track 3 (常识知识) Track 4 (指令跟随) Track 6 (纹理扰动) 平均成功率 (%)
\(\pi_0\) (Baseline) 47.0 21.2 29.1 17.3 32.2 29.4
\(\pi_{0.5}\) 40.6 22.6 18.0 16.1 25.6 24.5
\(\pi_0\) + GeCO (Ours) 61.0 22.0 34.0 20.0 45.0 36.0

在 Galaxea R1 Lite 双臂移动机器人真机高难度操作中,针对毫米级精度的双螺母旋拧装配(Nut Assembly)和带随机偏角试管整理插拔(Tube Arrangement)两项瓶颈任务进行了实测: - Nut Assembly:成功率由基线 G0 Plus 的 30% 跃升至 70%,平均 NFE 由 10.0 降至 5.06,单步平均推理延迟由 252.3 ms 降至 172.4 ms(降低 31.7%)。 - Tube Arrangement:成功率由基准的 25% 飙升至 80%,平均 NFE 降至 3.82,推理耗时从 254.6 ms 降至 150.3 ms(降低 41.0%)。

消融实验

在速度重缩放截断衰减参数 \(\alpha\)(对应公式中平滑与急变过渡阈值)对 LIBERO 基准性能影响的消融研究中,验证了平衡向量场平滑性与吸引子锐利度的重要性。

配置 (\(\alpha\)) Object (%) Goal (%) Spatial (%) Long (%) 平均成功率 (%) 说明
\(\alpha = 0.0\) 98.4 94.6 93.4 77.0 90.9 全程线性衰减,平滑但吸引子收敛较缓
\(\alpha = 0.2\) 96.8 95.4 95.8 80.2 92.1 早期衰减,保持稳定收敛
\(\alpha = 0.4\) 98.2 93.4 95.6 78.2 91.4 中等过渡区
\(\alpha = 0.6\) 98.2 94.2 95.6 77.8 91.5 场过于平滑,步数限制下收敛精度略降
\(\alpha = 0.8\) (本文) 97.8 94.0 96.0 81.0 92.2 达到极值清晰度与场平滑度的最佳平衡
\(\alpha = 1.0\) (无截断) 93.2 84.0 89.8 55.8 80.7 失去吸引子平衡点,无自然收敛终点,性能骤降

在 OOD 异常检测实验中(以 LIBERO-Goal 训练,LIBERO-Spatial 作为空间布局分布外输入): - 采用传统训练 Loss 代理指标(Moving Avg)的规划级 AUROC 仅为 0.53(TNR 14.4%, TPR 27.8%); - 采用 GeCO 优化残差范数结合 Moving Avg 滤波的 AUROC 高达 0.93(TNR 82.4%, TPR 89.7%),在 OOD 交互中平均提前避免了 42.4% 的无效执行时间;结合 Leaky Bucket 滤波后 TNR 进一步达 84.0%,节约 44.3% 的交互时间。

关键发现

  • 算力消耗与物理任务难度强相关:空间重定位任务中,NFE 高峰精准对齐预抓取对齐和精密放置瓶颈,自由空间回正仅需 1~3 步;在目标辨识任务中,决策起点因需消除视觉语义歧义产生算力波峰。
  • 速度衰减机制是时不变生成能够稳定收敛的理论前提:当移除速度衰减(\(\alpha = 1.0\))时,向量场在专家动作处不消失,导致优化算法产生持续高频震荡,Long 任务成功率从 81.0% 断崖式跌落至 55.8%。
  • 即插即用适应性优异:无需更改 \(\pi_0\) 骨干架构,仅在微调时固定时间通道输入 \(t=0\),即可使 VLA 大模型兼具大尺度视觉语义先验与本地自适应优化推理能力。

亮点与洞察

  • 将轨迹积分抽象为能量场下山:跳出扩散模型必须沿固定虚拟时间积分的思维定势,利用静态向量场将动作生成还原为最基础的一阶数值优化,兼顾了生成分布多样性与经典控制理论中的李雅普诺夫稳态吸引子特性。
  • 免训练的内生安全报警器:利用未收敛残差矢量直接表征认知不确定性,无需额外引入集成网络(ensemble)或不确定性估计判别器,实现计算与安全的零边际成本复用。
  • 极简工程迁移路径:对已有时间条件模型的“置零微调”方案展示了惊人的实用性,为当前工业界加速 VLA 实时部署提供了高价值的技术范式。

局限与展望

  • 理论边界仍有待完善:当前对于静态速度场在复杂非凸动作空间中的收敛速度、局部极小值捕获概率以及 OOD 检测的假阳性率尚缺乏严格的理论分析与泛化界限定理。
  • 步长调度的经验依赖:目前推理阶段仍依赖人工设计的递减学习率序列与固定的梯度范数早退门限,未来可探索自适应线搜索(Line Search)或基于 Hessian 信息的二阶预条件自适应优化器。

相关工作与启发

  • vs Diffusion Policy / Rectified Flow: 传统方法依赖显式时间变量积分,推理步数固定且无法中断;GeCO 学习无时间条件的静态场,推理是自由步长的凸/准凸势阱下山过程,步数随难度自适应调节。
  • vs Implicit Behavioral Cloning (IBC) / EBM: IBC 直接训练能量标量网络并计算显式梯度,训练极难收敛且采样昂贵;GeCO 直接对齐无旋速度矢量场,训练更加稳健且采样迅速。
  • vs Diff-DAgger (不确定性度量): Diff-DAgger 需在多步时间步上对扩散误差进行密集采样估计;GeCO 直接复用推理过程已经算出的当前步优化梯度范数,无需额外前向推理即可得到高灵敏度 OOD 信号。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 巧妙重构了生成式控制的基础数学范式,提出时不变流匹配与速度衰减吸引子设计。
  • 实验充分度: ⭐⭐⭐⭐⭐ 涵盖 LIBERO、RoboTwin 2.0、VLABench 三大具身基准及物理双臂机器人高精度真机验证。
  • 写作质量: ⭐⭐⭐⭐⭐ 逻辑严密,图文对应清晰,机理解释透彻深入。
  • 价值: ⭐⭐⭐⭐⭐ 为解决具身智能中扩散模型高延迟、高能耗与动作盲目积分痛点提供了极佳落地方案。