SWT: Self-Supervised Video Object Segmentation via Sliding, Wavelet and Transportation
Organizations: School of Computer Science and Technology, Soochow University, China
Abstract
Video Object Segmentation (VOS) aims to accurately segment target objects from consecutive video frames and track the changes of the objects in each frame of the video. Conventional VOS methods typically demand substantial quantities of pixel-level labeled video sequences for fully supervised learning, which limits the performance of the model in sparse video scenes, while existing VOS methods have limited adaptability to global changes in objects. Based on this observation, in this paper, we propose self-supervised VOS with Sliding window, Wavelet transform and optimal Transport (SWT), a self-supervised VOS framework entirely trained on static dataset using contrastive learning. Firstly, a rolling sample buffer reuses overlapping groups of independently sampled images across successive updates. Secondly, to address the long-distance modeling difficulty caused by simple convolutional structures, we introduce wavelet transform to expand the receptive field of convolutional kernels, thus improving the model's representational capability. Finally, we incorporate optimal transport to assist the model in finding the globally optimal match between the target across two frames, improving the model's ability to handle nonrigid deformations of objects. SWT only requires training on the COCO dataset once and achieves excellent results on five VOS datasets as well as an additional body part propagation dataset. The code will be released soon at https://github.com/machine928/SWT.git.
Figures & tables
| Item | Setting |
| Static training data | COCO train split; two random crops per image |
| Input augmentation | Random resized crop; area scale ; aspect ratio ; resized to |
| Normalization | Mean ; std. |
| Backbone and projection | ResNet-18; strides removed from layer3/layer4; projection hidden/output dimension 256; prediction hidden dimension 256 |
| Key encoder update | Momentum coefficient initialized as 0.99 and increased to 1 with a cosine schedule |
| Optimization | Adam; learning rate ; weight decay 0; 20 epochs; seed 42; mixed precision enabled |
| Method | BackBone | Training Dataset | DAVIS17-val | ||||
| MuG ( Lu et al., 2020 ) | ResNet-18 | OxUvA | 52.6 | 57.4 | 56.1 | 58.1 | 54.3 |
| MAST ( Lai et al., 2020 ) | ResNet-18 | YouTubeVOS | 63.3 | 73.2 | 67.6 | 77.7 | 65.5 |
| CRW ( Jabri et al., 2020 ) | ResNet-18 | Kinetics | 64.8 | 76.1 | 70.2 | 82.1 | 67.6 |
| ContrastCorr ( Wang et al., 2021a ) | ResNet-18 | TrackingNet | 60.5 | - | 65.5 | - | 63.0 |
| VFS ( Xu and Wang, 2021 ) | ResNet-18 | Kinetics | 64.0 | - | 69.4 | - | 66.7 |
| Method | Supervised Mode | DAVIS17-test | ||
| CRW ( Jabri et al., 2020 ) | Self-supervised | 52.3 | 59.6 | 55.9 |
| VFS ( Xu and Wang, 2021 ) | Self-supervised | 53.1 | 61.6 | 57.3 |
| SCC ( Son, 2022 ) | Self-supervised | 55.9 | 64.0 | 59.9 |
| SWT(Ours) | Self-supervised | 57.4 | 66.1 | 61.8 |
| Method | Supervised Mode | DAVIS16-val | ||
| SiamMask ( Wang et al., 2019a ) | Semi-supervised | 71.7 | 67.8 | 70.0 |
| OSVOS ( Caelles et al., 2017 ) | Semi-supervised | 79.8 | 80.6 | 80.2 |
| Colorization ( Vondrick et al., 2018 ) | Self-supervised | 38.9 | 30.0 | 34.9 |
| CorrFlow ( Lai and Xie, 2019 ) | Self-supervised | 47.1 | 49.9 | 48.0 |
| TimeCycle ( Wang et al., 2019b ) | Self-supervised | 55.8 | 51.1 | 53.5 |
| Method | Supervised Mode | Mean | Seen | Unseen | ||
| OSVOS ( Caelles et al., 2017 ) | Semi-supervised | 58.8 | 59.8 | 60.5 | 54.2 | 60.7 |
| PreMVOS ( Luiten et al., 2018 ) | Semi-supervised | 66.9 | 71.4 | 75.9 | 56.5 | 63.7 |
| TVOS ( Zhang et al., 2020 ) | Semi-supervised | 67.8 | 67.1 | 69.4 | 63.0 | 71.6 |
| Colorization ( Vondrick et al., 2018 ) | Self-supervised | 38.9 | 43.1 | 38.6 | 36.6 | 37.4 |
| CorrFlow ( Lai and Xie, 2019 ) | Self-supervised | 46.6 | 50.6 | 46.6 | 43.8 | 45.7 |
| Method | Supervised Mode | Mean | Seen | Unseen | ||
| Colorization ( Vondrick et al., 2018 ) | Self-supervised | 39.0 | 43.3 | 38.2 | 36.6 | 37.5 |
| CorrFlow ( Lai and Xie, 2019 ) | Self-supervised | 47.0 | 51.2 | 46.6 | 44.5 | 45.9 |
| RHMNet ( Zhou et al., 2025 ) | Scribble-supervised | 59.3 | 63.9 | 64.8 | 51.7 | 56.7 |
| MAST ( Lai et al., 2020 ) | Self-supervised | 64.9 | 64.3 | 65.3 | 61.5 | 68.4 |
| SWT(Ours) | Self-supervised | 71.7 | 69.5 | 67.7 | 71.6 | 77.8 |
| Method | Supervised Mode | VIP-val |
| mIoU | ||
| ResNet ( He et al., 2016 ) | Fully-supervised | 31.9 |
| ATEN ( Zhou et al., 2018 ) | Fully-supervised | 37.9 |
| TimeCycle ( Wang et al., 2019b ) | Self-supervised | 28.9 |
| UVC ( Li et al., 2019 ) | Self-supervised | 34.1 |
| CLTC ( Jeon et al., 2021 ) | Self-supervised | 37.8 |
| Module Variants | S | W | T | DAVIS17-val | DAVIS16-val |
| Baseline | ✗ | ✗ | ✗ | 72.0 | 78.9 |
| S | ✓ | ✗ | ✗ | 72.3 | 79.3 |
| W | ✗ | ✓ | ✗ | 72.6 | 79.6 |
| T | ✗ | ✗ | ✓ | 73.1 | 80.2 |
| S + W | ✓ | ✓ | ✗ | 73.0 | 80.3 |
| Augmentation Strategy | DAVIS17-val | DAVIS16-val |
| Detection Cropping | 71.9 | 78.8 |
| Enhanced Detection Cropping | 72.8 | 79.9 |
| Random Cropping | 73.3 | 80.5 |
| Marginal Distributions | ||
| Uniform | 73.2 | 80.4 |
| Method | Params | FLOPs | GPU memory | FPS |
| Baseline | 0.753M | 1.865G | 16.0GB | 48 |
| SWT | 1.160M | 2.064G | 21.4GB | 43 |
Video Object Segmentation
Video object segmentation assigns pixel-level masks to objects across frames while preserving their identities through motion, occlusion, and change.
See all 114 papers