☰
Drive-JEPA:用JEPA世界模型重构端到端自动驾驶的预测与规划
2026/10/21 7:06:52 网站建设 项目流程

最近圈子里讨论热度最高的一个智驾方案,就是某新势力品牌放出的Drive-JEPA。严格来说,它并不是又一个“多传感器融合+端到端规划”的缝合怪,而是把JEPA这套自监督表征学习思路,正式拉进了世界模型的训练体系里,并在下游直接接规划和控制。这篇我想结合自己在这条赛道上的观察和工程经验,把它拆开讲清楚:Drive-JEPA到底在哪个环节动了刀子、JEPA为什么比传统生成式世界模型更适合驾驶场景、它靠什么冲到SOTA,以及真正量产落地还差哪几口气。

如果你是做智驾算法、端到端模型或者自动驾驶仿真相关工作的,这篇文章可以帮你快速建立对“JEPA类世界模型+端到端驾驶”的技术框架认知;如果你只是对前沿AI方案感兴趣,我也尽量把概念掰开揉碎,保证你能看懂整个逻辑。

1. 为什么端到端驾驶的头号难题是“预测”而不是“感知”

1.1 感知已经“够用”,但规划总是“短视”

先说一个反直觉的行业观察:很多团队把资源堆在感知上,觉得把红绿灯、车道线、行人识别做到99%准确,驾驶就稳了。但真正跑路测的时候会发现,感知错误只占事故原因的一小部分,更常见的问题是“看到但没料到”——比如前车减速了,自车也检测到刹车灯亮了,但在接下来1到2秒里,规划器给出的轨迹要么太激进、要么太保守,完全不像一个老司机。

为什么?因为传统模块化方案里,感知、预测、规划、控制是四个独立模块。感知模块输出物体列表和车道线,预测模块基于这些输出一个“未来3秒轨迹”,规划模块再根据预测轨迹生成避障路径。每一层都做一次信息压缩,预测模块一旦猜错,规划就是错的,而且这个错误会沿着链路逐级放大。端到端神经网络出现后,梯度可以从最终方向盘的输出一路传回输入图像,省掉了模块间的“翻译损耗”,但大多数端到端方案本质上仍然是行为克隆(Imitation Learning):给它海量人类驾驶数据,让它学习“看到什么就怎么打方向盘”。

行为克隆的最大短板是缺乏反事实推演能力。它学会了“看到前车刹车就减速”,但没学会“如果前车突然急刹,我往左侧变道是否安全”这种需要脑内预演多步的结果。换句话说,当前这代端到端模型更像是“记忆了很多场景的熟练工”,而还不是“能提前想象路况演变的司机”。

1.2 世界模型为什么是“下一个台阶”

要补上这块短板,行业里给出的答案是世界模型(World Model)。世界模型这个概念在强化学习里早就有了,核心思想是让智能体在内部学一个“环境模拟器”:给定当前状态和动作,它可以预测未来状态;智能体就可以在脑内推演很多条可能的行动路径,选一条未来结果最好的去执行。

放到自动驾驶里,世界模型的价值就更清楚了。人开车的时候,真正厉害的是“预判”:

  • 路边停着的车突然亮起倒车灯,我减速并做好鸣笛准备;
  • 旁边车道的大车压线往我这边挤,我判断它会继续挤,所以让出空间;
  • 绿灯刚亮、行人还在斑马线上,我不是直接冲过去,而是预测他可能会继续走。

这些行为的共同点是:都在对未来做假设,并评估不同假设下的代价。世界模型就是试图在模型内部建立这种“对未来的模拟能力”。Drive-JEPA的取名逻辑很直白——它这套框架里,做未来预测的核心引擎就是JEPA,预测的对象不是图像像素,而是经过编码后的抽象世界状态。

1.3 从“感知-预测-规划”深度耦合的角度重新看

这里要补充一个我个人的判断:世界模型不是要替代掉感知和规划,而是把“预测”从辅助模块提升为核心主链路。传统方案里预测模块的输出是离散的、规则化的(比如目标物的轨迹点列表),而世界模型的预测是连续特征层面的“状态推演”。前者适合做规则判定,后者适合做端到端的梯度传递,也能跟规划器在同一个神经网络里联合优化。

这也是Drive-JEPA这类框架看起来并不“酷炫”但实际很深的原因:它没有取消传感器、也没有换成雷达点云输入,它重新设计的是信息在系统内部流动的方式——从图像到潜在特征,从潜在特征到未来状态,从未来状态到驾驶决策。这个链条一旦在数学上是可微的,整个系统就能做好“为了未来更安全,现在应该做什么”的全局优化。

2. JEPA进驾驶:从“像素世界模型”到“抽象世界模型”的范式切换

2.1 生成式世界模型的代价:为树叶和玻璃反光浪费算力

在JEPA出来之前,做世界模型的主流思路是生成式预测:让模型学会“看到前几帧,生成未来几帧的图像”。深度学习的做法一般有两种:

  • 用变分自编码器(VAE)之类的框架学习图像的隐空间,然后在这个空间里做预测,最后把预测出的隐向量解码成画面;
  • 用扩散模型或生成对抗网络直接预测未来帧,比如“给定连续3帧,生成下一秒的画面”。

这些方案在游戏仿真、视频预测里效果不错,但放进自动驾驶里有个致命问题:画面里的像素不是等价的。一张道路图像里,数百万像素中的绝大多数是天空、路面纹理、路边建筑、树叶、玻璃反光这类对驾驶决策完全没有直接影响的内容。可生成式模型去预测未来帧时,为了把“未来画面”重建到一个看起来像真的程度,必须把大量建模能力花在这些无关细节上。

打个比方,这就像你让一个学生准备一场关于交通规则的口试,他却把大量时间用来背诵考场墙上挂钟的分针刻度。不是没用,但性价比极低。在车载计算平台上,算力和功耗都是有严格预算的,把算力耗在做像素级重建,必然挤占真正用于决策的神经网络容量。

2.2 JEPA的底层逻辑:只预测“该预测的”

JEPA,全称是联合嵌入预测架构(Joint Embedding Predictive Architecture),它的核心思路可以用一句话概括:不在像素空间做预测,而是在经过语义抽象的嵌入空间做预测。

具体来说,JEPA的流程是:

  1. 把输入观测(比如一段道路视频)通过编码器(Encoder)压缩成一个高维向量,这个向量叫嵌入表示,里面保存的是“这帧画面里对决策有用的语义信息”;
  2. 把背景上下文(context,比如路面结构、导航路径、自车状态)通过另一个编码器编码成上下文表示;
  3. 用预测器从上下文表示出发,去预测目标帧的嵌入表示;
  4. 用损失函数衡量“预测出的嵌入”和“真实目标嵌入”之间的距离,训练整个网络。

最妙的地方在第4步:JEPA并不要求预测器和解码器“还原出完整的未来画面”,它只要求在嵌入空间里预测得准。那个嵌入空间是模型自己学出来的,模型可以选择保留“前方有行人横穿”这个关键语义,丢掉“行人的衣服是红色还是蓝色”这种次要信息。换句话说,JEPA让模型自己找到了“什么信息值得预测、什么信息可以忽略”的边界。

为了画面对比,我把生成式预测和JEPA预测的区别做成一个表:

对比维度生成式世界模型JEPA世界模型
预测目标未来图像像素/帧未来抽象嵌入表示
重建压力必须还原像素级细节只追求语义表征一致
算力消耗高(解码器反复参与)较低(无逐像素重建)
对无关纹理敏感度高低
下游决策可用性需要再套一层感知网络可直接与规划头对接

2.3 表示坍缩问题:JEPA落地时躲不开的工程坑

这里必须说一个JEPA家族都会遇到的问题:表示坍缩(Representation Collapse)。如果JEPA的损失函数只简单定义成“让预测嵌入和目标嵌入距离最近”,模型可能偷懒到把一切输入都编码成同一个常数向量。这样预测当然“零误差”,但嵌入空间里没有任何有价值的信息,模型彻底废了。

业内应对这个问题的经典思路主要有两种:

  • 非对称结构:让目标分支(target branch)使用更“强”的表示,比如用滑动平均的历史权重生成目标表示,让预测分支永远追不上目标分支的最新变化;
  • 信息保持正则:在损失函数之外,额外约束嵌入表示自身的信息量,比如保证一批样本里不同嵌入的方差不能太小、各维度的协方差要对角化,防止表示退化成一维信息。

I-JEPA也就是Yann LeCun那套自监督方案,被大家认可的一大原因是它通过“多块掩码预测(Multi-block Masking)”很自然地解决了坍缩问题:让编码器只看到可见块怎么办?它对可见块和掩码块分别走编码器,让预测器去预测被掩码块的表示;可见块的数量和位置是随机选的,迫使编码器学到的表征具有足够丰富的上下文语义才能准确预测被掩码部分。

具体到Drive-JEPA这种自动驾驶框架里,我倾向于认为它真正做扎实的地方,就是把这套防坍缩机制搬进了车辆感知的时序设定里——用“当前观测去预测未来状态的嵌入”,而不是“当前观测去重建未来的一张图”。这一步从原理上就规避了生成式模型的算力黑洞,而且保留下来的语义空间天然适合向下游决策模块直接输出。

3. Drive-JEPA的组件拆解:特征抽象、状态预测与规划是怎么串起来的

3.1 一条数据流链路:从传感器到方向盘

现在我把Drive-JEPA全链路按数据流拆开看。公开资料里没有给出完整结构图,但从“JEPA世界模型+端到端框架”这两点可以推断出一个合理的工程实现框架,我来按最常见的落地形态描述:

第一层是感知编码器(Encoder)。输入是多路相机图像,也可能包含毫米波雷达或激光雷达点云。这层的作用是把“原始观测”压缩成当前时刻的隐状态向量,记为 $s_t$。这个 $s_t$ 并不是传统BEV特征图那种“栅格化占用”,而是被拉平并融合了时序信息的语义向量。它可以理解为“此刻道路环境的紧凑摘要”。

第二层是上下文编码器(Context Encoder)。它接收的信息包括:自车历史轨迹、当前导航路径、车道结构、交通信号状态、以及前一个时刻的隐状态。输出成为一个上下文向量 $c_t$。一个很关键的工程细节是:上下文编码器应该有多少历史信息进入?太短会丢失意图,太长会让计算量膨胀。合理的做法是采用滑动窗口,比如过去2秒内的轨迹点采样,加上当前帧特征。

第三层是预测器(Predictor),这是Drive-JEPA最有区分度的部分。它从 $c_t$ 出发,直接预测未来几个关键时间步的隐状态,比如 $s_{t+100ms}$、$s_{t+300ms}$、$s_{t+1s}$。注意,这里的预测不是输出轨迹线,而是在抽象特征空间里的“预期状态”。这个“预期状态”可以理解成模型对“未来路况应该长什么样”的猜测。

第四层是规划与决策头(Planner)。它接收预测器输出的未来隐状态序列,结合当前车体状态和目标,在隐空间里做一次“选择”。输出包括:目标轨迹点、期望横纵向加速度、转向角等。到这里,从图像到方向盘的整个链路就闭环了。

这个流程在训练时可以看作一个多头任务:

  • JEPA分支负责自监督学习“预测未来隐状态”;
  • 规划分支负责监督学习“在给定预测状态序列下,选择与人类驾驶轨迹一致的决策”。

推理时,两个分支被拼成一个前向网络,一次前向传播直接得到控制量,天然满足端到端“一个模型、一次推理”的要求。这也是它比“模块化堆叠”更优的地方——整个链路可微,更新时是全局联合优化。

3.2 在隐空间里怎么做“规划”:想象之后的选择

真正让我觉得JEPA这类世界模型有趣的,是“规划”这一步。传统规划器在结构化世界里做:定义一大堆几何约束(避障、动力学边界、交规约束),然后用搜索或优化算法找轨迹。端到端学习式规划器则直接把“隐状态”映射到动作。

有了世界模型之后,规划可以在想象出来的多条未来分支里做挑选。我举一个具体场景来说明:左侧车道有一辆慢车,自车正准备变道超车,但右边盲区可能有一辆电瓶车。普通行为克隆模型会直接学人类在这类场景下的平均行为,可能会变道也可能不变道,结果模型给出的动作是“中间态”——既没变又没完全刹住,特别别扭。

而世界模型端到端框架会这样做:先预测出“变道”这一动作导致的未来隐状态序列“车道线、周围车速、盲区风险”;再预测“保持车道”这一动作的未来状态;规划器用一个人为设定的奖励/成本函数,比较两条想象轨迹的总代价,选择代价小的那条。这个思考过程虽然在实际部署时并不是真的把每条分支都展开(会做并行推断或深度网络直接粗选),但训练目标引导模型内部隐式地具备“想象对比”的能力。

从工程角度看,这种设计的优势是开放性问题变得可优化:你不再需要手写“什么时候可以变道”的判定规则,只需要定义好代价函数(安全距离权重、效率权重、舒适性权重),系统会在隐空间里学会权衡。

3.3 和“BEV特征+学习规划”那类方案的核心分歧

行业内前几年把“端到端”高地和“BEV(鸟瞰视角特征)”绑定得比较紧,大量方案是:把多相机图像转成BEV特征,再在这个特征图上做动态物体检测、轨迹预测和规划。这类方案在传感器融合上很成熟,但有一个边界:BEV特征图本质上仍然是一个“栅格空间”,网络很难处理“跨时间步的抽象演化”。比如一个行人在未来3秒可能走出一条弧线,BEV栅格图只能逐帧地预测,不能直接表示“这个人下一步会向左偏转”这种高层因果。

Drive-JEPA这类JEPA框架跟BEV方案最本质的差别,在于系统内是否存在一个显式的“未来状态预测器”,以及预测器是否作用于抽象语义空间而非几何栅格。BEV方案预测的是“未来物体位置”,JEPA方案预测的是“未来场景语义状态”。前者信息具体但琐碎,后者信息抽象但高维。再到下游,BEV方案往往需要再接一个规则优化器或轻量网络来生成轨迹,而JEPA框架可以直接把预测头和规划头焊在一个图里。

这不是说BEV方案没用。我的判断是:如果你要降低传感器融合成本、利用已有仿真验证工具,BEV仍是稳健选择;如果你目标是端到端SOTA、愿意承受训练复杂度和不确定性,JEPA路线的上限更高。

4. 训练管线与数据闭环:无标注预训练为什么能省掉海量标注

4.1 两阶段训练:自监督预训练让“海量数据”不再只是摆设

自动驾驶数据量极大,一家头部车企的新能源车队一天就能回传千万公里的驾驶视频。但传统做法的瓶颈是标注成本:一帧图像要标出所有物体框、车道线、道路拓扑,人力成本非常高,自动标注工具又需要人校验。因此大量天然数据只能躺在服务器里,利用率其实很低。

Drive-JEPA路线最大的工程红利是自监督预训练直接消耗未标注数据。JEPA的预测任务不需要任何人工标注,模型只需要“拿前几秒预测后几秒”,就能从海量驾驶视频里学到场景演化规律。这个阶段学到的东西是通用的、与驾驶动作无关的:路口如何衔接、车辆如何交互、不同天气如何影响可见性。

第二阶段才是下游对齐:用少部分人工标注的数据(目标轨迹、驾驶行为标签、操作检测)在预训练好的表征上做微调,让模型学会“基于这些表征输出正确控制量”。理论上,预训练阶段已经解决了大部分“理解场景”的问题,微调阶段只需要解决“会开”的问题。数据标注需求量可以比传统方法少一个量级。

4.2 数据闭环:世界模型本身就是一台“模拟器”

把自监督预训练应用好之后,还有一个隐蔽但重要的点:训练出来的世界模型可以作为仿真器,反向生成训练数据。

传统数据闭环长这样:路测车队遇到长尾场景(比如小动物窜出、交通灯故障、特种车辆逆行),把数据采集回来,打标注,加入训练集,再重新训练复测。这个闭环的缺点是:真实长尾场景可遇不可求,复现代价高,通过规则拼接场景又有真实感不足的问题。

而JEPA类的世界模型训练完成后,它内部其实保存了大量“驾驶环境演化的先验知识”。你可以给定一个罕见起始状态(比如“路面湿滑、前两车道同时事故、后车逼得很近”),让世界模型想象出未来几秒可能演化的状态序列,再把这些“想象中的样本”作为训练数据放进决策头继续训练。虽然想象出来的世界不够真实,但它的多样性、覆盖度远大于人工车祸重建。

用这种方式,一个成熟的世界模型系统会形成正循环:

  • 真实场景 → 预训练更新世界模型;
  • 世界模型 → 生成多样化场景 → 训练决策策略;
  • 决策策略 → 路测反馈 → 补充真实场景。

4.3 评测体系也要跟着变

说句实话,我现在看很多端到端方案的评测报告,都有一种“不是同一套标准”的困惑。老一套离线评测指标通常是在测试集上算“预测轨迹与真实轨迹的平均位移误差”,可误差小不等于开车安全——因为平均了之后,偶尔一次激进变道可能被平滑掉。

业界现在越来越认可的一种方式是闭环仿真测试:让模型在仿真环境里跟虚拟交通流对抗,看它会不会撞、会不会卡死、会不会违反交规。世界模型在这个环节里的角色很关键——如果仿真器本身足够接近真世界的“演化规律”,那么闭环测试结果才有意义。JEPA世界模型的表征能力在此可以作为仿真器的“神经核”:不需要手动建模每一个交通参与者的行为,而是让世界模型自己演化出逼真的交互。

评价Drive-JEPA是否为“最新SOTA”,我建议不要只看榜单账号发的三个数字,而是去看它的闭环测试逻辑是否成立:它预测出来的未来状态是否能在长时间跨度中保持稳定?在极端场景下是否会出现表征漂移?这才是真正拉开差距的地方。

5. 从公开演示看SOTA的含金量,以及它还缺什么

5.1 演示视频里能看到的“真实进步”

公开渠道放出的演示视频,通常分两类:一类是简单的直线跟车,另一类是复杂城市路口博弈。从Drive-JEPA相关演示给我的观感来看,进步主要体现在这三点:

第一,连续绕障动作没有“断点感”。很多老端到端方案在绕过第一个障碍物之后,需要200毫秒左右重新规划,这个时间差在车上会表现为一次犹豫、一个停顿。Drive-JEPA在特征层面预测未来状态,可以让规划器提前看到“绕过这个桩桶之后,右前方还有一个减速带”,从而一气呵成。第二,对“隐忧”的反应更平顺。比如旁边车道有一辆车但没打转向灯,模型似乎能通过预测其减速度提前松油门,而不是到撞上前的最后一刻才急刹车。第三,无保护左转这种长周期决策,它看起来敢于“边转边修正”,而不是像行为克隆模型那样转一半停下来。

但我要提醒一句:演示视频是“挑着放”的,公开环境表现不等于全场景稳定。尤其是雨雪天气、夜间极暗环境、传感器部分遮挡这类退化条件下,特征级的预测是否可靠,需要更长期的数据验证。

5.2 跟其他头部智驾方案拉开差异的三个点

现在行业内在一线城市跑路测的方案不少,我拆开看主流技术路线,新的JEPA方案跟其他顶级方案相比,差异化集中在三条线:

一是“预测器”进入主链路。很多方案的世界模型只是用来做“仿真生成”或“预训练特征”,真正推理时并不调用预测器。Drive-JEPA的做法更像“推理时也实时预测未来状态序列”,这让它的规划不是对当前帧的瞬时反应,而是带着“未来视野”的预判。

二是“隐空间决策”。它不做像素域的重建,也不需要像BEV方案那样维护一个显式栅格地图。这个选择直接砍掉了大批量的计算支出,换来的是更长的决策时间窗口。在同一颗Orin-X级别的芯片上,它能比生成式世界模型方案多跑一倍的前向传播频率,这是实测里很容易感知到的差距。

三是“自监督数据收益”。能在同等算力、同等数据量下拿到更好的表征质量,本质上是因为JEPA不浪费参数去建模“无关纹理”。这种差距在低算力部署条件下会被放大——你甚至可以压缩模型尺寸而不显著掉点。

5.3 落地仍然要过的三关

第一关是可解释性与安全评审。监管和车企内部安全团队都非常关心“模型为什么在这里刹车”。JEPA的隐空间预测天然比规则系统更难解释。我在很多项目评审里体会过这种痛苦:你很难向非算法背景的评审者说清楚“嵌入空间里的距离变小了所以决策产生了变化”。现在可行的解法是给JEPA加一个“轻量解释头”,把关键预测状态可视化回图像域,告诉评审者“系统预测到行人会横穿所以减速”。这个解释头必须轻量,不能主链路复用。

第二关是闭环工程复杂度。世界模型作为仿真器的精度是有质量边界的:如果世界模型本身对某种罕见场景学偏了,它生成的仿真样本会放大这个偏置,从而毒化决策模型。工程上需要设质量门禁:每个生成样本要过一层真伪判别器,与真实路段分布差异过大的样本直接丢弃。这一层会增加不少数据管线工作量。

第三关是系统冗余。世界上不存在不fail的模型。JEPA预测头如果出现异常输出,车辆必须有一条退路——要么退回轻量规则控制,要么启用另一套保守策略。这点上我强烈建议做量产的朋友一开始就把“预测头失效检测”写进架构:可以监控预测嵌入的方差,如果方差突然暴涨,说明模型进入了分布外区域,立刻降级。

从工程角度再总结几句个人体会

做自动驾驶算法这些年,我最大的感受是:方案本身牛不牛是一回事,能不能在车规级环境下稳定跑起来是另一回事。Drive-JEPA这种“JEPA世界模型+端到端框架”的组合,理论上的诱人之处在于它绕开了生成式模型的算力陷阱,把预测和决策整合成一个可微整体;工程上的挑战则在于隐空间的可解释性、数据闭环的质量控制,以及失效降级策略。

我会建议正在选技术路线的团队,先别急着全面转向JEPA世界模型。更务实的路径是:先用现有BEV方案保证量产交付,把JEPA作为“预训练表征模块”接入感知层,观察它在长尾场景和大数据预训练上的收益;在仿真环境里跑通闭环评估流程,确认预测头的可靠性之后,再逐步把规划头也迁到隐空间里来。这样既吃到新架构的果实,又不用把压在量产项目上的宝全部押在一个还未被多轮极端场景验证过的方向。

最后分享一个落地小技巧:在给JEPA类模型做数据管线时,一定要维护好“时间对齐索引”。因为要用前一段观测预测后一段状态,数据的传感器时间戳之间如果有毫秒级偏差,训练出来的预测器就会学到错误的时序关系,表现为“过早刹车”或“迟滞反应”。这个问题在真实路测数据里几乎必然存在,但常规的离线数据集检查很少注意到它。先把时间同步做好,再去调网络结构,会省下非常多排查问题的力气。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询