arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2309.09534cs.CV

面向视频动作识别的选择性体积混合

Selective Volume Mixup for Video Action Recognition

University of Science and Technology of China(中国科学技术大学)

Yi Tan, Zhaofan Qiu, Yanbin Hao, Ting Yao, Tao Mei

AI总结:

本文提出选择性体积混合(SV-Mix)视频增强策略,通过可学习的空间和时间选择性模块从两个视频中选取信息量最大的体积进行混合,以缓解小规模视频数据上的过拟合问题,并在多个基准上一致提升CNN和Transformer模型的视频动作识别性能。

摘要:

卷积神经网络(CNNs)和视觉Transformer的最新进展令人信服地证明了它们在大规模数据集上进行视频动作识别时具有很高的学习能力。然而,深度模型在训练视频数量有限的小规模数据集上往往会出现过拟合问题。一种常见的解决方案是分别对每一帧使用现有的图像增强策略,包括Mixup、Cutmix和RandAugment,但这些策略并不是专门针对视频数据优化的。在本文中,我们提出了一种新颖的视频增强策略,名为选择性体积混合(SV-Mix),以提高深度模型在训练视频有限情况下的泛化能力。SV-Mix设计了一个可学习的选择性模块,从两个视频中选择信息量最大的体积,并将这些体积混合起来生成新的训练视频。在技术实现上,我们提出了两个新模块,即用于为每个空间位置选择局部块的空间选择性模块,以及用于为每个时间戳混合整个帧并保持空间模式的时间选择性模块。在每一步中,我们随机选择其中一个模块来扩展训练样本的多样性。选择性模块与视频动作识别框架联合优化,以找到最优的增强策略。我们在广泛的视频动作识别基准上实证展示了SV-Mix增强的优势,并一致提升了基于CNN和基于Transformer的模型的性能。

英文摘要:

The recent advances in Convolutional Neural Networks (CNNs) and Vision Transformers have convincingly demonstrated high learning capability for video action recognition on large datasets. Nevertheless, deep models often suffer from the overfitting effect on small-scale datasets with a limited number of training videos. A common solution is to exploit the existing image augmentation strategies for each frame individually including Mixup, Cutmix, and RandAugment, which are not particularly optimized for video data. In this paper, we propose a novel video augmentation strategy named Selective Volume Mixup (SV-Mix) to improve the generalization ability of deep models with limited training videos. SV-Mix devises a learnable selective module to choose the most informative volumes from two videos and mixes the volumes up to achieve a new training video. Technically, we propose two new modules, i.e., a spatial selective module to select the local patches for each spatial position, and a temporal selective module to mix the entire frames for each timestamp and maintain the spatial pattern. At each time, we randomly choose one of the two modules to expand the diversity of training samples. The selective modules are jointly optimized with the video action recognition framework to find the optimal augmentation strategy. We empirically demonstrate the merits of the SV-Mix augmentation on a wide range of video action recognition benchmarks and consistently boot the performances of both CNN-based and transformer-based models.

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

↑