arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.03213cs.CV

ConsisDrive: 用于视频生成的实例掩码驾驶世界模型

ConsisDrive: Identity-Preserving Driving World Models for Video Generation by Instance Mask

University of Science and Technology of China(中国科学技术大学)

Zhuoran Yang, Yanyong Zhang

AI总结:

ConsisDrive通过实例掩码注意力和损失机制提升驾驶视频生成质量及自动驾驶性能。

摘要:

自动驾驶依赖于在大规模高质量多视角驾驶视频上训练的稳健模型。尽管世界模型为生成逼真驾驶数据提供了成本效益的解决方案,但它们常常受到身份漂移的影响,即由于缺乏实例级时间约束,相同物体在不同帧中会改变外观或类别。我们引入了ConsisDrive,一种旨在强制实例层面时间一致性的身份保持驾驶世界模型。我们的框架包含两个关键组件:(1) 实例掩码注意力,它在注意力块内应用实例身份掩码和轨迹掩码,以确保视觉令牌仅与其对应的实例特征在空间和时间维度上交互,从而保持物体身份一致性;(2) 实例掩码损失,它通过概率性实例掩码自适应强调前景区域,减少背景噪声同时保持整体场景保真度。通过整合这些机制,ConsisDrive实现了最先进的驾驶视频生成质量,并在nuScenes数据集上展示了在下游自动驾驶任务中的显著改进。我们的项目页面是https://shanpoyang654.github.io/ConsisDrive/page.html。

英文摘要:

Autonomous driving relies on robust models trained on large-scale, high-quality multi-view driving videos. Although world models provide a cost-effective solution for generating realistic driving data, they often suffer from identity drift, where the same object changes its appearance or category across frames due to the absence of instance-level temporal constraints. We introduce ConsisDrive, an identity-preserving driving world model designed to enforce temporal consistency at the instance level. Our framework incorporates two key components: (1) Instance-Masked Attention, which applies instance identity masks and trajectory masks within attention blocks to ensure that visual tokens interact only with their corresponding instance features across spatial and temporal dimensions, thereby preserving object identity consistency; and (2) Instance-Masked Loss, which adaptively emphasizes foreground regions with probabilistic instance masking, reducing background noise while maintaining overall scene fidelity. By integrating these mechanisms, ConsisDrive achieves state-of-the-art driving video generation quality and demonstrates significant improvements in downstream autonomous driving tasks on the nuScenes dataset. Our project page is https://shanpoyang654.github.io/ConsisDrive/page.html.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑