WA-JEPA

新WA-JEPA模型推动自动驾驶规划发展

研究人员开发了WA-JEPA，一种用于自动驾驶规划的新型世界-动作模型，它改进了视频联合嵌入预测架构（V-JEPA）范式。与之前使用随机掩码和确定性回归的V-JEPA模型不同，WA-JEPA采用了混合未来掩码预训练和条件流匹配，以实现更有效的未来预测。这种方法允许联合未来-动作预测，直接塑造与规划相关的世界表征，并在NAVSIM和HUGSIM等基准测试中取得了最先进的成果。