arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.08784cs.CV
收藏

视觉Transformer中基于查询引导的注意力算法

Query-guided Attention in Vision Transformers for Localizing Objects Using a Single Sketch

Indian Institute of Science(印度科学学院);Indian Institute of Technology, Jodhpur(印度理工学院焦特布尔分校)

Aditay Tripathi, Anand Mishra, Anirban Chakraborty

摘要:

在这项工作中,我们研究了自然图像上基于草图的对象定位问题,其中给定一个对象的粗糙手绘草图,目标是定位目标图像上同一对象的所有实例。由于手绘草图的抽象性,草图的风格和质量的变化,以及草图和自然图像之间存在的巨大领域差距,这个问题被证明是困难的。为了缓解这些挑战,现有的作品提出了基于注意力的框架,将查询信息的图像特征。然而,在这些作品中,查询功能被纳入后,图像特征已经被独立学习,导致不充分的对齐。相比之下,我们提出了一个草图引导的视觉Transformer编码器,使用交叉关注后,每个块的基于变压器的图像编码器,以学习查询条件的图像功能,导致更强的对齐与查询草图。此外,在解码器的输出处,对象和草图特征被细化,以使相关对象的表示更接近草图查询,从而改善定位。该模型还可以推广到训练过程中看不到的对象类别,因为我们的方法学习的目标图像特征是查询感知的。我们的本地化框架还可以通过可训练的新草图融合策略来利用多个草图查询。该模型是评估的图像从公共目标检测基准,即MS-COCO,使用草图查询从QuickDraw! Sketchy数据集与现有的定位方法相比,所提出的方法给出了一个$6.6\%$和$8.0\%$的改进mAP看到的对象使用草图查询QuickDraw!和草图数据集,以及AP@50中的$12.2\%$改进,用于训练期间"看不见"的大型对象。

英文摘要:

In this work, we investigate the problem of sketch-based object localization on natural images, where given a crude hand-drawn sketch of an object, the goal is to localize all the instances of the same object on the target image. This problem proves difficult due to the abstract nature of hand-drawn sketches, variations in the style and quality of sketches, and the large domain gap existing between the sketches and the natural images. To mitigate these challenges, existing works proposed attention-based frameworks to incorporate query information into the image features. However, in these works, the query features are incorporated after the image features have already been independently learned, leading to inadequate alignment. In contrast, we propose a sketch-guided vision transformer encoder that uses cross-attention after each block of the transformer-based image encoder to learn query-conditioned image features leading to stronger alignment with the query sketch. Further, at the output of the decoder, the object and the sketch features are refined to bring the representation of relevant objects closer to the sketch query and thereby improve the localization. The proposed model also generalizes to the object categories not seen during training, as the target image features learned by our method are query-aware. Our localization framework can also utilize multiple sketch queries via a trainable novel sketch fusion strategy. The model is evaluated on the images from the public object detection benchmark, namely MS-COCO, using the sketch queries from QuickDraw! and Sketchy datasets. Compared with existing localization methods, the proposed approach gives a $6.6\%$ and $8.0\%$ improvement in mAP for seen objects using sketch queries from QuickDraw! and Sketchy datasets, respectively, and a $12.2\%$ improvement in AP@50 for large objects that are `unseen' during training.

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

↑