pngwn/cifar-selftrain-teacher-t0

CIFAR-10 Self-Training Teacher (t=0)

Initial weak teacher of an iterative self-training trajectory on CIFAR-10, reproducing the deep-learning analogue of Why Self-Training Helps and Hurts (arXiv:2602.14029, Appendix A).

Clean CIFAR-10 test accuracy: 42.44% (iteration t=0).

Trained on 5000 CIFAR-10 images with 40% symmetric label noise. This is the starting point of the self-training trajectory.

Training configuration

• —Architecture: ResNet-18 with CIFAR stem (3x3 conv, no maxpool), trained from scratch
• —Iteration: t=0 of K=8
• —Data: 5000 fresh disjoint CIFAR-10 train images per iteration (uoft-cs/cifar10)
• —Teacher (t=0) labels: 40% symmetric noise; students: hard pseudo-labels from iterate t-1
• —SGD lr 0.05, momentum 0.9, weight decay 5e-4, 5-epoch warmup + cosine, batch 128, 60 epochs
• —Augmentation: random crop (pad 4) + horizontal flip
• —torch 2.13.0+cu130, seed 100

Full trajectory, configs, metrics and plots: https://huggingface.co/datasets/pngwn/self-training-denoising-forgetting

Usage

import torch, torch.nn as nn, torchvision

def build_model():
    m = torchvision.models.resnet18(weights=None)
    m.conv1 = nn.Conv2d(3, 64, kernel_size=3, stride=1, padding=1, bias=False)
    m.maxpool = nn.Identity()
    m.fc = nn.Linear(512, 10)
    return m

model = build_model()
sd = torch.load("model.pt", map_location="cpu", weights_only=True)
model.load_state_dict(sd)
model.eval()

Input: CIFAR-10 images normalized with mean (0.4914, 0.4822, 0.4465), std (0.2470, 0.2435, 0.2616).