Decoupled Optimization for Teacher-Student Semi-Supervised Learning via a Pioneer Student
Organizations: School of Computer Science and Technology Beijing Jiaotong University · School of Data Science and Intelligent Media Communication University of China
Abstract
Semi-supervised learning (SSL) relies on two core mechanisms: self-training under the Teacher-Student (T-S) framework and joint optimization of labeled and unlabeled losses. Despite their effectiveness, we find both mechanisms introduce distinct optimization pathologies. First, parameter coupling enforces strict synchronization between teacher and student, where strong regularization on the student degrades the teacher's fitting ability, thereby limiting the permissible generalization intensity. Second, the imbalance in gradient update consistency between labeled and unlabeled losses drives the shared parameters to prematurely converge to labeled-dominated local minima, creating a bottleneck for global optimization. To address both issues, we propose the Pioneer Student (PiS), an auxiliary branch that operates in an independent parameter space and periodically transfers accumulated knowledge back to the T-S model. Extensive experiments show that PiS is a universal plug-and-play module that consistently improves mainstream SSL methods.
Figures & tables
| Supervision Type | Augmentation | Acc. (%) | Pseudo-Label Acc. (%) |
|---|---|---|---|
| Endogenous | RA(2,3) | 78.13 | 77.97 |
| RA(3,5) | 78.46 | 77.92 | |
| RA(5,10) | 76.27 | 75.25 | |
| Exogenous | RA(5,10) | 78.80 | 77.92 |
| Modality | CV | Audio | NLP | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Dataset | CIFAR-100 | STL-10 | EuroSAT | UrbanSound8K | ACL-IMDb | |||||
| # Labels per Class | 2 | 4 | 4 | 10 | 2 | 4 | 10 | 40 | 10 | 50 |
| Baseline Methods | ||||||||||
| FlexMatch [ 14 ] | 73.24 1.12 | 81.76 0.36 | 85.60 3.11 | 91.83 0.78 | 94.83 0.57 | 94.42 0.81 | 60.77 0.96 | 76.30 1.66 | 92.18 0.77 | 92.59 0.38 |
| CoMatch [ 16 ] | 64.92 0.69 | 74.77 0.50 | 84.88 1.88 | 90.44 1.35 | 94.25 0.43 | 95.19 1.05 | 70.37 1.15 | 79.81 2.34 | 92.56 0.30 | 92.38 1.14 |
| SimMatch [ 33 ] | 76.22 1.08 | 82.94 0.78 | 88.23 3.20 | 92.45 1.86 | 92.34 0.60 | 94.73 0.89 | 63.37 6.10 | 80.05 2.11 | 92.07 0.55 | 92.92 0.33 |
| Dataset | ImageNet | |
|---|---|---|
| # Labels per Class | 10 | 100 |
| FreeMatch | 54.69 | 72.57 |
| RegMixMatch | 58.35 | 73.66 |
| FreePiS | 60.02 | 66.69 |
| FreePiS † | – | 73.24 |
| ID | Method/Variant | Mechanisms | FGVC | CIFAR-100 | ||||
|---|---|---|---|---|---|---|---|---|
| Anti-FGD | Anti-LDT | Warm-up | Acc (%) | Acc (%) | ||||
| ( a ) | FreeMatch | ✗ | ✗ | – | 51.65 | – | 78.46 | – |
| ( b ) | FreePiS w. Label Loss | ✓ | ✗ | ✓ | 54.66 | +3.01 | 79.02 | +0.56 |
| ( c ) | FreePiS w. Same Aug. | ✗ | ✓ | ✓ | 52.78 | +1.13 | 80.57 | +2.11 |
| ( d ) | FreePiS w/o Warm-up | ✓ | ✓ | ✗ | 52.25 | +0.60 | 80.66 | +2.20 |
| ( e ) | FreePiS (Full) | ✓ | ✓ | ✓ | 55.54 | +3.89 | 80.83 | +2.37 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | CIFAR-100 | STL-10 | EuroSAT | FGVC-Aircraft | ImageNet-1K |
|---|---|---|---|---|---|
| Image Size | 32 | 96 | 32 | 224 | 224 |
| Backbone (ViT [ 41 ] ) | S-P2-32 | B-P16-96 | S-P2-32 | B-P16-224 | B-P16-224 |
| Weight Decay | 5e-4 | 5e-4 | 5e-4 | 5e-2 | 5e-2 |
| Learning Rate | 5e-4 | 1e-4 | 5e-5 | 5e-4 | 3e-4 |
| Layer Decay Rate | 0.5 | 0.95 | 1.0 | 0.5 | 0.5 |
| Labeled Batch Size | 8 | 32 | |||
| Method | Time (s/iter) | +PiS | Memory (MiB) | +PiS |
|---|---|---|---|---|
| FixMatch | 0.0944 | 0.1256 (+33%) | 3359 | 4534 (+35%) |
| FreeMatch | 0.0960 | 0.1306 (+36%) | 3341 | 4543 (+36%) |
| RegMixMatch | 0.1344 | 0.1613 (+20%) | 4616 | 5446 (+18%) |
| Dataset | CIFAR-100 | STL-10 | EuroSAT | |||
|---|---|---|---|---|---|---|
| # Labels per Class | 2 | 4 | 4 | 10 | 2 | 4 |
| FreeMatch (200k iters) | 78.46 | 83.58 | 87.20 | 91.51 | 93.48 | 94.20 |
| FreePiS (147k iters) | 80.61 | 84.27 | 90.44 | 92.85 | 94.70 | 95.84 |
| FreePiS (200k iters) | 80.83 | 84.45 | 90.65 | 93.13 | 94.92 | 95.89 |
| Method | 9% ‡ | 50% ‡ | 71% | 80% | 85% |
|---|---|---|---|---|---|
| ( ) | ( ) | ( ) | ( ) | ( ) | |
| FreeMatch | 56.45 | 59.88 | 63.01 | 65.66 | 65.87 |
| RegMixMatch | 56.86 | 60.13 | 63.55 | 66.19 | 66.40 |
| FreePiS | 56.56 | 61.24 | 64.72 | 68.35 | 70.33 |
| Method | 50% ‡ | 38% ‡ | 31% ‡ | 23% | 13% |
|---|---|---|---|---|---|
| ( ) | ( ) | ( ) | ( ) | ( ) | |
| FreeMatch | 83.34 | 74.77 | 70.39 | 62.87 | 49.65 |
| RegMixMatch | 84.10 | 75.24 | 71.27 | 63.40 | 47.89 |
| FreePiS | 84.56 | 76.26 | 73.11 | 67.33 | 54.54 |
| Metric | Method | Augmentation Intensity ( ) | Dropout Rate | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 3 | 4 | 5 | 6 | 7 | 8 | 0.10 | 0.15 | 0.20 | 0.25 | 0.30 | ||
| Accuracy (%) | FreeMatch | 80.44 | 80.02 | 78.65 | 77.96 | 77.00 | 76.65 | 80.60 | 80.71 | 80.44 | 79.08 | 77.98 |
| FreePiS | 80.91 | 81.66 | 82.18 | 82.03 | 81.11 | 80.80 | 81.49 | 82.06 | 82.18 | 81.23 | 80.77 | |
| Reliability (%) | FreeMatch | 79.24 | 78.09 | 76.19 | 74.45 | 73.17 | 72.68 | 79.95 | 80.01 | 79.24 | 77.09 | 75.35 |
| FreePiS | 80.22 | 81.37 | 82.67 | 82.22 | 81.39 | 80.36 | 81.36 | 82.53 | 82.67 | 81.66 | 81.00 | |
| Method | Sync Mechanism | Accuracy (%) |
|---|---|---|
| FreeMatch | — | 78.46 |
| FreePiS + LI- =0.3 | Linear Interp. ( =0.3) | 78.80 |
| FreePiS + LI- =0.5 | Linear Interp. ( =0.5) | 79.34 |
| FreePiS + LI- =0.7 | Linear Interp. ( =0.7) | 80.19 |
| FreePiS + HO | Hard Overwrite ( =1.0) | 80.83 |
| Asset | Modality | Type | License / Terms | Reference |
| Datasets | ||||
| CIFAR-10/100 | CV | Dataset | Non-commercial research use | [ 8 ] |
| STL-10 | CV | Dataset | Non-commercial research use | [ 28 ] |
| ImageNet-1K | CV | Dataset | ImageNet Terms of Access (non-commercial use) | [ 37 ] |
| EuroSAT | CV | Dataset | MIT License | [ 34 ] |
| FGVC-Aircraft | CV | Dataset | Non-commercial research use | [ 39 ] |