DMT: Dynamic Mutual Training for Semi-Supervised Learning

Recent semi-supervised learning methods use pseudo supervision as core idea, especially self-training methods that generate pseudo labels. However, pseudo labels are unreliable. Self-training methods usually rely on single model prediction confidence to filter low-confidence pseudo labels, thus remaining high-confidence errors and wasting many low-confidence correct labels. In this paper, we point out it is difficult for a model to counter its own errors. Instead, leveraging inter-model disagreement between different models is a key to locate pseudo label errors. With this new viewpoint, we propose mutual training between two different models by a dynamically re-weighted loss function, called Dynamic Mutual Training (DMT). We quantify inter-model disagreement by comparing predictions from two different models to dynamically re-weight loss in training, where a larger disagreement indicates a possible error and corresponds to a lower loss value. Extensive experiments show that DMT achieves state-of-the-art performance in both image classification and semantic segmentation. Our codes are released at https://github.com/voldemortX/DST-CBC .

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Semi-Supervised Image Classification CIFAR-10, 4000 Labels DMT (WRN-28-2) Percentage error 5.79 # 23
Semi-Supervised Semantic Segmentation Cityscapes 100 samples labeled DMT (DeepLab v2 MSCOCO/ImageNet pre-trained) Validation mIoU 54.80% # 10
Semi-Supervised Semantic Segmentation Cityscapes 12.5% labeled DMT (DeepLab v2 MSCOCO/ImageNet pre-trained) Validation mIoU 63.03% # 22
Semi-Supervised Semantic Segmentation Pascal VOC 2012 12.5% labeled DMT Validation mIoU 72.70% # 19
Semi-Supervised Semantic Segmentation PASCAL VOC 2012 1464 labels DMT (DeepLab v2, ResNet-50) Validation mIoU 74.85 # 8
Semi-Supervised Semantic Segmentation Pascal VOC 2012 1% labeled DMT (DeepLab v2 MSCOCO pre-trained) Validation mIoU 63.04% # 3
Semi-Supervised Semantic Segmentation Pascal VOC 2012 2% labeled DMT (DeepLab v2 MSCOCO pre-trained) Validation mIoU 67.15% # 4
Semi-Supervised Semantic Segmentation Pascal VOC 2012 5% labeled DMT (DeepLab v2 MSCOCO/ImageNet pre-trained) Validation mIoU 69.92% # 5

Methods


No methods listed for this paper. Add relevant methods here