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

从自动驾驶序列的时间排序中进行自监督表示学习

Self-Supervised Representation Learning from Temporal Ordering of Automated Driving Sequences

Robert Bosch GmbH(罗伯特·博世有限公司);University of Freiburg(弗莱堡大学)

Christopher Lang, Alexander Braun, Lars Schillingmann, Karsten Haug, Abhinav Valada

AI总结:

针对自动驾驶序列稠密表示学习不足的问题,提出时间排序前置任务TempO,用无序候选框特征嵌入帧、基于transformer预测帧转移概率,在多个数据集上的检测和跟踪任务性能优于现有方法。

摘要:

自监督特征学习使感知系统能够从全球车队记录的海量原始数据中获益。尽管视频级自监督学习方法已在分类任务上展现出强泛化性,但从序列数据中学习稠密表示的潜力尚未得到充分探索。本研究中,我们提出TempO——一种时间排序前置任务,用于为感知任务预训练区域级特征表示。我们通过一组无序的候选框特征向量嵌入每一帧,这种表示形式天然适用于目标检测或跟踪系统,并在基于transformer的多帧架构中通过预测帧转移概率来构建序列排序,该架构的复杂度随序列长度的增长低于二次方。在BDD100K、nuImages和MOT17数据集上的大量评估表明,我们的TempO预训练方法优于单帧自监督学习方法以及有监督迁移学习初始化策略,在目标检测任务中mAP提升+0.7%,在多目标跟踪任务中HOTA分数提升+2.0%。

英文摘要:

Self-supervised feature learning enables perception systems to benefit from the vast raw data recorded by vehicle fleets worldwide. While video-level self-supervised learning approaches have shown strong generalizability on classification tasks, the potential to learn dense representations from sequential data has been relatively unexplored. In this work, we propose TempO, a temporal ordering pretext task for pre-training region-level feature representations for perception tasks. We embed each frame by an unordered set of proposal feature vectors, a representation that is natural for object detection or tracking systems, and formulate the sequential ordering by predicting frame transition probabilities in a transformer-based multi-frame architecture whose complexity scales less than quadratic with respect to the sequence length. Extensive evaluations on the BDD100K, nuImages, and MOT17 datasets show that our TempO pre-training approach outperforms single-frame self-supervised learning methods as well as supervised transfer learning initialization strategies, achieving an improvement of +0.7% in mAP for object detection and +2.0% in the HOTA score for multi-object tracking.

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

↑