目前以大名鼎鼎的YOLO为代表的基于CNN的实时监测网络需要NMS进行后处理,导致不能很好的优化网络,并且网络不够健壮,从而导致检测器的推理速度出现延迟。研究者也分析了Anchor-based和Anchor-free的YOLO的性能,发现Anchor并不是影响实时监测的关键要素,而NMS后处理彩色。
DETR很好的解决了后处理对于模型的限制,却受限于Transformer巨大的计算量,无法发挥实时监测性。
因此,作者想要重构detr,使其具有实时性。研究者发现,虽然多尺度特征的引入有利于加速训练收敛和提高性能,但它也能显著增加输入编码器的序列的长度。因此,Transformer编码器由于计算成本高而成为模型的计算瓶颈。因此,作者设计了一种高效的混合编码器来代替原来的Transformer编码器。通过解耦多尺度特征的尺度内交互和跨尺度融合,该编码器可以有效地处理不同尺度的特征。此外,之前的工作表明,解码器的对象查询初始化方案对检测性能至关重要(Detr存在训练收敛缓慢和Query难以优化的问题)。为了进一步提高性能,作者提出了对iou进行感知的查询选择。通过在训练期间提供IoU约束,为解码器提供更高质量的初始对象查询。此外,提出的检测器支持通过使用不同的解码器层来灵活地调整推理速度,而不需要再训练,这得益于DETR架构中解码器的设计,方便了实时检测器的实际应用。
NMS是目标检测中广泛采用的一种后处理算法,用于消除检测器输出的重叠预测框。而NMS的速度容易受到预测框的个数和两个超参数的影响。
在YOLOv5和YOLOV8进行验证,结合图和表格可以看到,置信度阈值越高,剩余的预测框的个数越少,NMS时间越短。而在同一置信度阈值的前提下,IOU阈值越高,NMS时间越长。
对于需要NMS后处理的实时检测器,无锚检测器的性能优于基于锚的检测器,因为前者占用的后处理时间明显少于后者,这在以往的工作中被忽略了。造成这种现象的原因是,基于锚的探测器比无锚的探测器产生更多的预测框。
4.1. Model Overview
RT-DETR由一个主干网、一个混合编码器和一个带有辅助预测头的Transformer解码器组成。模型架构的概述如图所示。具体来说,利用主干{S3、S4,S53}的最后三个阶段的输出特性作为编码器的输入。该混合编码器通过尺度内交互和跨尺度融合,将多尺度特征转换为一系列图像特征。随后,使用iou感知的查询选择从编码器输出序列中选择固定数量的图像特征,作为解码器的初始对象查询。最后,带有辅助预测头的解码器迭代地优化对象查询,生成预测框和置信度分数。
4.2. Effificient Hybrid Encoder
Computational bottleneck analysis. 多尺度特征会导致输入序列长度急剧增加,虽然能够加速训练的收敛性和提高性能,但是却大大提高了计算量。为了克服这一障碍,作者分析了多尺度Transformer编码器中存在的计算冗余,并设计了一组变体来证明尺度内和跨尺度特征的同时交互作用是计算效率低的。
混合设计:
在此基础上,作者重新考虑了该编码器的结构,并提出了一种新的高效混合编码器。如图3所示,所提出的编码器由两个模块组成,即基于注意的尺度内特征交互(AIFI)模块和基于cnn的跨尺度特征融合模块(CCFM)。AIFI进一步减少了基于变体D的计算冗余,它只在S5上执行尺度内交互。作者认为,将自注意操作应用于具有更丰富语义概念的高级特征,可以捕获图像中概念实体之间的联系,便于后续模块对图像中对象的检测和识别。同时,由于缺乏语义概念,存在与高级特征交互的重复和混淆的风险,低级特征的尺度内交互是不必要的。为了验证这一观点,作者只对变体D中的S5进行了尺度内交互作用,与普通的变体D相比,DS5显著降低了延迟(快了35%),但提高了准确性(AP提高了0.4%)。这一结论对于实时探测器的设计至关重要。CCFM还基于变体D进行了优化,在融合路径中插入了几个由卷积层组成的融合块。融合块的作用是将相邻的特征融合成一个新的特征,其结构如图4所示。融合块包含N个重新块,双路径输出通过元素添加进行融合。可以将这个过程表述如下:
4.3. IoU-aware Query Selection
DETR中的对象查询是一组可学习的嵌入,它们由解码器进行优化,并由预测头映射到分类分数和边界框。然而,这些对象查询很难进行解释和优化,因为它们没有明确的物理意义。其改进方案共同点是利用分类分数从编码器中选择前K个特征来初始化对象查询(或仅选择位置查询[40])。然而,由于分类分数的不一致分布和位置信心,一些预测框高分类分数但不接近真实框,导致高分类分数和低IOU的预测框被选择,而低分类分数和高IOU的预测框被丢弃。 这就损害了探测器的性能。为了解决这个问题,作者提出了IoU感知查询选择,让模型在训练期间对高IOU的特征产生高分类分数,对低IoU的特征产生低分类分数。因此,模型根据分类评分选择的前K个编码器特征对应的预测框具有较高的分类评分和较高的IoU评分。重新制定了检测器的优化目标(引入IOU)如下:
为了分析所提出的IoU感知查询选择的有效性,作者对IOU score和分类分数进行了可视化,如图所示。红点和蓝点分别通过应用普通查询选择和iou感知查询选择训练的模型中计算出来。根据可视化结果,我们发现最显著的特征是大量的蓝色点集中在图的右上角,而红色的点集中在右下角。这表明,用iou感知的查询选择训练的模型可以产生更多高质量的编码器特征。
4.4. Scaled RT-DETR
作者将Backbone由ResNet替换为了HGNetv2,此外,作者还使用了不同的宽度和深度对模型进行拓展。作者提出的不同尺度的RT-DETR保持了一个均匀的解码器,这便于模型蒸馏。
5.3. Ablation Study on Hybrid Encoder
A-E不同混合编码器变体的性能
5.4. Ablation Study on IoU-aware Query Selection