Training a Large Video Model on a Single Machine in a Day
Not on the map. Why?
Abstract (arXiv)
Videos are big, complex to pre-process, and slow to train on. State-of-the-art large-scale video models are trained on clusters of 32 or more GPUs for several days. As a consequence, academia largely ceded the training of large video models to industry. In this paper, we show how to still train a state-of-the-art video model on a single machine with eight consumer-grade GPUs in a day. We identify three bottlenecks, IO, CPU, and GPU computation, and optimize each. The result is a highly efficient video training pipeline. For comparable architectures, our pipeline achieves higher accuracies with frac18 of the computation compared to prior work. Code is available at https://github.com/zhaoyue-zephyrus/AVION.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Builds on 80 of its 87 references that are in the map. First posted on arXiv 28 Sep 2023.
Builds on
80 of 87 references.
References · 80
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2014Going Deeper with Convolutionscited by 4,231
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
38 more
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2017Quo Vadis, Action Recognition? A New Model and the Kinetics Datasetcited by 3,259
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2018SlowFast Networks for Video Recognitioncited by 1,862
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2020Reformer: The Efficient Transformercited by 1,658
- 2021ViViT: A Video Vision Transformercited by 1,610
- 2021Is Space-Time Attention All You Need for Video Understanding?cited by 1,609
- 2020Rethinking Attention with Performerscited by 1,584
- 2021Ego4D: Around the World in 3,000 Hours of Egocentric Videocited by 1,530
- 2022VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Trainingcited by 1,305
- 2022CoCa: Contrastive Captioners are Image-Text Foundation Modelscited by 1,265
- 2021Frozen in Time: A Joint Video and Image Encoder for End-to-End Retrievalcited by 1,175
- 2021Video Swin Transformercited by 1,090
- 2014Two-Stream Convolutional Networks for Action Recognition in Videoscited by 1,067
- 2019HowTo100M: Learning a Text-Video Embedding by Watching Hundred Million Narrated Video Clipscited by 946
- 2016Temporal Segment Networks: Towards Good Practices for Deep Action Recognitioncited by 899
- 2021Multiscale Vision Transformerscited by 896
- 2018TSM: Temporal Shift Module for Efficient Video Understandingcited by 852
- 2016Training Deep Nets with Sublinear Memory Costcited by 825
- 2021Florence: A New Foundation Model for Computer Visioncited by 802
- 2022Exploring Plain Vision Transformer Backbones for Object Detectioncited by 723
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
- 2020X3D: Expanding Architectures for Efficient Video Recognitioncited by 635
- 2021VideoCLIP: Contrastive Pre-training for Zero-shot Video-Text Understandingcited by 596
- 2020Rescaling Egocentric Visioncited by 529
- 2017Rethinking Spatiotemporal Feature Learning: Speed-Accuracy Trade-offs in Video Classificationcited by 521
- 2019Quantifying the Carbon Emissions of Machine Learningcited by 516
- 2021MViTv2: Improved Multiscale Vision Transformers for Classification and Detectioncited by 516
- 2019End-to-End Learning of Visual Representations from Uncurated Instructional Videoscited by 506
- 2022InternVideo: General Video Foundation Models via Generative and Discriminative Learningcited by 441
- 2022Masked Autoencoders As Spatiotemporal Learnerscited by 428
- 2022DeiT III: Revenge of the ViTcited by 417
- 2019Video Classification with Channel-Separated Convolutional Networkscited by 290
- 2020Self-Supervised MultiModal Versatile Networkscited by 276
- 2017Temporal Action Detection with Structured Segment Networkscited by 265
50 most cited of 80 All 80 (JSON)
Built on by
Citing papers per year: 2024: 11; 2025: 8; 2026: 1.
20 papers.
Sort:
By area:
- 2024Video Mamba Suite: State Space Model as a Versatile Alternative for Video Understandingcited by 106
- 2024Foundation Models for Video Understanding: A Surveycited by 46
- 2024EgoVideo: Exploring Egocentric Foundation Model and Downstream Adaptationcited by 44
- 2024Don't Look Twice: Faster Video Transformers with Run-Length Tokenizationcited by 42
- 2025Exo2Ego: Exocentric Knowledge Guided MLLM for Egocentric Video Understandingcited by 25
- 2024Distilling Vision-Language Models on Millions of Videoscited by 20
- 2025Modeling Fine-Grained Hand-Object Dynamics for Egocentric Video Representation Learningcited by 20
- 2024Vinci: A Real-time Embodied Smart Assistant based on Egocentric Vision-Language Modelcited by 19
- 2025EgoThinker: Unveiling Egocentric Reasoning with Spatio-Temporal CoTcited by 16
- 2024TIM: A Time Interval Machine for Audio-Visual Action Recognitioncited by 15
- 2024Mamba Fusion: Learning Actions Through Questioningcited by 5
- 2025EgoDTM: Towards 3D-Aware Egocentric Video-Language Pretrainingcited by 5
8 more
- 2024Extending Video Masked Autoencoders to 128 framescited by 3
- 2025LLaVAction: evaluating and training multi-modal large language models for action understandingcited by 3
- 2025EVA02-AT: Egocentric Video-Language Understanding with Spatial-Temporal Rotary Positional Embeddings and Symmetric Optimizationcited by 3
- 2025D2E: Scaling Vision-Action Pretraining on Desktop Data for Transfer to Embodied AIcited by 3
- 2024Symmetric Multi-Similarity Loss for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 2024cited by 2
- 2024FILS: Self-Supervised Video Feature Prediction In Semantic Language Spacecited by 1
- 2026MM-TS: Multi-Modal Temperature and Margin Schedules for Contrastive Learning with Long-Tail Datacited by 1
- 2025ContextRefine-CLIP for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 2025cited by 0
- 2026MM-TS: Multi-Modal Temperature and Margin Schedules for Contrastive Learning with Long-Tail Data9 Mar 2026
- 2025EgoThinker: Unveiling Egocentric Reasoning with Spatio-Temporal CoT27 Oct 2025
- 2025D2E: Scaling Vision-Action Pretraining on Desktop Data for Transfer to Embodied AI7 Oct 2025
- 2025EVA02-AT: Egocentric Video-Language Understanding with Spatial-Temporal Rotary Positional Embeddings and Symmetric Optimization17 Jun 2025
- 2025ContextRefine-CLIP for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 202512 Jun 2025
- 2025LLaVAction: evaluating and training multi-modal large language models for action understanding24 Mar 2025
- 2025EgoDTM: Towards 3D-Aware Egocentric Video-Language Pretraining19 Mar 2025
- 2025Exo2Ego: Exocentric Knowledge Guided MLLM for Egocentric Video Understanding12 Mar 2025
- 2025Modeling Fine-Grained Hand-Object Dynamics for Egocentric Video Representation Learning2 Mar 2025
- 2024Vinci: A Real-time Embodied Smart Assistant based on Egocentric Vision-Language Model30 Dec 2024
- 2024Extending Video Masked Autoencoders to 128 frames20 Nov 2024
- 2024Don't Look Twice: Faster Video Transformers with Run-Length Tokenization7 Nov 2024
8 more
- 2024Mamba Fusion: Learning Actions Through Questioning17 Sep 2024
- 2024EgoVideo: Exploring Egocentric Foundation Model and Downstream Adaptation26 Jun 2024
- 2024Symmetric Multi-Similarity Loss for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 202418 Jun 2024
- 2024FILS: Self-Supervised Video Feature Prediction In Semantic Language Space5 Jun 2024
- 2024Foundation Models for Video Understanding: A Survey6 May 2024
- 2024TIM: A Time Interval Machine for Audio-Visual Action Recognition8 Apr 2024
- 2024Video Mamba Suite: State Space Model as a Versatile Alternative for Video Understanding14 Mar 2024
- 2024Distilling Vision-Language Models on Millions of Videos11 Jan 2024
- 2024Foundation Models for Video Understanding: A Surveycited by 46
- 2024EgoVideo: Exploring Egocentric Foundation Model and Downstream Adaptationcited by 44
- 2024Don't Look Twice: Faster Video Transformers with Run-Length Tokenizationcited by 42
- 2025Exo2Ego: Exocentric Knowledge Guided MLLM for Egocentric Video Understandingcited by 25
- 2025Modeling Fine-Grained Hand-Object Dynamics for Egocentric Video Representation Learningcited by 20
- 2024TIM: A Time Interval Machine for Audio-Visual Action Recognitioncited by 15
- 2025EgoDTM: Towards 3D-Aware Egocentric Video-Language Pretrainingcited by 5
- 2024Extending Video Masked Autoencoders to 128 framescited by 3
- 2025EVA02-AT: Egocentric Video-Language Understanding with Spatial-Temporal Rotary Positional Embeddings and Symmetric Optimizationcited by 3
- 2025D2E: Scaling Vision-Action Pretraining on Desktop Data for Transfer to Embodied AIcited by 3
- 2024Symmetric Multi-Similarity Loss for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 2024cited by 2
- 2024FILS: Self-Supervised Video Feature Prediction In Semantic Language Spacecited by 1
12 most cited of 14 All 14 (JSON)
- 2026MM-TS: Multi-Modal Temperature and Margin Schedules for Contrastive Learning with Long-Tail Data9 Mar 2026
- 2025D2E: Scaling Vision-Action Pretraining on Desktop Data for Transfer to Embodied AI7 Oct 2025
- 2025EVA02-AT: Egocentric Video-Language Understanding with Spatial-Temporal Rotary Positional Embeddings and Symmetric Optimization17 Jun 2025
- 2025ContextRefine-CLIP for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 202512 Jun 2025
- 2025EgoDTM: Towards 3D-Aware Egocentric Video-Language Pretraining19 Mar 2025
- 2025Exo2Ego: Exocentric Knowledge Guided MLLM for Egocentric Video Understanding12 Mar 2025
- 2025Modeling Fine-Grained Hand-Object Dynamics for Egocentric Video Representation Learning2 Mar 2025
- 2024Extending Video Masked Autoencoders to 128 frames20 Nov 2024
- 2024Don't Look Twice: Faster Video Transformers with Run-Length Tokenization7 Nov 2024
- 2024EgoVideo: Exploring Egocentric Foundation Model and Downstream Adaptation26 Jun 2024
- 2024Symmetric Multi-Similarity Loss for EPIC-KITCHENS-100 Multi-Instance Retrieval Challenge 202418 Jun 2024
- 2024FILS: Self-Supervised Video Feature Prediction In Semantic Language Space5 Jun 2024
12 newest of 14 All 14 (JSON)