arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.04519cs.CV

基于自监督学习的自动驾驶雷达自举方法

Bootstrapping Autonomous Driving Radars with Self-Supervised Learning

University of Cambridge(剑桥大学);EPFL(洛桑联邦理工学院);UIUC(伊利诺伊大学厄巴纳-香槟分校);RadarEye(雷达眼公司);Nokia Bell Labs(诺基亚贝尔实验室)

Yiduo Hao, Sohrab Madani, Junfeng Guan, Mohammed Alloulah, Saurabh Gupta, Haitham Hassanieh

AI总结:

针对自动驾驶雷达模型训练标注成本高的问题,提出结合雷达-雷达与雷达-视觉对比损失的自监督预训练框架,利用未标注雷达数据学习通用表征,将下游目标检测的SOTA监督基线mAP提升5.8%。

摘要:

自动驾驶车辆利用雷达进行感知因具备雾天及恶劣天气下的工作能力,已吸引了越来越多的研究兴趣。然而,大规模雷达数据标注的成本与难度阻碍了雷达模型的训练。为突破这一瓶颈,我们提出一种自监督学习框架,利用大量未标注雷达数据预训练仅用于雷达的嵌入表示,服务于自动驾驶感知任务。该方法结合雷达-雷达与雷达-视觉对比损失,从未标注雷达热图及其配对的对应相机图像中学习通用表征。在用于下游目标检测任务时,我们证明所提出的自监督框架可将当前最优监督基线的mAP提升5.8%。代码可在https://github.com/yiduohao/Radical 获取。

英文摘要:

The perception of autonomous vehicles using radars has attracted increased research interest due its ability to operate in fog and bad weather. However, training radar models is hindered by the cost and difficulty of annotating large-scale radar data. To overcome this bottleneck, we propose a self-supervised learning framework to leverage the large amount of unlabeled radar data to pre-train radar-only embeddings for self-driving perception tasks. The proposed method combines radar-to-radar and radar-to-vision contrastive losses to learn a general representation from unlabeled radar heatmaps paired with their corresponding camera images. When used for downstream object detection, we demonstrate that the proposed self-supervision framework can improve the accuracy of state-of-the-art supervised baselines by $5.8\%$ in mAP. Code is available at \url{https://github.com/yiduohao/Radical}.

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

↑