ARVideo: Autoregressive Pretraining for Self-Supervised Video Representation Learning

التفاصيل البيبلوغرافية
العنوان: ARVideo: Autoregressive Pretraining for Self-Supervised Video Representation Learning
المؤلفون: Ren, Sucheng, Zhu, Hongru, Wei, Chen, Li, Yijiang, Yuille, Alan, Xie, Cihang
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computer Vision and Pattern Recognition
الوصف: This paper presents a new self-supervised video representation learning framework, ARVideo, which autoregressively predicts the next video token in a tailored sequence order. Two key designs are included. First, we organize autoregressive video tokens into clusters that span both spatially and temporally, thereby enabling a richer aggregation of contextual information compared to the standard spatial-only or temporal-only clusters. Second, we adopt a randomized spatiotemporal prediction order to facilitate learning from multi-dimensional data, addressing the limitations of a handcrafted spatial-first or temporal-first sequence order. Extensive experiments establish ARVideo as an effective paradigm for self-supervised video representation learning. For example, when trained with the ViT-B backbone, ARVideo competitively attains 81.2% on Kinetics-400 and 70.9% on Something-Something V2, which are on par with the strong benchmark set by VideoMAE. Importantly, ARVideo also demonstrates higher training efficiency, i.e., it trains 14% faster and requires 58% less GPU memory compared to VideoMAE.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2405.15160
رقم الأكسشن: edsarx.2405.15160
قاعدة البيانات: arXiv