• 2024/03/13: Fix some bugs and add 🤗HF model links.
• 🔥 2024/03/12: All the code and models are released.
  • image_sm: Single-modality Image Tasks
    • Image Classification: script & model
  • video_sm: Single-modality Video Tasks
    • Short-term Video Understanding: script & model
    • Long-term Video Understanding: script & model
    • Masked Modeling: script, model
  • video_mm: Multi-modality Video Tasks
    • Video-Text Retrieval: script & model

Addressing the dual challenges of local redundancy and global dependencies in video understanding, this work innovatively adapts the Mamba to the video domain. The proposed VideoMamba overcomes the limitations of existing 3D convolution neural networks and video transformers. Its linear-complexity operator enables efficient long-term modeling, which is crucial for high-resolution long video understanding. Extensive evaluations reveal VideoMamba's four core abilities: (1) Scalability in the visual domain without extensive dataset pretraining, thanks to a novel self-distillation technique; (2) Sensitivity for recognizing short-term actions even with fine-grained motion differences; (3) Superiority in long-term video understanding, showcasing significant advancements over traditional feature-based models; and (4) Compatibility with other modalities, demonstrating robustness in multi-modal contexts. Through these distinct advantages, VideoMamba sets a new benchmark for video understanding, offering a scalable and efficient solution for comprehensive video understanding.

If you find this repository useful, please use the following BibTeX entry for citation.

@misc{li2024videomamba,
      title={VideoMamba: State Space Model for Efficient Video Understanding}, 
      author={Kunchang Li and Xinhao Li and Yi Wang and Yinan He and Yali Wang and Limin Wang and Yu Qiao},
      year={2024},
      eprint={2403.06977},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

This project is released under the Apache 2.0 license

This repository is built based on UniFormer, Unmasked Teacher and Vim repository.