AnchorVLA: Bridging Discrete Decisions and Continuous Trajectories for Vision-Language-Action Planning

AnchorVLA 论文笔记

论文信息

  • 标题:AnchorVLA: Bridging Discrete Decisions and Continuous Trajectories for Vision-Language-Action Planning
  • 作者:刘琦¹²³*、李亚蓓⁴*、王洪松³、张恒⁵、贺磊¹²†(*同等贡献,†通讯作者)
  • 单位:¹清华大学车辆与运载学院 ²清华大学智能绿色车辆与移动全国重点实验室 ³东南大学计算机科学与工程学院 ⁴美团 ⁵东风汽车
  • arXiv:2607.03182(cs.RO)
  • Benchmark:Bench2Drive(CARLA闭环),达到 SOTA 成功率 SR=77.28,驾驶分数 DS=89.92(有竞争力但非最高)

一、核心问题与动机

自动驾驶规划需要把导航意图、交通规则、动态交互、语言指令,转化为可执行的连续轨迹。VLA(视觉-语言-动作)模型被引入以提升长尾泛化、常识推理、语义理解和可解释性,但现有方案存在两种范式,各有缺陷:

  • 规划头方式(Planning-Head-based):LLM/VLM 输出只作为条件/引导信号喂给独立的规划模块。问题:VLA 推理对轨迹空间的约束很弱,语义和动作难以对齐。
  • 全轨迹自回归方式(Full-trajectory Autoregressive):把轨迹离散化成坐标/路点 token,逐个自回归生成。问题:token 序列长、信息密度低,语义-动作对齐差,离散化误差累积,推理慢。

AnchorVLA 的解法:把动作表示从"低层坐标 token"提升为"高层轨迹模式锚点(trajectory-pattern anchor)"。每个锚点编码一个完整的局部运动模式(如车道保持、刹车、让行、转弯、超车),LLM 只需在语义化的锚点空间里推理决策,而不是生成冗长的路点序列。

一个直观的类比:以前要么是"大脑说一句’左转吧’然后手脚瞎猜"(规划头),要么是"大脑一毫米一毫米地指挥手脚"(全轨迹自回归);AnchorVLA 让大脑先说"走这个弯道模式"(锚点),手脚再在这个模式上精细调整。

相关工作定位

  • 端到端自动驾驶:UniAD、VAD、TransFuser++、DriveTransformer 等统一感知/预测/规划;DiffusionDrive、BridgeDrive、GoalFlow 等引入扩散/流匹配做生成式规划。
  • VLA 驾驶规划:ORION、SimLingo、DiffVLA 等属于"规划头"范式;AutoVLA、LinkVLA 属于"全轨迹自回归"范式。LinkVLA 是本文的直接基础和对比对象——AnchorVLA 在 LinkVLA 的 VLM backbone 上,用 DAAR + DARF 替换了其全轨迹自回归生成模块。
  • AnchorVLA 的 VLA backbone 直接复用 SimLingo,backbone 冻结,只替换规划接口。

二、方法

2.1 整体问题分解

不把轨迹分布 $p(\tau\mid x)$ 当作一个整体的连续预测问题,而是拆成"行为级决策 + 局部残差精细化"两阶段:

$$\tau = a_k + r_k \tag{1}$$$$p(\tau \mid x) = \sum_{k=1}^{K} p(a_k \mid x)\, p(r_k \mid x, a_k), \quad r_k = \tau - a_k \tag{2}$$

关键点:锚点不只是额外的条件输入,而是显式定义了残差坐标系和局部行为生成空间的决策接口

2.2 VLA Backbone

  • 编码器:InternVL2-1B(图像 → InternViT-300M;文本 → Qwen2-0.5B-Instruct)
  • GPS 目标点通过 MLP 投影为 token embedding,与语言 token 融合
  • 多模态 token 提供两类输出:
    1. 上下文特征(context features):LLM 中间层/末端隐状态向量,供 DAAR、DARF 使用
    2. CoT 推理 token:LLM 自回归生成的人类可读推理文字
  • 在自回归建模变体下,CoT 文字与锚点 token 处于同一自回归序列中(先文字推理,后锚点 token),因此锚点选择建立在模型已生成的推理内容之上,而不是独立模块
  • 上下文特征的隐状态维度约定取最后一个 token(锚点 token)位置的隐状态,量级对应 Qwen2-0.5B 的隐层大小(896 维,此数值来自 Qwen2-0.5B 已公开的模型结构,论文正文未直接给出具体数字)

2.3 DAAR:决策即锚点表示(Decision-as-Anchor Representation)

目标:把高层 VLA 推理映射到紧凑的轨迹模式锚点空间,用锚点分布估计代替坐标级自回归生成。

码本构建:对训练集轨迹做 K-means 聚类,得到码本 $\mathcal{A}=\{a_k\}_{k=1}^{K}$,K=100。每个锚点 $a_k$ 是聚类中心,本质是一条完整的 T 步时序轨迹(该簇内所有轨迹的"平均形状"),不是单个坐标点:

$$a_k = (a_k^1, \dots, a_k^T),\quad a_k^t \in \mathbb{R}^2$$

锚点-真值距离(逐时间步 L2 距离取平均):

$$d(\tau_{gt}, a_k) = \frac{1}{T}\sum_{t=1}^{T}\left\| \tau_{gt}^t - a_k^t \right\|_2 \tag{3}$$

软标签(注意是 softmax 形式,而非简单的"距离反比例"):不用单一硬标签,在最近的 top-N 邻域 $\mathcal{N}$ 上构造 softmax 形式的软目标:

$$q_k = \frac{\mathbb{I}[k\in\mathcal{N}]\exp\left(-d(\tau_{gt}, a_k)/\gamma\right)}{\sum_{j\in\mathcal{N}} \exp\left(-d(\tau_{gt}, a_j)/\gamma\right)} \tag{4}$$

其中 $\gamma$ 为温度系数。距离越近权重越大,邻域外锚点权重为 0,从而在离散锚点集合上保留轨迹空间的局部连续性,并对 K-means 聚类边界误差更鲁棒。

两种 DAAR 实例化方式(共享同一套软标签监督):

  1. Query-based 建模:可学习 query token 附加到输入序列,经 LLM 编码后取该位置输出特征过 MLP 得到 K 维 logits,softmax 得到锚点分布:

    $$s_k^{qry}=f_{MLP}(h_{qry})_k,\quad p_\theta^{qry}(a_k\mid x)=\text{softmax}(s^{qry})_k$$

    $$\mathcal{L}_{qry} = -\sum_k q_k \log p_\theta^{qry}(a_k\mid x)$$

    不使用 LLM 的 next-token 预测接口,纯粹是基于特征的分类器。

  2. Autoregressive 建模:向 LLM 词表引入 K 个特殊锚点 token $\{y_k\}$,与锚点一一固定映射。训练时插入一个 control token 标记锚点预测开始,其后紧跟最近锚点 token 以保持自回归格式;训练目标从 control token 位置在锚点 token 子集上的 logits 计算,同样用软标签交叉熵:

    $$p_\theta^{ar}(a_k\mid x)=\text{softmax}(s^{ar})_k,\qquad \mathcal{L}_{ar}=-\sum_k q_k\log p_\theta^{ar}(a_k\mid x)$$

    本质是复用 LLM 的 next-token 预测接口,但把输出空间限制在紧凑的锚点词表上。

推理:两种方式都输出 K 维锚点分布,取 top-M(M=6) 个锚点作为候选高层决策,保留行为级多模态性,再送入 DARF 并行生成候选轨迹。

训练/推理不对称性小结:训练阶段学习整个 K 维分布形状(软标签,无"选择"动作);推理阶段才做 top-M 选择。软标签训练让相邻锚点也获得合理概率,是 top-M 候选有意义的前提——若用硬标签训练,分布会退化为近似 one-hot,top-M 退化成 top-1,多候选优势消失。

2.4 DARF:决策锚定残差流(Decision-Anchored Residual Flow)

动机:确定性残差回归会把多个合理的修正方案平均成模糊结果(因为同一锚点+场景下可能存在多种合理的精调方式),因此用 flow matching 建模残差分布,支持多模态精细化。

给定 top-M 候选锚点集合 $\mathcal{I}_M$,对每个候选 $a_k$:

$$\tau_k = a_k + r_k \tag{9}$$

目标残差:$r_{gt}^k = \tau_{gt} - a_k$。采样噪声 $\epsilon\sim\mathcal{N}(0,I)$ 和时间步 $t\in[0,1]$,构造流路径:

$$z_t^k=(1-t)\epsilon+t\,r_{gt}^k,\quad v_k^*=r_{gt}^k-\epsilon,\quad \hat v_k=v_\theta(z_t^k,t,x,a_k) \tag{10}$$

匹配锚点监督:只在 top-M 中离真值最近的锚点 $k^\dagger=\arg\min_{k\in\mathcal{I}_M} d(\tau_{gt},a_k)$ 上计算 flow matching loss(L1):

$$\mathcal{L}_{flow}=\|\hat v_{k^\dagger}-v_{k^\dagger}^*\|_1 \tag{11}$$

置信度分支:为每个候选轨迹预测置信度 $s_k$,评估锚点与驾驶 context 的兼容性,训练时用 softmax 交叉熵监督到匹配锚点:

$$\mathcal{L}_{conf}=-\log\frac{\exp(s_{k^\dagger})}{\sum_{k\in\mathcal{I}_M}\exp(s_k)} \tag{12}$$

推理:从初始噪声 $z_0^k=\epsilon$ 出发,用少量 Euler 步积分速度场,得到候选轨迹后按置信度选最终输出:

$$z_{t+\Delta t}^k=z_t^k+\Delta t\,v_\theta(z_t^k,t,x,a_k),\quad \hat\tau_k=a_k+z_1^k,\quad \hat\tau=\hat\tau_{\arg\max_k s_k} \tag{13}$$

采样步数:主实验中 Flow Step = 2(Table 3、Table 4 明确写出),这是 DARF 相对全轨迹 flow matching 的效率优势来源之一。

Decoder 架构(附录 B):双分支解耦设计,共享输入锚点 $a_k$ 与多模态上下文特征 $H_x$:

  • Velocity 分支:噪声残差 $z_t^k$ 和锚点各自过 2D sine-cosine 位置编码 + MLP;每层用 AdaLN 式的 feature-wise affine modulation 把时间步特征注入隐状态: $$\text{Mod}(h,c)=h\odot(1+s(c))+b(c) \tag{15}$$ 锚点位置特征作为 query 的位置偏置叠加(而非直接替换残差隐状态):$q_l^k=h_l^k+h_{a,k}^{pos}$;query 对 $H_x$ 做 cross-attention 融合视觉/语言/导航信息;过残差连接、LayerNorm、FFN 后输出速度场 $\hat v_l^k\in\mathbb{R}^{T\times2}$。
  • Confidence 分支:完全不使用 $z_t^k$ 和 $t$——因为置信度只评估"锚点与驾驶 context 的兼容性",不评估某个具体噪声/去噪中间状态的质量(借鉴 DiffusionDrive 的候选选择策略)。以锚点特征 $h_{a,k}^{conf}$(独立 MLP 编码)为初始隐状态,对 $H_x$ 做 cross-attention,最终输出标量置信度。

2.5 完整轨迹空间 vs 锚点残差空间的区别

两者都用 flow matching,区别在于生成目标不同:

Full Flow(完整轨迹空间) DARF(锚点残差空间)
流路径 $z_t^{full}=(1-t)\epsilon+t\tau_{gt}$ $z_t^k=(1-t)\epsilon+t\,r_{gt}^k$
目标 绝对坐标轨迹本身 相对锚点的残差
目标分布形态 高度多峰、分散(直行/左转/刹车差异很大) 紧凑、近似单峰、以 0 为中心

论文(Section 3.4 + Figure 3 标注)指出:当锚点提供了良好的粗粒度参考时,残差目标比完整轨迹目标更紧凑,传输路径更短($d_{ResFlow}\le d_{FullFlow}$),因而少量步数的流积分(few-step Euler)更容易逼近准确。Table 4 消融在相同 2 步条件下验证了这一点:DARF(89.92/77.28)明显优于 Full Flow(88.34/72.73)。

追问:能否直接从锚点(而非噪声)出发,流向绝对坐标轨迹以进一步缩短路径?

  • 若从锚点这个确定点出发($z_0=a_k$,无随机性):给定相同锚点和 context,映射退化为确定性函数,本质等价于 Table 4 中的 Deterministic 基线(1 步,直接回归残差),效果最差(86.74/70.45)——因为丢失了随机源,多个合理修正被平均成模糊解。
  • 若从"以锚点为中心的噪声"出发($z_0=a_k+\epsilon$)流向绝对坐标:做变量代换 $\zeta_t=a_k+z_t^k$($a_k$ 对 $t$ 是常数,速度场不变),可证明这在数学上与 DARF 完全等价,只是坐标系表达方式不同,并非新方案。

结论:决定传输路径长短、采样步数多少的关键是"目标分布相对噪声中心的紧凑程度,以及是否保留了随机性",而不是"从哪个点出发";去掉噪声只会退化为效果更差的确定性回归。


三、实验

3.1 设置

  • 训练数据:沿用 SimLingo 设定——专家轨迹、指令条件的 “dream” 轨迹、以及 VQA/驾驶评论/无语言样本等混合语言监督
  • Benchmark:Bench2Drive(基于 CARLA 的闭环基准),报告 Driving Score (DS)、Success Rate (SR)、Efficiency、Comfort、Multi-Ability;SR 衡量任务是否完成,反映闭环可靠性
  • Backbone:复用 SimLingo 的 VLA backbone 并冻结,只修改规划接口
  • 码本/候选数:K-means 聚类 K=100 锚点;推理阶段候选数 M=6
  • 训练:两阶段。① 训练 DAAR 锚点决策模型,15 epoch,8×A100,batch size 16;② 冻结 backbone,训练 DARF 残差生成,15 epoch,4×A100,batch size 32

3.2 主结果(Table 1)

Method VLA Generative Planning DS SR(%) Efficiency Comfort
LinkVLA × 91.01 74.55 255.84 34.62
BridgeDrive × 87.99 74.99 236.49 20.98
AnchorVLA 89.92 77.28 251.14 28.94
  • SR 为所有对比方法中最高(77.28)
  • 相比 LinkVLA:DS 略低(89.92 vs 91.01),SR 更高(77.28 vs 74.55)
  • 相比 BridgeDrive:DS、SR 均更高
  • Comfort(28.94)低于部分基线,作者认为是生成式精细化灵活性与轨迹平滑度之间的权衡,未来可引入平滑约束改进

3.3 Multi-Ability 评估(Table 2)

Method Merging Overtake Brake Give-Way Traffic-Sign Mean
LinkVLA 60.00 80.00 93.33 50.00 83.68 73.40
BridgeDrive 69.92 66.67 90.00 50.00 89.47 73.15
AnchorVLA 65.00 81.11 90.00 50.00 85.00 74.22

均分最高(74.22),Overtake 最佳,但 Merging/Traffic-Sign 不敌 BridgeDrive,Brake 不敌 LinkVLA。

3.4 消融实验

决策建模方式(Table 3)——核心结论:性能提升不仅来自"使用锚点"本身,更来自把锚点预测紧密建模在 LLM token 空间里:

Method Extra Latency DS SR(%)
LinkVLA-AR(全轨迹自回归基线,无锚点) 361 ms 89.57 73.18
None(无锚点决策,直接把所有锚点扔进DARF) 87.49 70.91
Query Based 33 ms 88.67 73.81
Autoregressive 64 ms 89.92 77.28

Autoregressive 相比 Query-based,把锚点表示为 LLM 动作词表中的 token,让高层决策和锚点预测在同一 token 空间建模,SR 从 73.81 提升到 77.28;同时相比全轨迹自回归基线,额外延迟从 361ms 降到 64ms。

Flow Matching 形式(Table 4)——同为 Autoregressive 决策方法:

Flow Matching Method Flow Step DS SR(%)
Deterministic($t=0,z_t=0$ 直接回归) 1 86.74 70.45
Full Flow(完整轨迹空间) 2 88.34 72.73
DARF(锚点残差空间) 2 89.92 77.28

同样 2 步条件下,锚点残差空间显著优于完整轨迹空间,验证了残差空间紧凑性带来的收益(见 2.5 节)。

导航模态(Table 5):GPS 目标点(89.92/77.28)与导航文字指令(90.26/76.82)效果相当,说明 AnchorVLA 对不同形式的导航输入较为鲁棒。


四、失败案例分析(附录 A)

主要失败模式是锚点预测错误导致的误差传播:由于 DARF 在锚点定义的残差空间内生成轨迹,倾向于"精调"而非"推翻"已选定的行为,因此当预测锚点偏离路线或与语言指令不一致时,最终轨迹也会继承该错误行为;此外即便锚点大致正确,DARF 残差精调也可能引入局部几何偏差。作者认为改进方向是引入反思机制(reflective anchor selection),在 DARF 精细化之前重新评估语言推理、导航意图、场景 context 与候选锚点之间的一致性,修正潜在的错误锚点预测。


五、结论

AnchorVLA 是一个分层的决策锚定 VLA 规划框架,用"轨迹模式锚点"作为显式接口,桥接高层 VLA 推理与连续轨迹生成:

  • DAAR 负责行为级决策——把语义推理映射为紧凑锚点分布,两种实例化中"自回归建模"(锚点预测复用 LLM next-token 接口)显著优于"query-based 建模",说明关键收益来自决策与预测的紧密耦合,而非锚点概念本身
  • DARF 负责连续执行——在锚点定义的残差子空间上做 flow matching,仅需 2 步 Euler 积分即可生成精细轨迹,残差空间的紧凑性是"少步数也能生成好"的根本原因

在 Bench2Drive 闭环基准上取得 SOTA 成功率(SR=77.28)与有竞争力的驾驶分数(DS=89.92),同时把自回归推理的额外延迟从全轨迹自回归基线的 361ms 压缩到约 64ms,验证了轨迹模式锚点作为语言级推理与连续轨迹规划接口的有效性。

本工作受国家重点研发计划(2024YFB2505500)资助。


相关笔记

  • ResAD - 归一化残差轨迹建模 —— 同样采用"残差建模"思想,但 ResAD 的参考系是物理惯性先验轨迹,AnchorVLA 的参考系是 K-means 聚类得到的离散行为锚点;两者都验证了"缩小学习目标分布"对轨迹生成的收益
  • DiffusionDriveV2 论文阅读笔记 —— AnchorVLA 的置信度候选选择策略借鉴自 DiffusionDrive 系列;两者同属"锚点/截断+生成式精细化"路线
  • DFP - 扩散强迫规划与历史退火 —— 同为少步数生成式规划的代表工作,可对比扩散强迫与锚点残差流两种少步数策略