循环视频掩码自编码器
Recurrent Video Masked Autoencoders
用 Transformer 循环网络替代时空注意力做视频掩码自编码,在长时序上以线性计算代价稳定传播特征,可作为视频骨干预训练方案直接替换 VideoMAE/V-JEPA 的编码器。
原文依据 RVM couples an asymmetric masking objective with a transformer-based recurrent neural network to aggregate information over time
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论我们提出循环视频掩码自编码器(RVM):一种利用循环计算建模视频数据时序结构的视频表征学习新方法。RVM 将非对称掩码目标与基于 Transformer 的循环神经网络相结合,以在时间上聚合信息,并且仅使用简单的像素重建损失进行训练。这一设计得到了一种高效的“通用”编码器:RVM 在动作分类、点跟踪与目标跟踪等视频级任务上取得了与最优视频模型(如 VideoMAE、V-JEPA)相当的性能,同时在需要强几何特征与稠密空间特征的任务上达到或超过图像模型(如 DINOv2)的表现。值得注意的是,RVM 在小型模型规模下无需知识蒸馏即可取得强劲性能,相比同类视频掩码自编码器展现出最高 30 倍的参数效率。最后,我们证明 RVM 的循环特性使其能够以线性计算代价在长时序范围内稳定传播特征,从而克服标准时空注意力视频模型的部分局限。消融研究进一步揭示了驱动该模型成功的关键因素,定性结果则表明 RVM 学到了对场景语义、结构与运动的丰富表征。
英文原文摘要 1384 字符
We present Recurrent Video Masked-Autoencoders (RVM): a novel approach to video representation learning that leverages recurrent computation to model the temporal structure of video data. RVM couples an asymmetric masking objective with a transformer-based recurrent neural network to aggregate information over time, training solely on a simple pixel reconstruction loss. This design yields a highly efficient "generalist" encoder: RVM achieves competitive performance with state-of-the-art video models (e.g. VideoMAE, V-JEPA) on video-level tasks like action classification, and point and object tracking, while matching or exceeding the performance of image models (e.g. DINOv2) on tasks that require strong geometric and dense spatial features. Notably, RVM achieves strong performance in the small-model regime without requiring knowledge distillation, exhibiting up to 30xgreater parameter efficiency than competing video masked autoencoders. Finally, we demonstrate that \model's recurrent nature allows for stable feature propagation over long temporal horizons with linear computational cost, overcoming some of the limitations of standard spatio-temporal attention-based video models. Ablation studies further highlight the factors driving the model's success, with qualitative results showing that RVM learns rich representations of scene semantics, structure, and motion.