🐍VideoMamba

VideoMamba: State Space Model for Efficient Video Understanding

Kunchang Li, Xinhao Li, Yi Wang, Yinan He, Yali Wang, Limin Wang and Yu Qiao

Update

⚠️ 2024/03/25: ~~The current video models are fine-tuned without layer decay due to the bug, which may help to improve the performances as in MAE. We have fixed the bug but do not plan to retrain them.~~ We have applied it for VideoMamba-M but it does not help.
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

Introduction

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.

Cite

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}
}

License

This project is released under the Apache 2.0 license

Acknowledgement

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