解决检测与分割性能不均衡问题,提升联合模型精度
DI-MaskDINO: A Joint Object Detection and Instance Segmentation Model

- 引入平衡感知查询与优化模块,缓解检测与分割能力失衡
- 在COCO和BDD100K上分别提升1.2点检测、0.9点分割精度
- 适合追求高精度联合检测分割任务的研究者使用
本文观察到,在当前领先的联合检测与分割模型MaskDINO中,从Transformer解码器首层中间结果可见:检测性能落后于分割性能(即性能不平衡)。这一现象引发思考:早期解码层的不平衡是否限制了最终性能上限?通过定性与定量预实验验证了该不平衡对模型性能的负面影响。为此,本文提出DI-MaskDINO模型,核心思想是通过缓解检测-分割不平衡来提升整体性能。具体实现为在MaskDINO中加入所提出的去不平衡(DI)模块与平衡感知令牌优化(BATO)模块:DI生成平衡感知查询,BATO利用该查询引导初始特征令牌优化。随后,平衡感知查询与优化后的特征令牌分别作为解码器的Query与Key&Value,完成联合检测与实例分割。DI-MaskDINO在COCO和BDD100K基准上优于现有联合检测分割模型,相较SOTA模型MaskDINO,检测指标+1.2 $AP^{box}$,分割指标+0.9 $AP^{mask}$;同时相比SOTA检测模型DINO,检测提升+1.0 $AP^{box}$;相比SOTA分割模型Mask2Former,分割提升+3.0 $AP^{mask}$。
原文摘要 · Abstract (English)
This paper is motivated by an interesting phenomenon: the performance of object detection lags behind that of instance segmentation (i.e., performance imbalance) when investigating the intermediate results from the beginning transformer decoder layer of MaskDINO (i.e., the SOTA model for joint detection and segmentation). This phenomenon inspires us to think about a question: will the performance imbalance at the beginning layer of transformer decoder constrain the upper bound of the final performance? With this question in mind, we further conduct qualitative and quantitative pre-experiments, which validate the negative impact of detection-segmentation imbalance issue on the model performance. To address this issue, this paper proposes DI-MaskDINO model, the core idea of which is to improve the final performance by alleviating the detection-segmentation imbalance. DI-MaskDINO is implemented by configuring our proposed De-Imbalance (DI) module and Balance-Aware Tokens Optimization (BATO) module to MaskDINO. DI is responsible for generating balance-aware query, and BATO uses the balance-aware query to guide the optimization of the initial feature tokens. The balance-aware query and optimized feature tokens are respectively taken as the Query and Key&Value of transformer decoder to perform joint object detection and instance segmentation. DI-MaskDINO outperforms existing joint object detection and instance segmentation models on COCO and BDD100K benchmarks, achieving +1.2 $AP^{box}$ and +0.9 $AP^{mask}$ improvements compared to SOTA joint detection and segmentation model MaskDINO. In addition, DI-MaskDINO also obtains +1.0 $AP^{box}$ improvement compared to SOTA object detection model DINO and +3.0 $AP^{mask}$ improvement compared to SOTA segmentation model Mask2Former.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。