SIGMA
Paper link : on Huggingface :
Fonte do modelo
- Tipo
- Other
- Tamanho do arquivo
- 329 MB–1,10 GB
- Versão
- vRevision 520b55b
- Atualizado
- 19 de set. de 2026
- Downloads
- 0
- Fonte
- Hugging Face
Descrição da fonte
Official PyTorch Implementation of SIGMA(ECCV 2024).
Paper link : https://arxiv.org/html/2407.15447v1 on Huggingface : https://huggingface.co/papers/2407.15447
SIGMA Framework
🔥 Sinkhorn-Guided Masked Video Modeling
Video-based pretraining offers immense potential for learning strong visual representations on an unprecedented scale. Recently, masked video modeling methods have shown promising scalability, yet fall short in capturing higher-level semantics due to reconstructing predefined low-level targets such as pixels. To tackle this, we present Sinkhorn-guided Masked Video Modelling (SIGMA), a novel video pretraining method that jointly learns the video model in addition to a target feature space using a projection network. However, this simple modification means that the regular L2 reconstruction loss will lead to trivial solutions as both networks are jointly optimized. As a solution, we distribute features of space-time tubes evenly across a limited number of learnable clusters. By posing this as an optimal transport problem, we enforce high entropy in the generated features across the batch, infusing semantic and temporal meaning into the feature space. The resulting cluster assignments are used as targets for a symmetric prediction task where the video model predicts cluster assignment of the projection network and vice versa. Experimental results on ten datasets across three benchmarks validate the effectiveness of SIGMA in learning more performant, temporally-aware, and robust video representations improving upon state-of-the-art methods.
✨ Something-Something V2
| Method | Extra Data | Backbone | Resolution | #Frames x Clips x Crops | Epoch | Top-1 |
|---|---|---|---|---|---|---|
| VideoMAE | no | ViT-S | 224x224 | 16x2x3 | 2400 | 66.8 |
| VideoMAE | no | ViT-B | 224x224 | 16x2x3 | 800 | 69.6 |
| SIGMA | Img-1k | ViT-S | 224x224 | 16x2x3 | 2400 | 68.6 |
| SIGMA | Img-1k | ViT-B | 224x224 | 16x2x3 | 800 | 70.9 |
✨ Kinetics-400
| Method | Extra Data | Backbone | Resolution | #Frames x Clips x Crops | Epoch | Top-1 |
|---|---|---|---|---|---|---|
| VideoMAE | no | ViT-S | 224x224 | 16x5x3 | 1600 | 79.0 |
| VideoMAE | no | ViT-B | 224x224 | 16x5x3 | 800 | 80.0 |
| SIGMA | Img-1k | ViT-S | 224x224 | 16x5x3 | 800 | 79.4 |
| SIGMA | Img-1k | ViT-B | 224x224 | 16x5x3 | 800 | 81.6 |
🔨 Installation
Please follow the instructions in INSTALL.md.
➡️ Data Preparation
Please follow the instructions in DATASET.md for data preparation.
🔄 Pre-training
The pre-training instruction is in PRETRAIN.md.
⤴️ Fine-tuning with pre-trained models
The fine-tuning instruction is in FINETUNE.md.
📍Model Zoo
You can fined our pretrained and finetuned models in the files.
⚠️ Our code is based on VideoMAE code base.
✏️ Citation
If you think this project is helpful, please feel free to leave a star⭐️ and cite our paper:
@inproceedings{salehi2024sigma,
title={SIGMA: Sinkhorn-Guided Masked Video Modeling},
author={Salehi, Mohammadreza and Dorkenwald, Michael and Thoker, Fida Mohammad and Gavves, Efstratios and Snoek, Cees GM and Asano, Yuki M},
journal={European Conference of Computer Vision},
year={2024}
}
Fontes
1 fonte
Fontes
1 fonteVerificado 5 de set.
Artefatos de modelo
7 artefatos
Artefatos de modelo
7 artefatosk400_finetune_wth_Kinetics_Pretraining_Dino_multi_node_epoch_800.pth
pth · 330 MB · SHA-256 09056b8a3077…9336 · Hugging Face
Baixarpretrain/kinetics-400/kinetics_400_vit_b_sigma_with_dino.pth
pth · 1,08 GB · SHA-256 c9130cdae5bc…5550 · Hugging Face
Baixarpretrain/kinetics-400/kinetics_400_vit_b_sigma_with_mlp.pth
pth · 1,10 GB · SHA-256 df3d632daadc…f428 · Hugging Face
Baixarpretrain/ssv2/ssv2_vit_b_sigma_with_dino_and_mgmae_mask.pth
pth · 1,04 GB · SHA-256 19604189ebe7…4b1b · Hugging Face
Baixarpretrain/ssv2/ssv2_vit_b_sigma_with_dino.pth
pth · 1,08 GB · SHA-256 a90cbe8907f6…81fb · Hugging Face
Baixarpretrain/ssv2/ssv2_vit_b_sigma_with_mlp.pth
pth · 1,10 GB · SHA-256 1c967ea52a20…40ee · Hugging Face
Baixarssv2_finetune_wth_Kinetics_Pretraining_Dino_multi_node_epoch_800.pth
pth · 329 MB · SHA-256 485d359dbf70…337b · Hugging Face
BaixarTrechos de fonte
2 trechos
Trechos de fonte
2 trechos--- language: - en pipeline_tag: video-classification --- # Official PyTorch Implementation of SIGMA(ECCV 2024). Paper link : https://arxiv.org/html/2407.15447v1 on Huggingface : https://huggingface.co/papers/2407.15447  ### 🔥 Sinkhorn-Guided Masked Video Modeling Video-based pretraining offers immense potential for learning strong visual representations on an unprecedented scale. Recently, masked video modeling methods have shown promising scalability, yet fall short in capturing higher-level semantics due to reconstructing predefined low-level targets such as pixels. To tackle this, we present Sinkhorn-guided Masked Video Modelling (SIGMA), a novel video pretraining method that jointly learns the video model in addition to a target feature space using a projection network. However, this simple modification means that the regular L2 reconstruction loss will lead to trivial solutions as both networks are jointly optimized. As a solution, we distribute features of space-time tubes evenly across a limited number of learnable clusters. By posing this as an optimal transport problem, we enforce high entropy in the generated features across the batch, infusing semantic and temporal meaning into the feature space. The resulting cluster assignments are used as targets for a symmetric prediction task where the video model predicts cluster assignment of the projection network and vice versa. Experimental results on ten datasets across three benchmarks validate the effectiveness of SIGMA in learning more performant, temporally-aware, and robust video representations improving upon state-of-the-art methods. <!-- ### ⚡️ A Simple, Efficient and Strong Baseline in SSVP VideoMAE uses the simple masked autoencoder and **plain ViT** backbone to perform video self-supervised learning. Due to the extremely high masking ratio, the pre-training time of VideoMAE is **much shorter** than contrastive learning methods (**3.2x** speedup). VideoMAE can serve as **a simple but strong baseline** for future research in self-supervised video pre-training. ### 😮 High performance, but NO extra data required VideoMAE works well for video datasets of different scales and can achieve **87.4%** on Kinects-400, **75.4%** on Something-Something V2, **91.3%** on UCF101, and **62.6%** on HMDB51. To our best knowledge, VideoMAE is the **first** to achi...
Source context: 0 downloads · 3 likes · Pipeline video-classification · Repo SMSD75/SIGMA
