尺度等效蒸馏:半监督目标检测的新方法

0 下载量 155 浏览量 更新于2025-01-16 收藏 1.52MB PDF 举报
"这篇论文提出了一种名为尺度等效蒸馏(Scale-Equivalent Distillation, SED)的半监督目标检测方法,旨在解决在有限标注数据情况下进行目标检测的挑战,特别是面对类不平衡和对象大小变化的问题。SED是一种端到端的知识蒸馏框架,能够提高对不同大小对象的检测鲁棒性,并且通过一致性正则化处理大规模方差。同时,它还通过自我蒸馏来减少假阴性样本的噪声,提高定位精度。为了应对类别不平衡,该方法采用了重加权策略,隐式筛选出未标注数据中的潜在前景区域。在MS-COCO数据集上,与其他最新方法相比,SED表现出显著的优势,尤其是在只有5%和10%的标注数据情况下。" **1. 引言** 深度学习在大规模标注数据集如ImageNet的支持下,取得了显著的进步。然而,目标检测任务由于需要精确的边界框标注,其数据获取成本远高于分类任务。半监督学习为此提供了一个解决方案,尝试在少量标注数据的情况下进行学习。尽管半监督分类取得了显著成就,但在目标检测领域,即SS-OD,仍面临更多挑战。 **2. 挑战与解决方案** - **类不平衡**:在目标检测中,背景通常占据大部分图像,导致类别严重不平衡。SED通过重加权策略缓解这个问题,优先关注潜在的前景区域。 - **对象大小变化**:不同对象的尺度差异影响检测性能。SED引入尺度等效蒸馏,通过一致性正则化确保不同大小图像间的检测一致性。 - **假阴性样本与定位精度**:大量假阴性样本(误判为背景的对象)和低定位精度是SS-OD的难题。SED使用自我蒸馏方法减少这类噪声,提升定位准确性。 **3. 方法论** - **尺度等效蒸馏(SED)框架**:这个框架旨在让模型在不同尺度的图像上表现一致,提高对大小变化的鲁棒性。 - **一致性正则化**:通过一致性约束,使模型在不同变换的同一图像上预测结果保持一致,降低大规模方差。 - **自我蒸馏**:通过模型自身的输出指导学习过程,减少假阴性样本的噪声,提高定位精度。 - **重加权策略**:对未标注数据进行加权,强调潜在前景区域,以平衡类别分布。 **4. 实验与结果** 在MS-COCO数据集上的实验验证了SED的有效性,即使在仅使用5%或10%的标注数据时,也明显优于现有的SS-OD方法。 **结论** SED为半监督目标检测提供了一种有效的方法,通过处理类不平衡、对象大小变化和假阴性样本等问题,提高了模型在有限标注数据条件下的检测性能。这种方法的简洁性和有效性为未来的研究提供了新的方向。