Generative Spatiotemporal Data Augmentation
Organizations: University of Chicago · University of Michigan, Ann Arbor · DEVCOM Army Research Laboratory
Abstract
We explore spatiotemporal data augmentation using video foundation models to diversify both camera viewpoints and scene dynamics. Unlike existing approaches based on simple geometric transforms or appearance perturbations, our method leverages off-the-shelf video diffusion models to generate realistic 3D spatial and temporal variations from a given image dataset. Incorporating these synthesized video clips as supplemental training data yields consistent performance gains in low-data settings, such as UAV-captured imagery where annotations are scarce. Beyond empirical improvements, we provide practical guidelines for (i) choosing an appropriate spatiotemporal generative setup, (ii) transferring annotations to synthetic frames, and (iii) addressing disocclusion - regions newly revealed and unlabeled in generated views. Experiments on COCO subsets and UAV-captured datasets show that, when applied judiciously, spatiotemporal augmentation broadens the data distribution along axes underrepresented by traditional and prior generative methods, offering an effective lever for improving model performance in data-scarce regimes.
Figures & tables
| Augmentation | Raw | Instagen | ControlAug | Ours |
|---|---|---|---|---|
| Image Recall | 88.60 | 97.14 | 98.62 | 98.98 |
| Object Recall | 72.83 | 78.42 | 86.46 | 88.56 |
| KID |
| Dataset | Method | mAP@50 | mAP@50:95 |
|---|---|---|---|
| COCO5k | Trad | 36.3 | 24.8 |
| Trad + ControlAug (8x) | 40.1 ( ) | 27.6 ( ) | |
| Trad + InstaGen (8x) | 40.3 ( ) | 28.1 ( ) | |
| Trad + Ours (8x) | 41.1 ( ) | 28.6 ( ) | |
| VisDrone | Trad | 45.9 | 28.5 |
| Trad + ControlAug (8x) | 48.5 ( ) | 30.5 ( ) |
| Raw | ControlAug [ 14 ] | InstaGen [ 15 ] | |
|---|---|---|---|
| Base | 24.8 | 27.6 | 28.1 |
| +Ours | 28.6 | 29.7 | 29.1 |
| +3.8 | + 2.1 | +1.0 |
| Dataset size | 5k | 10k | 15k | 20k |
|---|---|---|---|---|
| Raw | 19.2 | 25.4 | 28.4 | 30.2 |
| Augmented | 22.7 | 27.9 | 30.6 | 32.2 |
| +3.5 | +2.5 | +2.2 | +2.0 |
| Dataset | Method | mAP@50 | mAP@50:95 |
|---|---|---|---|
| COCO5k | Spatial (8x) | 41.1 (+4.8) | 28.6 (+3.8) |
| Temporal (8x) | 38.7 (+2.4) | 26.7 (+1.9) | |
| VisDrone | Spatial (40x) | 49.2 (+3.3) | 31.1 (+2.6) |
| Temporal (40x) | 49.6 (+3.7) | 31.3 (+2.8) |
| raw | w/o 1:1 mixing | w/ 1:1 mixing | |
|---|---|---|---|
| mAP@50 | 36.3 | 39.9 (+3.6) | 41.1 (+4.8) |
| mAP@50:95 | 24.8 | 27.9 (+3.1) | 28.6 (+3.8) |
| Data | #frames per img | mAP@50 | mAP@50:95 |
|---|---|---|---|
| Raw | – | 36.3 | 24.8 |
| Sample every 3 frames | 8 | 40.7 | 28.2 |
| Sample every 4 frames | 8 | 41.1 | 28.5 |
| Sample every 5 frames | 8 | 41.1 | 28.6 |
| Use first 36 frames | 36 | 40.5 | 28.3 |
| Removed samples | w/o filtering | 5% | 10% | 20% |
|---|---|---|---|---|
| mAP@50 | 41.1 | 40.5 | 41.4 | 40.4 |
| mAP@50:95 | 28.6 | 28.3 | 29.0 | 28.2 |
| -NN | Linear | |||
| Augmentation | Top-1 | Top-5 | Top-1 | Top-5 |
| Trad Aug | 23.13 | 38.81 | 27.86 | 49.28 |
| Gen Aug | 25.87 | 42.20 | 29.90 | 52.39 |
| +2.74 | +3.39 | +2.04 | +3.11 | |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset size | 5k | 10k | 15k | 20k |
|---|---|---|---|---|
| Raw | 19.8 | 30.0 | 34.1 | 36.1 |
| Augmented | 24.5 (+4.7) | 31.7 (+1.7) | 35.3 (+1.2) | 38.9 (+2.8) |
| Dataset size | 5k | 10k | 15k | 20k | 25k | 30k | 35k |
|---|---|---|---|---|---|---|---|
| Raw | 18.7 | 24.2 | 27.5 | 29.8 | 31.2 | 32.7 | 33.6 |
| Augmented | 20.7 | 26.3 | 29.3 | 31.4 | 32.6 | 33.4 | 34.2 |
| Conf thr. | 0.5 | 0.6 | 0.7 | 0.8 | w/o pseudo-labels |
|---|---|---|---|---|---|
| mAP@50 | 41.1 | 40.7 | 41.4 | 40.7 | 41.1 |
| mAP@50:95 | 28.7 | 28.3 | 28.9 | 28.3 | 28.6 |
| Dataset | Method | mAP@50 | mAP@50:95 |
|---|---|---|---|
| COCO5k | w/o masking | 41.1 | 28.6 |
| polygon masking | 40.4 | 28.0 | |
| pixel-wise masking | 39.6 | 27.4 | |
| VisDrone | w/o masking | 49.6 | 31.3 |
| polygon masking | 49.6 | 31.3 | |
| pixel-wise masking | 49.5 | 31.1 |
| Training set size | Augmented | mAP@50 | mAP@50:95 |
|---|---|---|---|
| 100 | 71.1 | 39.4 | |
| 100 | ✓ | 74.5 (+3.4) | 44.1 (+4.7) |
| 200 | 81.9 | 51.3 | |
| 200 | ✓ | 86.8 (+4.9) | 56.8 (+5.5) |
| 300 | 89.8 | 61.3 | |
| 300 | ✓ | 90.7 (+0.9) | 63.4 (+2.1) |
| Data | Pretrained | mAP@50 | mAP@50:95 |
|---|---|---|---|
| Raw | 45.9 | 28.5 | |
| Augmented (40x) | 49.6 (+3.7) | 31.3 (+2.8) | |
| Raw | ✓ | 48.2 | 30.1 |
| Augmented (40x) | ✓ | 50.2 (+2.0) | 31.7 (+1.6) |
| Data | mAP@50 | mAP@50:95 |
|---|---|---|
| Raw | 45.9 | 28.5 |
| Resize (8x) | 48.2 (+2.3) | 30.3 (+1.8) |
| Random crop (8x) | 48.9 (+3.0) | 30.7 (+2.2) |
| Random crop (40x) | 49.6 (+3.7) | 31.3 (+2.8) |
| Method | ControlAug | InstaGen | Ours-LTX | Ours-DimX |
|---|---|---|---|---|
| mAP | 30.5 | 29.8 | 31.3 | 31.1 |
| +2.0 | +1.3 | +2.8 | +2.6 | |
| Gen time/src-img | 169 s | 45 s | 110 s | 1210 s |
Data Augmentation
Data augmentation expands or transforms training examples to improve learning, data efficiency, and robustness to distribution shifts.
See all 249 papers