融合视觉变压器与图卷积,提升视频目标检测精度与速度
TGBFormer: Transformer-GraphFormer Blender Network for Video Object Detection
- 用时空变换器捕捉全局上下文,用图结构建模局部时空关系
- 在ImageNet VID上达86.5% mAP,单卡运行约41.0 FPS
- 适合需要兼顾检测精度与实时性的视频分析场景
近年来,视频目标检测得益于卷积神经网络(CNN)和视觉变换器(ViT)的发展取得了显著进展。通常,CNN擅长捕捉局部特征但难以建模全局表示;而ViT能有效捕捉长距离全局特征,却在局部细节表达上存在不足。现有方法仅依赖CNN或ViT进行特征聚合,无法同时利用全局与局部信息,导致检测性能受限。本文提出一种用于视频目标检测的Transformer-GraphFormer Blender Network(TGBFormer),通过三项关键技术改进:首先,设计时空变换器模块以聚合全局上下文信息,构建具有长程依赖的全局表示;其次,引入时空GraphFormer模块,利用局部时空关系进行特征聚合,生成互补于变换器输出的局部表示;最后,设计全局-局部特征融合模块,自适应地结合基于变换器的全局表示与基于GraphFormer的局部表示。大量实验表明,TGBFormer在ImageNet VID数据集上达到新纪录,实现86.5% mAP,且在单张Tesla A100 GPU上运行速度约为41.0 FPS。
原文摘要 · Abstract (English)
Video object detection has made significant progress in recent years thanks to convolutional neural networks (CNNs) and vision transformers (ViTs). Typically, CNNs excel at capturing local features but struggle to model global representations. Conversely, ViTs are adept at capturing long-range global features but face challenges in representing local feature details. Off-the-shelf video object detection methods solely rely on CNNs or ViTs to conduct feature aggregation, which hampers their capability to simultaneously leverage global and local information, thereby resulting in limited detection performance. In this paper, we propose a Transformer-GraphFormer Blender Network (TGBFormer) for video object detection, with three key technical improvements to fully exploit the advantages of transformers and graph convolutional networks while compensating for their limitations. First, we develop a spatial-temporal transformer module to aggregate global contextual information, constituting global representations with long-range feature dependencies. Second, we introduce a spatial-temporal GraphFormer module that utilizes local spatial and temporal relationships to aggregate features, generating new local representations that are complementary to the transformer outputs. Third, we design a global-local feature blender module to adaptively couple transformer-based global representations and GraphFormer-based local representations. Extensive experiments demonstrate that our TGBFormer establishes new state-of-the-art results on the ImageNet VID dataset. Particularly, our TGBFormer achieves 86.5% mAP while running at around 41.0 FPS on a single Tesla A100 GPU.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。