论文介绍了一种名为 FAST (Frequency-space Action Sequence Tokenization) 的新型动作分词(Tokenization)方法,旨在解决高频、复杂具身智能任务中自回归(Autoregressive)模型(如 VLA)的训练难题。


一、 论文核心要点总结

  1. 核心痛点:
    • 现有的 VLA 模型(如 OpenVLA)多采用简单的“逐维度、逐时间步”分词(Binning),在高频控制(如 50Hz)下,相邻动作高度相关。
    • 这导致自回归模型的次标记预测(Next-token Prediction)目标变得极其容易(只需复制上一个动作),使模型陷入局部最优,无法学习到真正的运动动力学。
  2. 创新方案 (FAST):
    • 频域转换:利用离散余弦变换 (DCT) 将动作时序信号从空域转换到频域。
    • 时序压缩:DCT 能够集中信号能量,通过舍弃高频微小系数实现压缩,减少冗余。
    • 离散化流程:归一化 -> DCT -> 量化 -> 扁平化(低频优先) -> BPE (字节对编码)。
  3. 主要贡献:
    • FAST+ 通用分词器:在 100 万条真实机器人轨迹上预训练,支持不同构型、不同控制频率。
    • 性能突破:使自回归模型能胜任复杂的灵巧操作任务(如折衣服),且训练速度比扩散模型(Diffusion-based VLA,如 π0\pi_0π0​)快 5 倍。
    • 泛化性:在 DROID 数据集上实现了显著的零样本(Zero-shot)泛化能力。

二、 提问-回答

Q1: 为什么要对动作进行压缩(Tokenization)?直接像普通 VLA 那样做 Binning 有什么问题?

A1: 在低频任务中,Binning 是可行的。但在**高频(>20Hz)和长动作块(Action Chunking)**预测中,动作序列在时域上高度冗余。如果直接对每一帧进行离散化,自回归模型预测下一个 token 时,由于信号变化极小,模型只需通过简单的“恒等映射”就能获得极低的 Loss,这会导致训练塌缩,无法捕捉到复杂的运动模式。FAST 通过频域压缩,将冗余信息剔除,强制模型学习控制序列的本质频谱特征。

Q2: 为什么选择 DCT(离散余弦变换)而不是小波变换(Wavelets)或学习型的 VQ-VAE?

A2:

  • 相比 VQ-VAE:学习型的分词器训练复杂,且对超参数极度敏感,容易出现量化误差导致的控制不稳定。DCT 是解析解,无需训练,且具有极佳的能量集中特性。
  • 相比小波变换:DCT 在处理平滑信号(如典型的机器人轨迹)时,压缩效率更高,且在标准信号处理(如 JPEG)中已经证明了其鲁棒性。
  • 简洁性:DCT 计算速度快,且支持变长动作序列,非常适合作为大规模 VLA 的预处理模块。
Q3: 论文中提到的分词流程里,BPE(Byte-Pair Encoding)的作用是什么?它不是处理文本的吗?

A3: 这是本论文的一个妙处。在 DCT 变换并量化后,会产生大量的零系数(高频部分)和重复的系数组合。

  • BPE 的作用是无损压缩这些离散的系数序列。
  • 它将频繁出现的系数序列组合成一个新的“宏 Token”。
  • 这样做不仅显著缩短了自回归预测的序列长度(论文显示压缩率提升了 2x-13x),还平衡了词表分布,提高了模型的学习效率。
Q4: 在扁平化(Flattening)DCT 系数矩阵时,论文强调要“低频优先(Low-frequency first)”,这背后有什么直觉?

A4: 这符合自回归预测的逻辑。DCT 的低频系数决定了动作序列的总体趋势和轮廓(比如手臂的大致移动方向),而高频系数决定了细节抖动和精准修正。

  • 在推理时,先生成低频 Token,相当于让模型先确定“我要去哪”,再生成高频 Token 来微调“我怎么更准”。
  • 实验证明,这种从宏观到微观的生成顺序比逐时间步生成更稳定,能显著减少轨迹的漂移。
Q5: FAST 相比扩散模型(Diffusion Policy/VLA)在训练和推理上的优劣是什么?

A5:

  • 训练端:FAST 采用纯自回归训练,无需像 Diffusion 模型那样进行多次去噪迭代。实验表明,在达到相同性能的情况下,FAST 节省了 5 倍的计算资源。
  • 推理端(劣势):这是 FAST 目前的短板。自回归预测 30-60 个 Token 耗时约 750ms(在 3B 规模模型上),而 π0\pi_0π0​ 的扩散预测只需约 100ms。不过,这可以通过投机采样(Speculative Decoding)或模型剪枝等 LLM 加速技术来优化。
  • 效果:在大规模数据上,FAST 的自回归模型在指令遵循(Language following)方面表现更好,因为它直接复用了预训练语言模型强大的对齐能力。
Q6: 如果我们要在一个新的机械臂上部署 FAST,需要调整哪些超参数?

A6: 核心超参数只有两个:

  1. 量化缩放因子 γ\gammaγ (Scale):控制 DCT 系数保留的精度。γ\gammaγ 越大,精度越高,但 Token 数量也越多(压缩率降低)。
  2. BPE 词表大小:通常设为 1024。
    此外,预处理时的**分位数归一化(Quantile Normalization)**非常重要,它能消除离群动作值的影响,使分词器能处理不同构型机械臂的动作幅值差异。
Q7: 作为算法工程师, FAST 在具身智能领域的长远意义是什么?

A7: FAST 证明了离散化动作空间依然是实现大一统具身智能(Generalist Agent)的有力途径。它解决了自回归模型在高频实时控制上的“信息密度平衡”问题。通过 FAST+ 这种通用分词器,我们可以像处理多语言文本一样处理多形态机器人的动作数据,真正实现具身智能领域的“大模型化”。


Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐