提升弱监督目标检测精度,解决分类任务间的混淆问题
Self-Classification Enhancement and Correction for Weakly Supervised Object Detection
- 引入类内二分类模块增强类别区分能力
- 推理时融合双任务结果减少误判
- 在VOC数据集上性能优于现有方法
近年来,弱监督目标检测(WSOD)因标注成本低而受到关注。当前主流方法依赖两阶段多类别分类(MCC)任务,即多实例学习与在线分类精炼。然而,这些方法忽略了两个任务间的潜在分类歧义,未能充分发挥各自优势。本文提出一种新框架,首先设计自分类增强模块,引入类内二分类(ICBC)任务,提升网络对正样本与错位样本的判别力,并与MCC任务形成协同增强;其次在推理阶段提出自分类校正算法,融合两个任务结果以有效降低误分类预测。在主流的VOC 2007 & 2012数据集上的大量实验表明,该框架性能显著优于现有方法。
原文摘要 · Abstract (English)
In recent years, weakly supervised object detection (WSOD) has attracted much attention due to its low labeling cost. The success of recent WSOD models is often ascribed to the two-stage multi-class classification (MCC) task, i.e., multiple instance learning and online classification refinement. Despite achieving non-trivial progresses, these methods overlook potential classification ambiguities between these two MCC tasks and fail to leverage their unique strengths. In this work, we introduce a novel WSOD framework to ameliorate these two issues. For one thing, we propose a self-classification enhancement module that integrates intra-class binary classification (ICBC) to bridge the gap between the two distinct MCC tasks. The ICBC task enhances the network's discrimination between positive and mis-located samples in a class-wise manner and forges a mutually reinforcing relationship with the MCC task. For another, we propose a self-classification correction algorithm during inference, which combines the results of both MCC tasks to effectively reduce the mis-classified predictions. Extensive experiments on the prevalent VOC 2007 & 2012 datasets demonstrate the superior performance of our framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。