给6D位姿预测模型加个不确定度模块,让机器人导航更准。
Aleatoric Uncertainty from AI-based 6D Object Pose Predictors for Object-relative State Estimation
- 在现有位姿模型后接两个独立感知机,实现不确定性推断。
- 实测在真实与合成数据上均提升状态估计精度,优于固定噪声方法。
- 计算开销极小,适合部署在边缘设备上使用。
深度学习已广泛应用于机器人领域,尤其在基于视觉的物体相对导航中,依赖深度神经网络(DNN)预测物体与机器人间的6维位姿作为状态估计器的测量输入。准确获取此类基于DNN测量的随机不确定性,对概率状态估计算法至关重要。本文提出仅通过在现有预训练的DL位姿预测器的平移和旋转分支后附加两个多层感知机,即可高效实现对6D位姿的随机不确定性推断,且无需重新训练主干网络。所推断出的位姿及其协方差矩阵被用于扩展卡尔曼滤波(EKF)的状态估计。该方法引入的计算开销极小,使状态估计算法可在边缘设备上运行,并受益于动态变化的测量不确定性。在合成数据与真实数据上的实验表明,相比固定协方差方法,本方案显著提升了物体相对状态估计性能。
原文摘要 · Abstract (English)
Deep Learning (DL) has become essential in various robotics applications due to excelling at processing raw sensory data to extract task specific information from semantic objects. For example, vision-based object-relative navigation relies on a DL-based 6D object pose predictor to provide the relative pose between the object and the robot as measurements to the robot's state estimator. Accurately knowing the uncertainty inherent in such Deep Neural Network (DNN) based measurements is essential for probabilistic state estimators subsequently guiding the robot's tasks. Thus, in this letter, we show that we can extend any existing DL-based object-relative pose predictor for aleatoric uncertainty inference simply by including two multi-layer perceptrons detached from the translational and rotational part of the DL predictor. This allows for efficient training while freezing the existing pre-trained predictor. We then use the inferred 6D pose and its uncertainty as a measurement and corresponding noise covariance matrix in an extended Kalman filter (EKF). Our approach induces minimal computational overhead such that the state estimator can be deployed on edge devices while benefiting from the dynamically inferred measurement uncertainty. This increases the performance of the object-relative state estimation task compared to a fix-covariance approach. We conduct evaluations on synthetic data and real-world data to underline the benefits of aleatoric uncertainty inference for the object-relative state estimation task.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。