ARVideo: Autoregressive Pretraining for Self-Supervised Video Representation Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ren, Sucheng, Zhu, Hongru, Wei, Chen, Li, Yijiang, Yuille, Alan, Xie, Cihang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914809871597568
author Ren, Sucheng
Zhu, Hongru
Wei, Chen
Li, Yijiang
Yuille, Alan
Xie, Cihang
author_facet Ren, Sucheng
Zhu, Hongru
Wei, Chen
Li, Yijiang
Yuille, Alan
Xie, Cihang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2405_15160
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ARVideo: Autoregressive Pretraining for Self-Supervised Video Representation Learning
Ren, Sucheng
Zhu, Hongru
Wei, Chen
Li, Yijiang
Yuille, Alan
Xie, Cihang
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.
title ARVideo: Autoregressive Pretraining for Self-Supervised Video Representation Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.15160