端到端检测视频中未见物体的关系,提升开放词汇场景下的泛化能力。
End-to-end Open-vocabulary Video Visual Relationship Detection using Multi-modal Prompting
- 统一轨迹检测与关系分类,用关系查询增强模型对对象间关系的感知。
- 在VidVRD和VidOR上达到新高,跨数据集测试仍保持稳定性能。
- 适合需要处理未知物体和关系的视频理解研究者使用。
开放词汇视频视觉关系检测旨在突破标注类别的限制,识别视频中已知与未知对象间的未见关系。现有方法通常依赖在封闭数据集上训练的轨迹检测器获取物体轨迹,再输入大规模预训练视觉语言模型进行开放词汇分类,严重依赖预训练轨迹检测器,导致对新类别泛化能力不足,性能下降。为解决此问题,本文提出一个端到端的开放词汇框架,将物体轨迹检测与关系分类统一。框架包含一种关系感知的开放词汇轨迹检测器,基于查询的Transformer解码器,利用CLIP视觉编码器实现帧级开放词汇物体检测,并引入轨迹关联模块。通过在解码器中嵌入关系查询并设计辅助关系损失,使模型显式感知对象间关系。此外,提出一种开放词汇关系分类器,借助CLIP丰富的语义知识发现新关系。设计多模态提示方法:时空视觉提示用于视觉表征,视觉引导语言提示用于语言输入。在两个公开数据集VidVRD和VidOR上的大量实验验证了框架有效性。该框架还应用于更具挑战性的跨数据集场景,进一步证明其泛化能力。
原文摘要 · Abstract (English)
Open-vocabulary video visual relationship detection aims to expand video visual relationship detection beyond annotated categories by detecting unseen relationships between both seen and unseen objects in videos. Existing methods usually use trajectory detectors trained on closed datasets to detect object trajectories, and then feed these trajectories into large-scale pre-trained vision-language models to achieve open-vocabulary classification. Such heavy dependence on the pre-trained trajectory detectors limits their ability to generalize to novel object categories, leading to performance degradation. To address this challenge, we propose to unify object trajectory detection and relationship classification into an end-to-end open-vocabulary framework. Under this framework, we propose a relationship-aware open-vocabulary trajectory detector. It primarily consists of a query-based Transformer decoder, where the visual encoder of CLIP is distilled for frame-wise open-vocabulary object detection, and a trajectory associator. To exploit relationship context during trajectory detection, a relationship query is embedded into the Transformer decoder, and accordingly, an auxiliary relationship loss is designed to enable the decoder to perceive the relationships between objects explicitly. Moreover, we propose an open-vocabulary relationship classifier that leverages the rich semantic knowledge of CLIP to discover novel relationships. To adapt CLIP well to relationship classification, we design a multi-modal prompting method that employs spatio-temporal visual prompting for visual representation and vision-guided language prompting for language input. Extensive experiments on two public datasets, VidVRD and VidOR, demonstrate the effectiveness of our framework. Our framework is also applied to a more difficult cross-dataset scenario to further demonstrate its generalization ability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。