TrackMAE: Video Representation Learning via Track Mask and Predict
Organizations: University of Li`ege · KAUST
Abstract
Masked video modeling (MVM) has emerged as a simple and scalable self-supervised pretraining paradigm, but only encodes motion information implicitly, limiting the encoding of temporal dynamics in the learned representations. As a result, such models struggle on motion-centric tasks that require fine-grained motion awareness. To address this, we propose TrackMAE, a simple masked video modeling paradigm that explicitly uses motion information as a reconstruction signal. In TrackMAE, we use an off-the-shelf point tracker to sparsely track points in the input videos, generating motion trajectories. Furthermore, we exploit the extracted trajectories to improve random tube masking with a motion-aware masking strategy. We enhance video representations learned in both pixel and feature semantic reconstruction spaces by providing a complementary supervision signal in the form of motion targets. We evaluate on six datasets across diverse downstream settings and find that TrackMAE consistently outperforms state-of-the-art video self-supervised learning baselines, learning more discriminative and generalizable representations. Code available at https://github.com/rvandeghen/TrackMAE
Figures & tables
| Method | Target | Spatial-centric | Motion-centric | ||
| K400 | HMDB | SSv2 | GYM | ||
| Pixel Reconstruction | |||||
| VideoMAE [ 49 ] | Pixel | 20.7 | 37.7 | 17.5 | 23.9 |
| MVD [ 51 ] | Pixel | 18.7 | 28.6 | 12.2 | 22.7 |
| MGMAE [ 23 ] | Pixel | 24.9 | 41.3 | 16.8 | 26.1 |
| EVEREST [ 25 ] | Pixel | 14.1 | 30.3 | 14.5 | 23.3 |
| Method | Backbone | Targets | SSv2 Top-1 | K400 Top-1 |
| VideoMAE [ 49 ] | ViT-B | Pixel | 68.5 | 80.0 |
| OmniMAE [ 17 ] | ViT-B | Pixel | 69.0 | 80.8 |
| MGM [ 14 ] | ViT-B | Pixel | 71.1 | 80.8 |
| MGMAE [ 23 ] | ViT-B | Pixel | 68.9 | 81.2 |
| TrackMAE (ours) | ViT-B | Pixel | 70.1 | 80.8 |
| MME [ 45 ] | ViT-B | HOG | 70.5 | 81.5 |
| Reconstruction Targets | ||
| Trajectory only | 46.5 | 53.1 |
| Pixels | 46.0 | 52.2 |
| Pixels + Trajectory | 48.9 | 55.7 |
| CLIP | 52.7 | 57.1 |
| CLIP + Trajectory | 55.8 | 61.1 |
| Target | Masking | ||
| Pixel | Tube | 46.0 | 52.2 |
| Pixel | Motion-aware | 46.6 | 52.6 |
| Pixel + Trajectory | Tube | 48.9 | 55.7 |
| Pixel + Trajectory | Motion-aware | 49.4 | 56.2 |
| Grid Size | Upsampling | ||
| None | 48.9 | 55.7 | |
| None | 49.5 | 56.7 | |
| None | 50.0 | 57.0 | |
| ( ) | 50.6 | 57.6 | |
| ( ) | 50.4 | 57.4 |
| Method | Domain shift | Sample efficiency ( ) | Action granularity | Task shift | ||||
| SSv2 | Gym99 | UCF | GYM | FX-S1 | UB-S1 | UCF-RC | Charades | |
| Pixel Reconstruction | ||||||||
| VideoMAE [ 49 ] | 68.6 | 86.6 | 74.6 | 25.9 | 42.8 | 65.3 | 0.172 | 17.8 |
| MVD [ 51 ] | 70.0 | 82.5 | 67.1 | 17.5 | 31.3 | 50.5 | 0.184 | 16.1 |
| MGMAE [ 23 ] | 68.9 | 87.2 | 77.2 | 24.1 | 33.7 | 79.5 | 0.181 | 17.9 |
| MGM [ 14 ] | 71.1 | 89.1 | 78.4 | 26.4 | 38.6 | 86.9 | 0.152 | 22.5 |
| SSv2 Pretraining | K400 Pretraining | |||||
| Method | Backbone | Targets | Epochs | SSv2 Top-1 | SSv2 Top-1 | K400 Top-1 |
| Supervised | ||||||
| Mformer [ 37 ] | Mformer-B | - | - | - | 66.7 | 79.7 |
| VideoSwin [ 34 ] | Swin-B | - | - | - | 69.6 | 80.6 |
| TimeSformer [ 4 ] | ViT-B | - | - | - | 59.5 | 80.7 |
| MViTv1 [ 15 ] | MViTv1-B | - | - | - | 67.7 | 80.2 |
| Shared | ||
| Optimizer | AdamW | |
| Base learning rate | ||
| Weight decay | 0.05 | |
| Momentum (Betas) | ||
| Batch size | 512 | |
| LR schedule | cosine decay | |
| config | K400 | HMDB | SSv2 | GYM | |
| optimizer | AdamW | ||||
| base learning rate | 1 10 -3 | ||||
| weight decay | 0.05 | ||||
| optimizer momentum | |||||
| layer-wise lr decay | 0.75 | ||||
| batch size | 128 | ||||
| Parameter | Value | ||
| Optimizer | AdamW | ||
| Base learning rate | |||
| Weight decay | 0.05 | ||
| Momentum (Betas) | |||
| Layer-wise LR decay | 0.75 | ||
| LR schedule | cosine decay | ||
| Dataset | Abbrev. | #Classes | #Train | #Test |
| Kinetics - 400 | K400 | 400 | 240K | 19K |
| UCF - 101 | UCF | 101 | 9.5K | 3.8K |
| HMDB - 51 | HMDB | 51 | 4.8K | 2K |
| Something - Something V2 | SSv2 | 174 | 169K | 24.8K |
| FineGYM | GYM | 99 | 20.5K | 8.5K |
| Dataset | Experiment | Setup Group | Task | #Classes | Finetune | Test | Metric |
| FineGym [ 42 ] | Gym99 | Full | Action Class. | 99 | 20,484 | 8,521 | Top-1 Acc. |
| UCF 101 [ 44 ] | UCF ( ) | Sample Efficiency | Action Class. | 101 | 1,000 | 3,783 | Top-1 Acc. |
| FineGym [ 42 ] | Gym ( ) | Sample Efficiency | Action Class. | 99 | 1,000 | 8,521 | Top-1 Acc. |
| FineGym [ 42 ] | FX - S1 | Action Granularity | Action Class. | 11 | 1,882 | 777 | Mean-per-class |
| FineGym [ 42 ] | UB - S1 | Action Granularity | Action Class. | 15 | 3,511 | 1,471 | Mean-per-class |
| UCFRep [ 57 ] | UCF - RC | Task Shift | Repetition Counting | – | 421 | 105 | Mean Error |
Video Representation Learning
Video representation learning develops features that capture appearance, motion, and temporal structure for recognition, prediction, and control.
See all 143 papers