用预训练ViT调制CNN特征,提升开放词汇目标检测能力
Modulating CNN Features with Pre-Trained ViT Representations for Open-Vocabulary Object Detection
- 双分支结构:可训练CNN+冻结ViT,结合两者优势
- 在OV-COCO上新类别检测达44.3 AP₅₀,优于现有方法
- 适合需要识别未见类别的目标检测场景
由于大规模图文对比训练,预训练视觉语言模型(如CLIP)具备优异的开放词汇识别能力。现有方法多将预训练VLM作为主干网络并冻结其权重,但无法利用标注数据优化表示。为此,本文提出新型双分支主干网络VMCNet,包含可训练的卷积分支、冻结的预训练ViT分支及特征调制模块(VMC)。CNN分支通过标注数据优化,而ViT分支保持大规模预训练带来的泛化能力。VMC模块将多尺度CNN特征与ViT分支表示进行调制,增强对新类别物体的发现能力。在两个主流基准测试中表现优异:在OV-COCO上,使用ViT-B/16和ViT-L/14分别达到44.3 AP₅₀^{novel}和48.5 AP₅₀^{novel};在OV-LVIS上,对应结果为27.8和38.4 mAP_r。
原文摘要 · Abstract (English)
Owing to large-scale image-text contrastive training, pre-trained vision language model (VLM) like CLIP shows superior open-vocabulary recognition ability. Most existing open-vocabulary object detectors attempt to utilize the pre-trained VLMs to attain generalized representation. F-ViT uses the pre-trained visual encoder as the backbone network and freezes it during training. However, its frozen backbone doesn't benefit from the labeled data to strengthen the representation for detection. Therefore, we propose a novel two-branch backbone network, named as \textbf{V}iT-Feature-\textbf{M}odulated Multi-Scale \textbf{C}onvolutional Network (VMCNet), which consists of a trainable convolutional branch, a frozen pre-trained ViT branch and a VMC module. The trainable CNN branch could be optimized with labeled data while the frozen pre-trained ViT branch could keep the representation ability derived from large-scale pre-training. Then, the proposed VMC module could modulate the multi-scale CNN features with the representations from ViT branch. With this proposed mixed structure, the detector is more likely to discover objects of novel categories. Evaluated on two popular benchmarks, our method boosts the detection performance on novel category and outperforms state-of-the-art methods. On OV-COCO, the proposed method achieves 44.3 AP$_{50}^{\mathrm{novel}}$ with ViT-B/16 and 48.5 AP$_{50}^{\mathrm{novel}}$ with ViT-L/14. On OV-LVIS, VMCNet with ViT-B/16 and ViT-L/14 reaches 27.8 and 38.4 mAP$_{r}$.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。