arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.03318cs.CVcs.MM

通过改进内容查询与相似查询聚合增强DETR变体

Enhancing DETRs Variants through Improved Content Query and Similar Query Aggregation

Ant Group(蚂蚁集团)

Yingying Zhang, Chuangji Shi, Xin Guo, Jiangwei Lao, Jian Wang, Jiaotuan Wang, Jingdong Chen

AI总结:

本文提出自适应内容查询模块SACQ和相似查询聚合策略,通过自注意力池化生成内容查询并合并相似候选,在COCO上使六种DETR变体平均提升超过1.0 AP。

摘要:

查询的设计对于DETR及其变体的性能至关重要。每个查询由两部分组成:内容部分和位置部分。传统上,内容查询用零向量或可学习嵌入进行初始化,缺乏必要的内容信息,导致性能次优。本文提出了一种新颖的即插即用模块——自适应内容查询(SACQ),以解决这一局限。SACQ模块利用Transformer编码器的特征,通过自注意力池化生成内容查询。这使得候选查询能够适应输入图像,从而获得更全面的内容先验,并更好地聚焦于目标对象。然而,这种增强的集中性给使用匈牙利匹配的训练过程带来了挑战,因为匈牙利匹配只选择单个候选并抑制其他相似候选。为克服这一问题,我们提出了一种查询聚合策略,与SACQ协同工作。该策略合并来自不同查询的相似预测候选,从而简化优化过程。我们在COCO数据集上的大量实验证明了所提方法在六种不同DETR变体及多种配置下的有效性,平均提升超过1.0 AP。

英文摘要:

The design of the query is crucial for the performance of DETR and its variants. Each query consists of two components: a content part and a positional one. Traditionally, the content query is initialized with a zero or learnable embedding, lacking essential content information and resulting in sub-optimal performance. In this paper, we introduce a novel plug-and-play module, Self-Adaptive Content Query (SACQ), to address this limitation. The SACQ module utilizes features from the transformer encoder to generate content queries via self-attention pooling. This allows candidate queries to adapt to the input image, resulting in a more comprehensive content prior and better focus on target objects. However, this improved concentration poses a challenge for the training process that utilizes the Hungarian matching, which selects only a single candidate and suppresses other similar ones. To overcome this, we propose a query aggregation strategy to cooperate with SACQ. It merges similar predicted candidates from different queries, easing the optimization. Our extensive experiments on the COCO dataset demonstrate the effectiveness of our proposed approaches across six different DETR's variants with multiple configurations, achieving an average improvement of over 1.0 AP.

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

↑