少即是多:聚焦注意力的高效 DETR
Sparse DETR 是第一个使用 Transformer 架构的完整目标检测器,其只更新预期被解码器引用的令牌,从而提高模型性能,使性能优于 Deformable DETR。
Nov, 2021
本论文介绍了一种改进的 DETR 检测器,它采用了 “简洁” 的设计,使用单尺度特征图和全局交叉注意力计算,并没有特定的局部约束,与之前基于 DETR 的主要检测器相比,没有重新引入多尺度和局部性的架构偏好。我们展示了两种简单的技术在简洁设计中的出人意料的有效性,以弥补多尺度特征图和局部性约束缺失的问题。第一种是将盒子到像素相对位置偏差(BoxRPB)项添加到交叉注意力公式中,它很好地指导每个查询与对应的对象区域相互关注,并提供了编码灵活性。第二种是基于遮蔽图像模型(MIM)的骨干预训练,有助于学习具有细粒度定位能力的表示,并且对于补救对多尺度特征图的依赖是至关重要的。通过结合这些技术和最新的训练方法和问题形成策略,改进的 “简洁” DETR 检测器在原始 DETR 检测器的基础上取得了显著的改进。通过利用 Object365 数据集进行预训练,使用 Swin-L 骨干网络达到了 63.9 的平均精度(mAP),与依赖于多尺度特征图和基于区域的特征提取的最先进的检测器性能相媲美。代码在此 https URL 中提供。
Aug, 2023
通过引入分层显著性筛选细化和稳定的两阶段初始化查询模块,该研究论文在 DETR 类方法中取得了显著的性能提升,克服了规模偏差和语义不匹配的问题,并在多个任务特定的检测数据集上实现了 4.0% ~ 4.4% 的平均准确率(AP)提高以及在 COCO 2017 数据集上实现了 49.2% 的 AP 提高,并且使用更少的 FLOPs 达到了更高的计算效率。
Mar, 2024
本文提出了 Lite DETR,一种简单而高效的端到端目标检测框架,用于减少检测头的 GFLOPs,同时保持 99%的原始性能。通过设计一种有效的编码器块,以交错的方式更新高级和低级功能,并开发了一种关键感知可变形关注机制,以更好地融合跨尺度特征。全面的实验验证了所提出的 Lite DETR 的有效性和效率,而高效的编码器策略可以很好地推广到现有的 DETR-based 模型中。
Mar, 2023
本研究通过分析数据效率影响因素,发现从图像局部区域进行稀疏特征采样可以减小检测模型对数据的依赖性,并提出了一种简单但有效的标注增强方法,实现最小改动的联合注意力构造,从而提高检测模型在小数据集和样本丰富数据集上的性能。
Mar, 2022
Deformable DETR 提出了一种改进的 Transformer 注意力模块,只关注于围绕参考点的一小集采样点,能够更好地实现物体检测,尤其是在小物体上,并在 COCO 基准测试上得到了有效的实验结果。
Oct, 2020
该论文介绍了一种条件交叉注意力机制,名为条件 DETR,以应对目标检测中 DETR 训练速度较慢的问题,并通过实验证明,相较于传统的 DETR,条件 DETR 在训练速度上提升了 10 倍,并可通过 https 链接获取源代码。
Aug, 2021
本研究针对小物体检测中 RT-DETR 模型的准确性不足提出了两个关键改进:首先,引入细粒度路径增强方法以提供更多详细信息来精确定位小物体;其次,采用自适应特征融合算法来有效整合不同尺度的特征信息,从而提高模型对不同尺度目标的检测准确率。
Jan, 2024
基于 Transformer 的目标检测器 (DETR) 在机器视觉任务中表现出显著性能,但其在处理遮挡和对抗扰动等不同图像干扰方面存在问题。我们通过多种实验和将 DETR 与基于卷积神经网络 (CNN) 的检测器(如 YOLO 和 Faster-RCNN)进行基准测试来研究这个问题。我们发现 DETR 在处理遮挡图像的信息丢失干扰方面表现良好。然而,在涂有对抗标记的图像上,网络需要产生一组新的不必要的键、查询和值,导致网络方向错误。与图像损坏基准测试中 YOLOv5 相比,DETR 的性能也较差。此外,我们发现 DETR 在进行预测时严重依赖于主要查询,导致查询之间的贡献不平衡,因为主要查询接收大部分梯度流。
Oct, 2023
本文提出使用密集先验初始化对象容器的方法,利用 Efficient DETR 同时利用密集检测和稀疏集检测,实现只有 3 个编码器层和 1 个解码器层的端到端物体检测,提高了算法效率和性能。在 MS COCO 数据集上的实验证明了该算法在拥挤场景下仍能稳定工作,并取得最先进算法相近的成绩,且在 CrowdHuman 数据集上表现优异。
Apr, 2021