arXiv:2605.23507cs.CV2026-05

MDS-DETR用掩码抑制重复,让检测变压器更快更准。

MDS-DETR: DETR with Masked Duplicate Suppressor

  • 单解码器内融合一对一与一对多监督,避免额外计算
  • 引入置信度因果掩码过滤重复预测,提升召回率2.8个点
  • 无需辅助解码器,训练快20%,适合追求效率的部署场景

DETR是一种强大的端到端目标检测器,但其一对一匹配策略导致收敛慢、召回率低。现有方法通过一对多标签分配增加正样本,但需额外解码器,训练成本高且推理时丢弃。为此,我们提出MDS-DETR,将一对一和一对多监督整合于单一解码器中。具体地,引入掩码重复抑制器(MDS),通过置信度引导的因果掩码在自注意力中注入非对称性,过滤一对多层生成的重复预测,实现可解释、无重复的端到端预测。在使用ResNet-50主干网络的MS COCO数据集上,12轮训练下,MDS-DETR相比Deformable-DETR提升2.8 mAP,训练时间仅增加5%;相比当前最优的MR.DETR提升0.3 mAP,且训练速度更快20%。代码与模型已开源。

原文摘要 · Abstract (English)

The DEtection TRansformer (DETR) is a powerful end-to-end object detector, yet its one-to-one matching strategy suffers from slow convergence and low recall. A common approach to address this issue is to use one-to-many label assignment to provide more positive samples. However, existing methods that use one-to-many matching as an auxiliary objective lead to increased training costs, with their auxiliary decoders discarded during inference. To address this limitation, we propose MDS-DETR, which leverages both one-to-one and one-to-many supervision within a single decoder. Specifically, we introduce a Masked Duplicate Suppressor (MDS) that injects asymmetry into self-attention via confidence-based causal masking. MDS filters out the duplicates generated by the one-to-many supervised layer, enables explainable, duplicate-free predictions in a fully end-to-end framework. MDS-DETR outperforms existing one-to-many DETR variants such as MS-DETR, MR.DETR and Relation-DETR, without relying on any additional queries or auxiliary decoders. Under a 12-epoch training schedule on MS COCO with a ResNet-50 backbone, MDS-DETR achieves a +2.8 mAP improvement over Deformable-DETR with only a 5\% increase in training time, and outperforms the state-of-the-art MR.DETR by +0.3 mAP while being even 20\% faster in training. Our code and models are available at \href{https://github.com/dcholee/mds-detr}{https://github.com/DChoLee/MDS-DETR}.

目标检测TransformerDETR高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。