arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.04514cs.CV
收藏

DetCLIPv2:通过词-区域对齐实现可扩展的开放词汇目标检测预训练

DetCLIPv2: Scalable Open-Vocabulary Object Detection Pre-training via Word-Region Alignment

Hong Kong University of Science and Technology(香港科技大学);Huawei Noah’s Ark Lab(华为诺亚方舟实验室);Shenzhen Campus of Sun Yat-Sen University(中山大学深圳校区)

Lewei Yao, Jianhua Han, Xiaodan Liang, Dan Xu, Wei Zhang, Zhenguo Li, Hang Xu

AI总结:

本文提出DetCLIPv2框架,通过端到端学习大规模图文对的细粒度词-区域对齐并采用混合监督,在LVIS基准上取得40.4%零样本AP,大幅超越现有方法及全监督模型。

摘要:

本文提出了DetCLIPv2,一个高效且可扩展的训练框架,它结合大规模图文对来实现开放词汇目标检测(OVD)。与以往通常依赖预训练视觉-语言模型(如CLIP)或通过伪标注过程利用图文对的OVD框架不同,DetCLIPv2以端到端的方式直接从海量图文对中学习细粒度的词-区域对齐。为此,我们采用区域提议与文本词之间的最大词-区域相似度来指导对比目标。为了使模型在学习广泛概念的同时获得定位能力,DetCLIPv2在统一的数据格式下,通过来自检测、定位和图文对数据的混合监督进行训练。通过交替方案联合训练并对图文对采用低分辨率输入,DetCLIPv2高效且有效地利用了图文对数据:DetCLIPv2比DetCLIP多利用13倍的图文对,且训练时间相近并提升了性能。使用13M图文对进行预训练,DetCLIPv2展现出卓越的开放词汇检测性能,例如,采用Swin-T骨干网络的DetCLIPv2在LVIS基准上达到40.4%的零样本AP,分别比先前的工作GLIP/GLIPv2/DetCLIP高出14.4/11.4/4.5% AP,甚至大幅超越了其全监督的对应模型。

英文摘要:

This paper presents DetCLIPv2, an efficient and scalable training framework that incorporates large-scale image-text pairs to achieve open-vocabulary object detection (OVD). Unlike previous OVD frameworks that typically rely on a pre-trained vision-language model (e.g., CLIP) or exploit image-text pairs via a pseudo labeling process, DetCLIPv2 directly learns the fine-grained word-region alignment from massive image-text pairs in an end-to-end manner. To accomplish this, we employ a maximum word-region similarity between region proposals and textual words to guide the contrastive objective. To enable the model to gain localization capability while learning broad concepts, DetCLIPv2 is trained with a hybrid supervision from detection, grounding and image-text pair data under a unified data formulation. By jointly training with an alternating scheme and adopting low-resolution input for image-text pairs, DetCLIPv2 exploits image-text pair data efficiently and effectively: DetCLIPv2 utilizes 13X more image-text pairs than DetCLIP with a similar training time and improves performance. With 13M image-text pairs for pre-training, DetCLIPv2 demonstrates superior open-vocabulary detection performance, e.g., DetCLIPv2 with Swin-T backbone achieves 40.4% zero-shot AP on the LVIS benchmark, which outperforms previous works GLIP/GLIPv2/DetCLIP by 14.4/11.4/4.5% AP, respectively, and even beats its fully-supervised counterpart by a large margin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑