用轻量级语义增强提升开放词汇目标检测在垂直领域的适应性。
Parameter-Efficient Semantic Augmentation for Enhancing Open-Vocabulary Object Detection
- 构建多尺度提示库,从图像特征金字塔中提取细粒度局部语义。
- 在垂直领域数据集上平均提升12.3%的AP,优于现有方法。
- 动态路由机制避免模型泛化能力下降,适合工业场景迁移。
开放词汇目标检测(OVOD)使模型能够识别任意类别,包括未见类别。尽管现有方法在通用场景(如OV-COCO)上表现良好,但在存在显著领域偏移的下游任务中性能大幅下降。这主要源于领域特定任务中类别标签稀缺且语义薄弱,以及现有模型无法捕捉超出粗粒度类别的辅助语义。为此,我们提出HSA-DINO,一种参数高效的语义增强框架。具体而言,设计多尺度提示库,利用图像特征金字塔捕捉层次化语义,并选择领域特定的局部语义提示,逐步丰富文本表示。此外,引入语义感知路由器,在推理时动态选择合适的语义增强策略,防止参数更新损害预训练模型的泛化能力。我们在OV-COCO、多个垂直领域数据集及修改的基准设置上评估HSA-DINO,结果表明其在领域适应性与开放词汇泛化之间取得更优平衡,显著优于此前最优方法。
原文摘要 · Abstract (English)
Open-vocabulary object detection (OVOD) enables models to detect any object category, including unseen ones. Benefiting from large-scale pre-training, existing OVOD methods achieve strong detection performance on general scenarios (e.g., OV-COCO) but suffer severe performance drops when transferred to downstream tasks with substantial domain shifts. This degradation stems from the scarcity and weak semantics of category labels in domain-specific task, as well as the inability of existing models to capture auxiliary semantics beyond coarse-grained category label. To address these issues, we propose HSA-DINO, a parameter-efficient semantic augmentation framework for enhancing open-vocabulary object detection. Specifically, we propose a multi-scale prompt bank that leverages image feature pyramids to capture hierarchical semantics and select domain-specific local semantic prompts, progressively enriching textual representations from coarse to fine-grained levels. Furthermore, we introduce a semantic-aware router that dynamically selects the appropriate semantic augmentation strategy during inference, thereby preventing parameter updates from degrading the generalization ability of the pre-trained OVOD model. We evaluate HSA-DINO on OV-COCO, several vertical domain datasets, and modified benchmark settings. The results show that HSA-DINO performs favorably against previous state-of-the-art methods, achieving a superior trade-off between domain adaptability and open-vocabulary generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。