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

StageInteractor:具有跨阶段交互的基于查询的目标检测器

StageInteractor: Query-based Object Detector with Cross-stage Interaction

State Key Laboratory for Novel Software Technology, Nanjing University(南京大学计算机软件新技术国家重点实验室);MYbank, Ant Group(蚂蚁集团网商银行);Shanghai AI Lab(上海人工智能实验室)

Yao Teng, Haisong Liu, Sheng Guo, Limin Wang

AI总结:

本文提出StageInteractor,通过跨阶段标签分配和轻量级适配器复用动态算子,增强基于查询检测器的建模能力,在MS COCO上显著提升检测精度。

摘要:

以往的目标检测器基于密集网格点或大量预设锚框进行预测,其中大多数检测器采用一对多标签分配策略进行训练。相反,近期基于查询的目标检测器依赖于一组稀疏的可学习查询和一系列解码器层。在训练过程中,一对一标签分配被独立地应用于每一层以实现深度监督。尽管基于查询的目标检测取得了巨大成功,然而这种一对一标签分配策略要求检测器具备强大的细粒度判别能力和建模能力。为解决上述问题,本文提出了一种新的具有跨阶段交互的基于查询的目标检测器,命名为StageInteractor。在前向传播过程中,我们提出了一种高效的方法,通过复用带有轻量级适配器的动态算子来提升这种建模能力。在标签分配方面,在一对一标签分配之后应用了一个跨阶段标签分配器。借助该分配器,训练目标类别标签在多个阶段之间被收集,然后在每个解码器层重新分配给合适的预测结果。在MS COCO基准上,我们的模型将基线提升了2.2 AP,并在使用ResNet-50作为骨干网络、100个查询和12个训练轮次的设置下达到44.8 AP。在更长的训练时间和300个查询的设置下,StageInteractor分别使用ResNeXt-101-DCN和Swin-S达到51.1 AP和52.2 AP。

英文摘要:

Previous object detectors make predictions based on dense grid points or numerous preset anchors. Most of these detectors are trained with one-to-many label assignment strategies. On the contrary, recent query-based object detectors depend on a sparse set of learnable queries and a series of decoder layers. The one-to-one label assignment is independently applied on each layer for the deep supervision during training. Despite the great success of query-based object detection, however, this one-to-one label assignment strategy demands the detectors to have strong fine-grained discrimination and modeling capacity. To solve the above problems, in this paper, we propose a new query-based object detector with cross-stage interaction, coined as StageInteractor. During the forward propagation, we come up with an efficient way to improve this modeling ability by reusing dynamic operators with lightweight adapters. As for the label assignment, a cross-stage label assigner is applied subsequent to the one-to-one label assignment. With this assigner, the training target class labels are gathered across stages and then reallocated to proper predictions at each decoder layer. On MS COCO benchmark, our model improves the baseline by 2.2 AP, and achieves 44.8 AP with ResNet-50 as backbone, 100 queries and 12 training epochs. With longer training time and 300 queries, StageInteractor achieves 51.1 AP and 52.2 AP with ResNeXt-101-DCN and Swin-S, respectively.

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

↑