Follow the Geometry, Not the Model: Cold Start Semi-Supervised Learning
Organizations: School of Computer Science and Engineering, The Hebrew University of Jerusalem, Jerusalem 91904, Israel
Abstract
Modern semi-supervised learning (SSL) couples pseudo-label generation and classifier training, using the classifier's own confidence to select the pseudo-labels that are then used to update the model. In the cold-start regime, where at most a few labels per class are available, this coupling is ill-posed, since the classifier cannot supervise itself before it has learned. To address this problem, we propose VAST (Veracity-Aware Semi-Supervised Training), which decouples these two stages. Probabilistic beliefs over the unlabeled set are first inferred directly from the geometry of a frozen self-supervised embedding and only then distilled into an inductive classifier. The construction rests on the Veracity Matrix, a kernel-based structure that aggregates label evidence across the data manifold and admits an interpretation as a Dirichlet posterior under a per-observation powered-likelihood model. Additionally, we introduce Veracity Propagation, a self-terminating belief-spreading step that extends coverage beyond the kernel neighborhood of the labeled set. Under a controlled protocol in which all methods receive identical frozen embeddings and labeled sets, VAST outperforms the strongest graph-based SSL baselines at every operating point across three datasets, with statistically significant gains in 7 of 9 comparisons, while producing a deployable inductive classifier rather than requiring transductive graph inference. Compared with end-to-end confidence-gated SSL, we further find that these methods underperform in this setting and, in our experiments, do not consistently exceed labeled-only performance.
Figures & tables
| CIFAR-100 | TinyImageNet | ImageNet100 | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Method | 1/cls | 2/cls | 4/cls | 1/cls | 2/cls | 4/cls | 1/cls | 2/cls | 4/cls |
| MLP (labeled only) | |||||||||
| 1-NN | |||||||||
| Laplace | |||||||||
| Poisson | |||||||||
| C&S | |||||||||
| Method | 100 | 200 | 400 |
|---|---|---|---|
| FlexMatch (WRN-28-2) | 16.2 | 33.4 | – |
| FlexMatch (WRN-28-8) | 9.4 | 40.8 | |
| FreeMatch (WRN-28-8) | |||
| CoMatch | 30.7 | 38.9 | |
| SimMatch | 17.2 | 46.9 | |
| ExMatch ∗ |
| Configuration | 1/cls | 2/cls | 4/cls |
|---|---|---|---|
| VAST (distill + prop.) | |||
| Distill only (matched kernel) | |||
| Distill only (local RBF, best) |
| Configuration | 1/cls | 2/cls | 4/cls |
|---|---|---|---|
| VAST (distill + prop.) | |||
| VAST w/o degree normal. |
| Method | SimCLR | Barlow | BYOL | DINOv2 | VAE2 |
| 1 label per class | |||||
| 1-NN | – | ||||
| Poisson | |||||
| C&S | – | ||||
| VAST | – | ||||
| 4 labels per class | |||||
Appendix figures & tables22 assets
Supplementary material from the paper’s appendix.
Appendix
| Architecture | Parameters |
| ResNet-18 backbone ( VAST /BYOL) | M |
| BYOL projector (2-layer MLP, 512 256 256) | M |
| BYOL predictor (same architecture) | M |
| Full BYOL network (backbone + projector + predictor) | M |
| Labeled-only MLP classifier head | – M |
| WRN-28-2 ( FlexMatch ) | M |
| CIFAR-100 | ImageNet100 | TinyImageNet | |
| Encoder | BYOL | BYOL | SimCLR |
| 50,000 | 130,000 | 100,000 | |
| PR | 138.1 | 60.6 | 75.5 |
| @90% | 252 | 164 | 214 |
| @95% | 339 | 256 | 306 |
| @99% | 460 | 425 | 441 |
| Hard degree | Soft degree | |||||||
|---|---|---|---|---|---|---|---|---|
| Dataset | Encoder | mean | max | p99 | mean | max | p99 | |
| CIFAR-100 | BYOL | 13 | 20.3 | 518 | 65 | 12.2 | 280.5 | 37.1 |
| ImageNet100 | BYOL | 5 | 8.0 | 78 | 23 | 4.8 | 44.3 | 13.7 |
| TinyImageNet | SimCLR | 3 | 5.1 | 50 | 17 | 3.0 | 28.7 | 9.9 |
| CIFAR-100 | ImageNet100 | TinyImageNet | |
|---|---|---|---|
| 1 | 0.616 | 0.611 | 0.343 |
| 3 | 0.581 | 0.572 | 0.312 |
| 5 | 0.563 | 0.551 | 0.295 |
| 10 | 0.536 | 0.522 | 0.272 |
| 15 | 0.520 | 0.505 | 0.258 |
| Gate | 100 | 200 | 400 |
|---|---|---|---|
| adaptive | |||
| MLP (labeled only) |
| Stage | Time | Space | Frequency |
|---|---|---|---|
| Graph construction | with brute-force pairwise distances; reduced to sub-quadratic in practice by any standard exact or approximate nearest-neighbor index (e.g. FAISS, HNSW) without changes to the rest of the algorithm | , sparse | once |
| Veracity Matrix seeding | , averaging to under the sparse -NN graph | for the matrix itself the dominant memory structure for typical (e.g. floats on CIFAR-100, vs. pre-symmetrization directed edges, after symmetrization) | once |
| Propagation | only active receivers are updated each round, and monotonically shrinks as points cross and freeze; is a loose upper bound | reuses seeding | once |
| Classifier training | for epochs at per-example cost identical in form to any baseline of the same architecture | model-dependent | every epoch |
| Inference (new test point) | one forward pass through | model-dependent | every query |
| CIFAR-100 | ImageNet100 | |||||
|---|---|---|---|---|---|---|
| Method | 1/cls | 2/cls | 4/cls | 1/cls | 2/cls | 4/cls |
| Poisson ( ) | † | |||||
| Laplace (LP) | ||||||
| C&S | ||||||
| LP-DSL | ||||||
| VAST (distill + prop.) | ||||||
| CIFAR-100 | ImageNet100 | |||||
|---|---|---|---|---|---|---|
| Configuration | 1/cls | 2/cls | 4/cls | 1/cls | 2/cls | 4/cls |
| VAST (distill + prop.) | ||||||
| Distill only (matched kernel) | ||||||
| Distill only (local RBF, best-effort) | ||||||
| CIFAR-100 | |||
|---|---|---|---|
| Weighting | 1/cls | 2/cls | 4/cls |
| Per-sample (reported) | |||
| Uniform ( ) | |||
| TinyImageNet | |||
| Weighting | 200 lbl | 400 lbl | 800 lbl |
| Per-sample (reported) | |||
| CIFAR-100 (BYOL) | |||
|---|---|---|---|
| Peakedness function | 1/cls | 2/cls | 4/cls |
| Max, (reported) | |||
| Margin, | |||
| Entropy, | |||
| Poisson (reference) | |||
| Embed. | Method | 100 | 200 | 400 |
|---|---|---|---|---|
| BYOL | VAST | |||
| Barlow Tw. | VAST | |||
| Poisson | ||||
| PoissonMBO | ||||
| Laplace | ||||
| SimCLR | VAST |
| Embedding | 100 | 200 | 400 |
|---|---|---|---|
| BYOL | |||
| VAE2 |
| Method | 100 lbl | 200 lbl | 400 lbl |
|---|---|---|---|
| Laplace | |||
| LP-DSL | † | ||
| C&S | |||
| Poisson | |||
| VAST (distill + prop.) |
Semi-Supervised Learning
Semi-supervised learning uses both labeled and unlabeled data to improve prediction when reliable annotations are scarce.
See all 164 papers