TorchSSL

收录理由

TorchSSL 是 FlexMatch 的官方 PyTorch 实现，FlexMatch 是 NeurIPS 2021 上提出课程伪标签方法的论文。这个库把 PiModel、Mean-Teacher、MixMatch、FixMatch 等九种经典半监督学习算法整合在同一个接口下，方便直接对比各自的实验结果。不过有个现实问题：项目已经停止维护，README 明确指向了继任者 USB，后者训练耗时大约是前者的八分之一，而且覆盖了 NLP 和音频任务。因此，TorchSSL 更适合当作一个可读的历史参考，用来理解这些算法的内部结构、复现论文里已发布的数字，而不是作为新工作的起点。即便如此，它仍然有存在价值——当伪标签和一致性正则化这类方法的代码细节都并排放在一个代码库里时，研究它们确实会省力不少。