arXiv:2503.10152cs.CV2025-03中稿 · TMM 2025被引 11

通过三层语义蒸馏,让检测器学会识别训练外的新物体。

A Hierarchical Semantic Distillation Framework for Open-Vocabulary Object Detection

  • 分三层次从CLIP模型中蒸馏实例、类别和图像级语义
  • 在OV-COCO上新类识别准确率提升至46.4%(ResNet50)
  • 适合需要识别未见物体的开放词汇检测场景

开放词汇目标检测(OVD)旨在识别训练标注之外的物体。现有方法通常直接将检测器与预训练视觉语言模型(如CLIP)对齐,以继承其泛化识别能力,但未能有效学习语义知识。本文提出分层语义蒸馏框架HD-OVD,从三个层面挖掘CLIP模型的通用知识:第一层,通过建模视觉空间中单个物体间关系,学习细粒度实例级语义;第二层,引入文本空间中的新类别感知分类,吸收CLIP文本编码器的类别级泛化语义;第三层,通过图像级对比蒸馏,提取包含多物体及其上下文的图像级语义。得益于三层次精细化的语义蒸馏,HD-OVD在实例、类别和图像层面均继承了CLIP的泛化能力。在OV-COCO数据集上,采用ResNet50主干网络时,新类平均精度(novel AP)达46.4%,显著优于现有方法。我们还进行了大量消融实验,分析各组件作用。

原文摘要 · Abstract (English)

Open-vocabulary object detection (OVD) aims to detect objects beyond the training annotations, where detectors are usually aligned to a pre-trained vision-language model, eg, CLIP, to inherit its generalizable recognition ability so that detectors can recognize new or novel objects. However, previous works directly align the feature space with CLIP and fail to learn the semantic knowledge effectively. In this work, we propose a hierarchical semantic distillation framework named HD-OVD to construct a comprehensive distillation process, which exploits generalizable knowledge from the CLIP model in three aspects. In the first hierarchy of HD-OVD, the detector learns fine-grained instance-wise semantics from the CLIP image encoder by modeling relations among single objects in the visual space. Besides, we introduce text space novel-class-aware classification to help the detector assimilate the highly generalizable class-wise semantics from the CLIP text encoder, representing the second hierarchy. Lastly, abundant image-wise semantics containing multi-object and their contexts are also distilled by an image-wise contrastive distillation. Benefiting from the elaborated semantic distillation in triple hierarchies, our HD-OVD inherits generalizable recognition ability from CLIP in instance, class, and image levels. Thus, we boost the novel AP on the OV-COCO dataset to 46.4% with a ResNet50 backbone, which outperforms others by a clear margin. We also conduct extensive ablation studies to analyze how each component works.

开放词汇检测语义蒸馏CLIP

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