Scaling 4D Representations

João Carreira, Dilara Gokay, Michael King, Chuhan Zhang, Ignacio Rocco +30 more
12/19/2024
cs.CVcs.AIcs.LG

Official code: google-deepmind/representations4d (174 stars, Apache-2.0, last commit 2026-03-07).

Abstract

Scaling has not yet been convincingly demonstrated for pure self-supervised learning from video. However, prior work has focused evaluations on semantic-related tasks $\unicode{x2013}$ action classification, ImageNet classification, etc. In this paper we focus on evaluating self-supervised learning on non-semantic vision tasks that are more spatial (3D) and temporal (+1D = 4D), such as camera pose estimation, point and object tracking, and depth estimation. We show that by learning from very large video datasets, masked auto-encoding (MAE) with transformer video models actually scales, consistently improving performance on these 4D tasks, as model size increases from 20M all the way to the largest by far reported self-supervised video model $\unicode{x2013}$ 22B parameters. Rigorous apples-to-apples comparison with many recent image and video models demonstrates the benefits of scaling 4D representations. Pretrained models are available at https://github.com/google-deepmind/representations4d .

View on arXivView PDF

Code Implementations(1)

Apache-2.0
1749Python, Jupyter NotebookJun 23, 20257 months agoApache-2.0

Similar Papers

Cite this paper

@article{carreira2024scaling,
  title  = {Scaling 4D Representations},
  author = {João Carreira and Dilara Gokay and Michael King and Chuhan Zhang and Ignacio Rocco and Aravindh Mahendran and Thomas Albert Keck and Joseph Heyward and Skanda Koppula and Etienne Pot and Goker Erdogan and Yana Hasson and Yi Yang and Klaus Greff and Guillaume Le Moing and Sjoerd van Steenkiste and Daniel Zoran and Drew A. Hudson and Pedro Vélez and Luisa Polanía and Luke Friedman and Chris Duvarney and Ross Goroshin and Kelsey Allen and Jacob Walker and Rishabh Kabra and Eric Aboussouan and Jennifer Sun and Thomas Kipf and Carl Doersch and Viorica Pătrăucean and Dima Damen and Pauline Luc and Mehdi S. M. Sajjadi and Andrew Zisserman},
  year   = {2024},
  eprint = {2412.15212},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.15212v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.15212.svg)](https://codeofpaper.com/paper/2412.15212)

Discussion