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

基于多分辨率可变形注意力与查询聚合的高效多目标姿态估计

Efficient Multi-Object Pose Estimation using Multi-Resolution Deformable Attention and Query Aggregation

University of Bonn(波恩大学)

Arul Selvam Periyasamy, Vladimir Tsaturyan, Sven Behnke

AI总结:

针对多目标姿态估计问题,提出融合多分辨率可变形注意力与查询聚合机制的视觉Transformer模型,在YCB-Video数据集上取得当前最优结果,兼顾长程依赖学习与计算效率。

摘要:

目标姿态估计是计算机视觉领域的一个长期存在的问题。近年来,基于注意力的视觉Transformer模型已在众多计算机视觉应用中取得了当前最优(state-of-the-art)结果。利用注意力机制的置换不变性,一类视觉Transformer模型将多目标姿态估计建模为集合预测问题。然而,现有的多目标姿态估计视觉Transformer模型完全依赖注意力机制。另一方面,卷积神经网络则将各类归纳偏置硬编码到其架构中。本文研究在用于多目标姿态估计的视觉Transformer模型中融入归纳偏置,这既能促进长程依赖的学习,又能规避计算成本高昂的全局注意力。具体而言,我们采用多分辨率可变形注意力,其注意力操作仅在少量变形参考点之间执行。此外,我们提出一种查询聚合机制,能够在不增加计算复杂度的前提下增加目标查询的数量。我们在具有挑战性的YCB-Video数据集上评估了所提出的模型,并取得了当前最优结果。

英文摘要:

Object pose estimation is a long-standing problem in computer vision. Recently, attention-based vision transformer models have achieved state-of-the-art results in many computer vision applications. Exploiting the permutation-invariant nature of the attention mechanism, a family of vision transformer models formulate multi-object pose estimation as a set prediction problem. However, existing vision transformer models for multi-object pose estimation rely exclusively on the attention mechanism. Convolutional neural networks, on the other hand, hard-wire various inductive biases into their architecture. In this paper, we investigate incorporating inductive biases in vision transformer models for multi-object pose estimation, which facilitates learning long-range dependencies while circumventing the costly global attention. In particular, we use multi-resolution deformable attention, where the attention operation is performed only between a few deformed reference points. Furthermore, we propose a query aggregation mechanism that enables increasing the number of object queries without increasing the computational complexity. We evaluate the proposed model on the challenging YCB-Video dataset and report state-of-the-art results.

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

↑