中文

MaskVD:用于高效视频目标检测的区域掩码

计算机视觉与模式识别 2024-07-18 v1 机器学习

摘要

视频任务计算负载高,因此在资源受限且对延迟敏感的边缘设备上部署具有挑战性,尤其是对于需要最先进视觉 Transformer (ViT) 的任务。 Several research efforts have tried to address this challenge by leveraging the fact that large portions of the video undergo very little change across frames, leading to redundant computations in frame-based video processing. In particular, some works leverage pixel or semantic differences across frames, however, this yields limited latency benefits with significantly increased memory overhead. This paper, in contrast, presents a strategy for masking regions in video frames that leverages the semantic information in images and the temporal correlation between frames to significantly reduce FLOPs and latency with little to no penalty in performance over baseline models. In particular, we demonstrate that by leveraging extracted features from previous frames, ViT backbones directly benefit from region masking, skipping up to 80% of input regions, improving FLOPs and latency by 3.14x and 1.5x. We improve memory and latency over the state-of-the-art (SOTA) by 2.3x and 1.14x, while maintaining similar detection performance. Additionally, our approach demonstrates promising results on convolutional neural networks (CNNs) and provides latency improvements over the SOTA up to 1.3x using specialized computational kernels.

关键词

引用

@article{arxiv.2407.12067,
  title  = {MaskVD: Region Masking for Efficient Video Object Detection},
  author = {Sreetama Sarkar and Gourav Datta and Souvik Kundu and Kai Zheng and Chirayata Bhattacharyya and Peter A. Beerel},
  journal= {arXiv preprint arXiv:2407.12067},
  year   = {2024}
}