论文标题

昏迷:自我教学的不确定性估计值

STUN: Self-Teaching Uncertainty Estimation for Place Recognition

论文作者

Cai, Kaiwen, Lu, Chris Xiaoxuan, Huang, Xiaowei

论文摘要

位置识别是同时定位和映射(猛击)和空间感知的关键。但是,野外的地方识别通常会因图像变化(例如改变观点和街头外观)而产生错误的预测。将不确定性估计纳入地点识别的生命周期是减轻变化对位置识别性能的影响的有前途的方法。但是,这种静脉的现有不确定性估计方法要么是计算效率低下(例如蒙特卡洛辍学),要么以降低准确性为代价。本文提出了Stun,这是一个自学框架,学会同时预测位置并估算给定输入图像的预测不确定性。为此,我们首先使用标准的度量学习管道培训教师网,以生产嵌入培训。然后,在经过审计的教师网络监督的情况下,培训了一个具有额外差异分支的学生网,可以通过样本估算嵌入者的先验并估算不确定性样本。在在线推论阶段,我们仅使用学生网与不确定性结合产生位置预测。与对不确定性一无所知的位置识别系统相比,我们的框架具有自由估计的不确定性估计而无需牺牲任何预测准确性。我们对大规模匹兹堡30K数据集的实验结果表明,昏迷在识别精度和不确定性估计质量方面的表现都优于最新方法。

Place recognition is key to Simultaneous Localization and Mapping (SLAM) and spatial perception. However, a place recognition in the wild often suffers from erroneous predictions due to image variations, e.g., changing viewpoints and street appearance. Integrating uncertainty estimation into the life cycle of place recognition is a promising method to mitigate the impact of variations on place recognition performance. However, existing uncertainty estimation approaches in this vein are either computationally inefficient (e.g., Monte Carlo dropout) or at the cost of dropped accuracy. This paper proposes STUN, a self-teaching framework that learns to simultaneously predict the place and estimate the prediction uncertainty given an input image. To this end, we first train a teacher net using a standard metric learning pipeline to produce embedding priors. Then, supervised by the pretrained teacher net, a student net with an additional variance branch is trained to finetune the embedding priors and estimate the uncertainty sample by sample. During the online inference phase, we only use the student net to generate a place prediction in conjunction with the uncertainty. When compared with place recognition systems that are ignorant to the uncertainty, our framework features the uncertainty estimation for free without sacrificing any prediction accuracy. Our experimental results on the large-scale Pittsburgh30k dataset demonstrate that STUN outperforms the state-of-the-art methods in both recognition accuracy and the quality of uncertainty estimation.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源