论文标题
注意力图平均旋转图形神经网络
Rotation Averaging with Attention Graph Neural Networks
论文作者
论文摘要
在本文中,我们为大规模多旋转平均提出了一种实时和健壮的解决方案。直到最近,使用常规迭代优化算法解决了多个旋转平均问题。此类方法采用了基于对传感器噪声和异常分布的假设选择的强大成本函数。实际上,这些假设并不总是很好地符合实际数据集。最近的工作表明,可以使用图神经网络学习噪声分布。由于平均网络对初始化较差,因此该解决方案需要第二个网络以进行异常检测和去除。在本文中,我们提出了一个单阶段的图形神经网络,该网络可以在存在噪声和异常值的情况下坚固地进行旋转。我们的方法使用所有观察结果,通过使用加权平均和网络设计中的注意机制来抑制异常值的影响。结果是一个网络,该网络比以前的神经方法更快,更健壮,可以接受样本少的训练,最终在准确性和推理时间上优于常规迭代算法。
In this paper we propose a real-time and robust solution to large-scale multiple rotation averaging. Until recently, Multiple rotation averaging problem had been solved using conventional iterative optimization algorithms. Such methods employed robust cost functions that were chosen based on assumptions made about the sensor noise and outlier distribution. In practice, these assumptions do not always fit real datasets very well. A recent work showed that the noise distribution could be learnt using a graph neural network. This solution required a second network for outlier detection and removal as the averaging network was sensitive to a poor initialization. In this paper we propose a single-stage graph neural network that can robustly perform rotation averaging in the presence of noise and outliers. Our method uses all observations, suppressing outliers effects through the use of weighted averaging and an attention mechanism within the network design. The result is a network that is faster, more robust and can be trained with less samples than the previous neural approach, ultimately outperforming conventional iterative algorithms in accuracy and in inference times.