统一多模态对齐,提升自动驾驶感知能力
Toward Unified Multimodal Representation Learning for Autonomous Driving
- 用张量结构联合对齐文本、图像、点云三模态
- 在三个场景中实现优于传统方法的对齐效果
- 适合做自动驾驶多模态预训练的研究者
对比语言-图像预训练(CLIP)在对齐视觉与文本表示方面表现优异。近期研究将此范式拓展至3D视觉,以提升自动驾驶的场景理解能力。常见策略是使用模态间的成对余弦相似度来指导3D编码器训练,但仅关注单对模态的相似性,难以保证整个多模态空间的一致性与统一性。本文提出一种对比张量预训练(CTP)框架,通过统一嵌入空间同时对齐多个模态,以增强端到端自动驾驶性能。相较于成对余弦相似度对齐,本方法将二维相似度矩阵扩展为多模态相似度张量,并引入张量损失实现所有模态间的联合对比学习。为验证框架,我们基于现有自动驾驶数据集构建了一个包含文本-图像-点云三元组的数据集。实验结果表明,所提出的统一多模态对齐框架在两种场景下均表现良好:(i) 将3D编码器与预训练的CLIP编码器对齐;(ii) 从零开始预训练所有编码器。
原文摘要 · Abstract (English)
Contrastive Language-Image Pre-training (CLIP) has shown impressive performance in aligning visual and textual representations. Recent studies have extended this paradigm to 3D vision to improve scene understanding for autonomous driving. A common strategy is to employ pairwise cosine similarity between modalities to guide the training of a 3D encoder. However, considering the similarity between individual modality pairs rather than all modalities jointly fails to ensure consistent and unified alignment across the entire multimodal space. In this paper, we propose a Contrastive Tensor Pre-training (CTP) framework that simultaneously aligns multiple modalities in a unified embedding space to enhance end-to-end autonomous driving. Compared with pairwise cosine similarity alignment, our method extends the 2D similarity matrix into a multimodal similarity tensor. Furthermore, we introduce a tensor loss to enable joint contrastive learning across all modalities. For experimental validation of our framework, we construct a text-image-point cloud triplet dataset derived from existing autonomous driving datasets. The results show that our proposed unified multimodal alignment framework achieves favorable performance for both scenarios: (i) aligning a 3D encoder with pretrained CLIP encoders, and (ii) pretraining all encoders from scratch.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。