让开放词汇目标检测模型更准识别物体属性
HA-FGOVD: Highlighting Fine-grained Attributes via Explicit Linear Composition for Open-Vocabulary Object Detection
- 通过显式线性组合突出文本中细粒度属性词
- 在FG-OVD数据集上统一提升多种主流模型性能
- 方法通用可迁移,适合需属性识别的场景
开放词汇目标检测(OVD)模型因训练数据广泛、参数量大,被视为大型多模态模型。主流OVD模型侧重物体粗粒度类别,忽略颜色、材质等细粒度属性,导致无法精准识别带属性描述的物体。然而,这些模型在大规模图文对上预训练,其隐空间可将全局文本特征表示为细粒度属性标记的线性组合,但未显式强调属性。为此,本文提出一种适用于冻结主流OVD模型的通用显式方法:首先利用大语言模型在输入文本中零样本提示突出属性词;其次通过策略性调整标记掩码,使文本编码器提取全局与属性特异性特征,并在显式线性空间中组合成新特征向量用于检测,对应标量可手工设定或学习。值得注意的是,这些标量可在不同OVD模型间无缝迁移,证明该线性组合具有普适性。在FG-OVD数据集上的实证表明,本方法统一提升了多种主流模型的细粒度属性级检测能力,达到新最佳性能。
原文摘要 · Abstract (English)
Open-vocabulary object detection (OVD) models are considered to be Large Multi-modal Models (LMM), due to their extensive training data and a large number of parameters. Mainstream OVD models prioritize object coarse-grained category rather than focus on their fine-grained attributes, e.g., colors or materials, thus failed to identify objects specified with certain attributes. However, OVD models are pretrained on large-scale image-text pairs with rich attribute words, whose latent feature space can represent the global text feature as a linear composition of fine-grained attribute tokens without highlighting them. Therefore, we propose in this paper a universal and explicit approach for frozen mainstream OVD models that boosts their attribute-level detection capabilities by highlighting fine-grained attributes in explicit linear space. Firstly, a LLM is leveraged to highlight attribute words within the input text as a zero-shot prompted task. Secondly, by strategically adjusting the token masks, the text encoders of OVD models extract both global text and attribute-specific features, which are then explicitly composited as two vectors in linear space to form the new attribute-highlighted feature for detection tasks, where corresponding scalars are hand-crafted or learned to reweight both two vectors. Notably, these scalars can be seamlessly transferred among different OVD models, which proves that such an explicit linear composition is universal. Empirical evaluation on the FG-OVD dataset demonstrates that our proposed method uniformly improves fine-grained attribute-level OVD of various mainstream models and achieves new state-of-the-art performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。