跳转至

ECoSim: Data Efficient Fine-Tuning for Controllable Traffic Simulation

会议: ECCV2026
论文: https://eccv.ecva.net/virtual/2026/poster/3709
PDF: https://media.eventhosts.cc/Conferences/ECCV2026/pdfs/2701.pdf
项目: https://ecosim-web.github.io/
领域: 自动驾驶
关键词: 可控交通仿真、参数高效微调、特征调制、行为潜变量、上下文检索

一句话总结

ECoSim 冻结已有交通生成模型,用恒等初始化的 FiLM 适配器和闭环训练接入草图、行为潜变量或文本,在仅用 1% 配对控制数据时将 SMART 草图控制的 mADE 从 1.5044 降至 0.2561,并通过上下文匹配把行为迁移到新的交通场景。

研究背景与动机

交通仿真不仅要生成“像真实驾驶”的轨迹,还要能指定某辆车转弯、让行或改变速度,才能针对同一环境测试不同未来。VBD、SMART 一类模型已经学到相当丰富的交通分布,但无条件采样不保证产生测试者想要的行为。推理时施加梯度或约束可以引导生成,却需要设计控制目标并增加采样开销;ProSim 等直接条件生成方法则依赖较多配对控制数据,为新模态重新训练的成本较高。

这里真正困难的不是把一个文本向量接进网络,而是在低数据条件下改变指定车辆的行为,同时保住原模型的多车交互能力。交通模型会把自己上一轮的预测作为下一轮输入,初始阶段很小的特征扰动都可能沿闭环累积。另一方面,“遵循条件”也不等于“产生合理场景”:同一个转弯行为放进不匹配的道路环境,即使轨迹跟得很紧,也可能破坏驾驶质量。

本文把已经学到的驾驶先验与新增控制能力分开处理:前者保留在冻结骨干中,后者交给小型适配器,并在迁移行为前检查场景是否兼容。核心 idea:用接近恒等映射的特征调制逐步学会控制,用闭环训练适应自身预测误差,再用上下文检索约束行为迁移的适用环境,而不是为每种控制信号重新学习交通分布。

方法详解

整体框架

输入是地图、多车状态及指定车辆的控制信号,输出是持续交互的多车未来轨迹。直接控制时,用户提供草图或文本;行为迁移时,先从相似交通环境中选取控制来源,再将其草图、行为潜变量或文本送入模态编码器。编码结果通过 FiLM 改变冻结解码器的中间特征,适配器在闭环轨迹生成过程中学习,不更新交通骨干。

系统包含上下文匹配迁移、多模态控制表示、恒等初始化调制与闭环适配训练四个相互衔接的设计。下图把可选的迁移入口与直接输入画在同一流程中;最后一个训练节点表示训练方式,部署时使用已经训练好的适配器滚动生成,不需要访问真实未来或在线更新参数。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["目标交通场景"] --> B["上下文匹配迁移"]
    B --> C["多模态控制表示"]
    U["直接输入草图或文本"] --> C
    C --> D["恒等初始化调制"]
    D --> E["闭环适配训练"]
    E --> F["受控多车轨迹"]

关键设计

1. 上下文匹配迁移:先选择能容纳目标行为的环境

已有轨迹中的行为不能不加区分地移植到任意车辆。ECoSim 根据目标车辆及其场景生成上下文嵌入,在数据库中寻找嵌入相似的车辆;检索流程还包含保证动态可行性的启发式过滤,再按相似度选择候选。匹配成功后,迁移的是行为控制信号,目标场景中的周围车辆仍按原有模型交互生成,而不是把整个来源场景机械复制过来。正文没有给出过滤阈值及检索编码器的完整配置,因此这里不补写这些实现细节。

这个入口服务于两种使用方式。反事实生成检索不同驾驶意图,在同一场景内探索替代未来;长尾生成则先利用 SMART 对轨迹的似然筛出最难预测的场景,再把其中的行为潜变量转移到兼容环境。这里“难预测”只是筛选稀有候选的代理指标,不能直接等同于危险程度。上下文匹配也只是提高兼容性,不构成几何可行性或安全性的形式保证。

2. 多模态控制表示:把空间路径、交互行为和语言意图接到同一接口

每种模态都映射到 256 维控制空间。草图是在车辆局部坐标系中表示的稀疏未来路点,通过轻量时序编码器汇聚成控制向量;它提供空间走向,但不要求环境中的其他车辆照抄录制轨迹。文本由带 LoRA 的 DistilBERT 编码,取其 [CLS] 表示并线性投影,语言编码器的 LoRA 参数与闭环控制目标共同优化,使抽象指令与实际运动逐步建立联系。这里没有宣称用文本恢复唯一的精确轨迹。

第三种表示由单独训练的 BehaviorVAE 提供。它同时读取场景内车辆的未来运动与环境上下文,为各车辆推断高斯潜变量;联合编码让潜变量携带相对几何和交互信息,而不只是一辆车的独立运动曲线。轨迹重建迫使编码保留行为信息,KL 正则化和退火则约束潜空间、促进平滑变化。训练好的编码器可以从已有轨迹一次导出配对潜变量,正文也描述了从先验采样新行为的可能性;本文的行为迁移实验主要使用检索得到的行为编码。只有被指定控制的车辆接收相应向量,其余车辆使用零向量,但零条件本身不应被解读成训练后严格不改变任何中间特征的证明。

3. 恒等初始化调制:先保留驾驶先验,再逐步改变解码特征

控制向量不直接替换骨干的轨迹输出,而是为每一层解码特征预测逐通道缩放和偏移。FiLM 的仿射作用可写为:

\[ \widetilde{\mathbf h}^{(l)}=\boldsymbol\gamma^{(l)}(z_i)\odot\mathbf h^{(l)}+\boldsymbol\beta^{(l)}(z_i). \]

其中,\(z_i\) 是指定车辆的控制表示,\(\mathbf h^{(l)}\) 是第 \(l\) 层特征,两个轻量 MLP 输出与特征通道数相同的调制参数。该式按正文对仿射变换的说明整理,缓存中的公式排版存在符号丢失。适配器以 \(\boldsymbol\gamma\approx1\)、\(\boldsymbol\beta\approx0\) 初始化,使训练起点接近原模型;随后才逐步学会放大、抑制或平移与目标行为相关的特征。

这比单纯“少训练一些参数”多了一层闭环稳定性考虑:如果随机条件分支一开始就大幅改变轨迹,下一步输入也会偏离骨干熟悉的状态,容易使错误连续放大。本文在 VBD 扩散模型与 SMART 自回归模型的每个 Transformer 解码层输出处使用同一类调制,骨干保持冻结,只学习控制编码器及 FiLM。虽然支持不同生成架构,但并不意味着两种骨干共享全部训练过程或一组适配器权重。

4. 闭环适配训练:让控制器在自身生成的历史中纠偏

仅在真实历史上监督下一步,不能覆盖部署时模型不断消费自身预测的情况。ECoSim 在闭环展开过程中施加条件、生成轨迹,再用真实未来监督适配器,使其既学会听从控制,又学会在先前预测已发生偏差时继续生成。它复用了相应骨干的闭环训练机制,而不是把所有架构硬改成统一的优化算法。

对于扩散骨干,每次重规划先生成 8 条候选联合轨迹,按与真实轨迹的匹配损失选择最接近的一条,用选中轨迹计算适配器训练损失;这是训练时的候选选择,不是部署时利用真实未来挑选答案。自回归骨干则在动态生成的多车历史下,以真实状态 token 为目标优化交叉熵,并通过采样偏置提高与条件一致的轨迹被采到的机会。这样训练面对的是闭环中的误差分布,而不只是开环预测精度。

一个完整示例

可以把一次反事实仿真理解为:同一辆目标车、同一份道路与周围车辆状态,比较“继续当前运动”和“执行另一种意图”两种未来。以下只是按论文机制展开的解释,不是作者另报的一条定量案例。论文的反事实实验选择 250 辆查询车辆,每辆最多取 4 种上下文匹配分数最高的不同意图;这样得到的不是随机换一条路径,而是先从数据中寻找当前环境较可能容纳的行为来源。来源车辆提供控制信号,目标车辆和周围环境仍留在原场景里,因此检索匹配解决的是“这种行为适不适合在这里尝试”,生成器解决的是“在这里具体怎样执行”。

假设选择的是一种转弯意图。草图接口把来源轨迹压缩成局部坐标中的稀疏路点,让模型知道要朝哪个方向运动;潜变量接口则先由 BehaviorVAE 联合编码来源运动和场景,把该车辆的交互模式压缩成行为表示。两种输入不是必须同时提供的两个阶段,而是不同的可选控制模态。模型随后为指定车辆生成 FiLM 参数,在冻结解码器中调节相应表示;未被指定的车辆不接收这条行为条件,却仍会随着共同生成的场景历史继续运动。最终效果因此既取决于控制信号,也取决于目标环境中的交互,而不能理解为复制来源轨迹的坐标。

随着仿真前进,当前状态逐渐来自模型自己的预测,而不再始终是真实记录。如果前一步让目标车稍慢或位置偏移,后一步必须在这个新状态上继续实现意图;闭环适配训练正是让适配器见到这种情况。动态条件表示每一步重新编码控制信号,静态表示则一次编码后结合时间嵌入,两者在跟踪误差上的区别由消融测量,而不是预设动态一定在所有指标上更好。评估时应把“目标是否跟随成功”与“生成过程是否合理”分开:Control ADE 检查是否贴近迁移行为,动作成功率检查高层意图,碰撞及驶出道路指标检查代价。即使第一项很好,也不能据此断言后两项可靠,这正是原文反事实结果需要保留安全边界的原因。

损失函数 / 训练策略

BehaviorVAE 与控制适配器分开训练,前者使用 10,000 个 WOMD 训练场景,以条件变分目标结合轨迹重建和 KL 正则化,并采用 KL 退火。由于缓存中对应公式损坏,且解码条件的记号不完整,此处只保留原文明确说明的目标组成,不补写一个貌似精确的 ELBO。

主实验先把 VBD 与 SMART 额外训练成闭环骨干 VBD-CL、SMART-tiny-CLSFT,再冻结它们训练适配器。表中的 1% 指配对控制微调数据比例,不涵盖骨干预训练、闭环骨干准备或 BehaviorVAE 的训练成本。数据规模实验覆盖 50 至 50,000 个场景,即约 0.01% 至 10%;其 100% 参考线是使用相同控制适配器、同时以 LoRA 微调骨干的参考模型,不是完全零额外代价的比较对象。

实验关键数据

主实验

评测使用 WOMD,并遵循 WOSAC 闭环仿真协议。Meta 越高表示多维驾驶行为的分布真实性越好;mADE 是候选轨迹对目标的最小平均位移误差,越低越好。相对改善按 \(\mathrm{Gain}=(\mathrm{mADE}_{base}-\mathrm{mADE}_{ctrl})/\mathrm{mADE}_{base}\times100\%\) 计算,不是百分点增益。

标准控制评测中的草图和潜变量由留出集真实未来提取,文本来自 ProSim-Instruct-520k。它测量“给定目标后能否跟随”,不能直接证明模型无需目标信息就能预测真实未来;检索迁移实验才对应不依赖目标车辆真实未来的控制用途。以下为原文表 1,各条件模型都使用 1% 配对控制数据,增益相对各自无条件骨干计算。

骨干 控制方式 Meta ↑ mADE ↓ 相对 mADE 改善 ↑
VBD-CL 无条件 0.7186 2.7223 —
VBD-CL 草图 0.7432 0.9645 64.56%
VBD-CL 潜变量 0.7440 0.9221 66.12%
VBD-CL 文本 0.7306 2.0005 26.51%
SMART-tiny-CLSFT 无条件 0.7728 1.5044 —
SMART-tiny-CLSFT 草图 0.7984 0.2561 82.97%
SMART-tiny-CLSFT 潜变量 0.7912 0.3238 78.48%
SMART-tiny-CLSFT 文本 0.7767 1.3537 10.02%

同一 ProSim 架构的补充对照中,原始模型使用 100% 配对数据,草图 mADE 为 1.099;FiLM 仅用 1% 配对数据得到 1.130,是接近而非胜出。文本 mADE 则为 2.330 对 1.824。不过原始 ProSim 使用官方 checkpoint,FiLM 使用重新训练的无条件骨干,各自无条件 mADE 分别为 2.679 和 2.641,因此“同骨干”指同架构,不能当作完全相同权重起点的严格对照。

消融实验

原文表 5 比较控制表示的更新时间:Dynamic 每一步重新编码,Static 编码一次并加入时间嵌入。该节默认使用自回归骨干与约 5,000 个训练场景;以下数值属于独立消融,不与主表混为同一配置。

模态 配置 Meta ↑ mADE ↓
草图 Dynamic 0.7799 0.3970
草图 Static 0.7719 1.0598
潜变量 Dynamic 0.7848 0.3151
潜变量 Static 0.7880 0.4842

动态编码两种模态的 mADE 都更低,但潜变量静态编码的 Meta 反而略高,说明更精确跟随与分布真实性不完全同向。调制方式消融还报告:加法注入在 50,000 步、单张 RTX 4090 约 9 小时后仍未收敛,恒等初始化 FiLM 在不足 12,000 步、约 2.2 小时内已取得更好的 mADE;这是该消融设置下的训练观察,不是通用加速保证。

下表为原文表 4 的长尾迁移分析:先取 SMART 最难预测的 0.1% 场景,再整理 8 类行为,每类检索前 30 个兼容候选。Control ADE 是生成轨迹相对迁移目标的平均位移误差;Traj ADE 是相对无条件生成轨迹的偏离幅度,不是越低越好的跟踪误差。

上下文选择 Control ADE ↓ Traj ADE 碰撞率 ↓ 驶出道路率 ↓ PDMScore ↑
随机 3.8723 16.33 0.3175 0.1553 39.06
匹配 1.6994 11.30 0.1574 0.0661 65.27

碰撞与驶出道路指标保留原文的小数率,PDMScore 是论文改造后的闭环驾驶质量评分;当前正文缓存没有其完整聚合定义,不应默认与其他 PDM 基准绝对可比。

关键发现

  • 空间控制强于抽象文本:SMART 草图的相对改善为 82.97%,文本只有 10.02%。这说明控制信息的具体程度很关键,而不是所有模态都达到同样的轨迹约束能力。
  • 潜变量的数据效率不等于零成本:规模实验描述其在 0.01% 配对微调数据时已接近全数据参考,但 BehaviorVAE 另用 10,000 个场景训练,不能省略这笔前置成本。
  • 上下文匹配明显改善长尾迁移,但匹配后的碰撞率仍为 0.1574。它降低失败频率,并没有消除强行执行不适合行为的问题。
  • 反事实表 3 的安全表述需要保留边界:无条件碰撞率为 0.0563,草图和潜变量控制分别为 0.0836、0.0774,文本为 0.0520。尽管其 PDMScore 与覆盖度提高,也不能概括为所有控制方式均不损害安全。

亮点与洞察

  • 恒等初始化的价值来自闭环放大效应,而不只是优化器更容易收敛。先保持熟悉的驾驶分布,再逐步学习控制,能避免随机控制分支立即破坏后续输入状态。
  • 行为潜变量把已有运动数据转换成可复用的控制监督,降低对逐条语言标注的依赖。它保留交互上下文,也解释了为什么应把潜变量学习成本与适配器数据效率分开讨论。
  • 论文区分“跟随录制目标”和“生成替代未来”两种评测目的。反事实场景不再以贴近真实未来为唯一目标,而结合行为成功、覆盖度及驾驶质量判断生成是否有用。

局限与展望

  • 作者明确承认,不可行或含糊条件可能使模型优先满足控制、牺牲交互合理性。作者提出强化学习微调或推理引导作为后续方向,但当前结果不提供安全约束保证。
  • 实验主要控制单个指定车辆。虽然生成器处理多车交互,结果不能直接外推为协调多辆受控车辆的联合控制能力。
  • 反事实 Coverage 按轨迹核密度估计超过阈值的网格数量衡量空间覆盖,行为成功率统计是否执行指定高层动作;它们并不覆盖所有驾驶合理性。正文没有展开完整阈值、动作判定和闭环 PDMScore 细节,跨实现复现仍需补充材料。
  • 当前本地缓存包含完整正文和参考文献,但没有正文引用的补充材料;FiLM、VAE 公式有提取损坏。因此本文不猜测补充实现、不从曲线读出未经列明的精确数值,也不把定性稳定性叙述替代实际安全指标。

相关工作与启发

  • 对比 ProSim:ProSim 直接训练可提示的闭环生成器,ECoSim 强调冻结已有驾驶先验后增加条件接口。配对数据需求较低,但已有模型训练与条件编码器的成本并未消失。
  • 对比 ControlNet 与 LoRA:ECoSim 借鉴保持初始函数近似不变的控制适配思路,实际采用逐层 FiLM,并针对交通生成做闭环训练。它不是提出一种新的通用低秩分解方法。
  • 对比 RealGen:两者都利用检索支持场景控制;ECoSim 把兼容上下文检索与草图、行为潜变量、文本等控制接口结合,并在闭环生成中验证迁移。关键是“检索什么环境后再控制”,而不只是找到一条相似轨迹。

评分

  • 新颖性: 3.5/5 — FiLM 和条件 VAE 本身成熟,价值主要在跨交通骨干的闭环适配及行为迁移组合。
  • 实验充分度: 4/5 — 覆盖多模态、两类主骨干、数据规模与迁移消融,但协调多车控制及安全边界验证仍有限。
  • 写作质量: 3.5/5 — 主流程清晰,部分安全措辞比表中数值更乐观,补充实现未包含在当前缓存中。
  • 价值: 4/5 — 对复用已有交通模型、低配对数据增加控制入口具有明确工程参考价值。