arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.03469cs.LGcs.AIcs.CV

Fast FixMatch:基于课程批大小的更快半监督学习

Fast FixMatch: Faster Semi-Supervised Learning with Curriculum Batch Size

Rice University(莱斯大学)

John Chen, Chen Dun, Anastasios Kyrillidis

AI总结:

提出课程批大小(CBS)利用训练动态减少半监督学习计算开销,结合强标签增强与CPL构建Fast FixMatch,在保持同等错误率下实现2.1-3.4倍训练加速并提升数据利用率。

摘要:

半监督学习(SSL)的进展几乎完全消除了SSL与监督学习之间的差距,且仅使用了一小部分标签。然而,最近的性能提升往往以显著增加训练计算量为代价。为了解决这个问题,我们提出了课程批大小(Curriculum Batch Size, CBS),这是一种利用深度神经网络自然训练动态的无标签批大小课程。在训练开始时使用较小的无标签批大小,并在训练过程中逐渐增加到结束。无论数据集、模型或训练轮数如何,都使用固定的课程,并在所有设置中均展示了训练计算量的减少。我们将CBS、强标签增强和课程伪标签(Curriculum Pseudo Labeling, CPL)应用于FixMatch,并将新的SSL算法命名为Fast FixMatch。我们进行了消融研究,表明强标签增强和/或CPL并不能显著减少训练计算量,但与CBS协同作用时,它们能达到最优性能。与之前的最先进方法相比,Fast FixMatch还实现了显著更高的数据利用率。在CIFAR-10上,当仅保留40、250和4000个标签而移除其余所有标签时,与原始FixMatch相比,Fast FixMatch在保持相同的最先进错误率的同时,将训练计算量减少了2.1倍至3.4倍。在CIFAR-100、SVHN和STL-10上也取得了类似的结果。最后,Fast MixMatch在联邦SSL任务和在线/流学习SSL任务中实现了2.6倍至3.3倍的训练计算量减少,这进一步证明了Fast MixMatch在不同场景和任务中的泛化能力。

英文摘要:

Advances in Semi-Supervised Learning (SSL) have almost entirely closed the gap between SSL and Supervised Learning at a fraction of the number of labels. However, recent performance improvements have often come \textit{at the cost of significantly increased training computation}. To address this, we propose Curriculum Batch Size (CBS), \textit{an unlabeled batch size curriculum which exploits the natural training dynamics of deep neural networks.} A small unlabeled batch size is used in the beginning of training and is gradually increased to the end of training. A fixed curriculum is used regardless of dataset, model or number of epochs, and reduced training computations is demonstrated on all settings. We apply CBS, strong labeled augmentation, Curriculum Pseudo Labeling (CPL) \citep{FlexMatch} to FixMatch \citep{FixMatch} and term the new SSL algorithm Fast FixMatch. We perform an ablation study to show that strong labeled augmentation and/or CPL do not significantly reduce training computations, but, in synergy with CBS, they achieve optimal performance. Fast FixMatch also achieves substantially higher data utilization compared to previous state-of-the-art. Fast FixMatch achieves between $2.1\times$ - $3.4\times$ reduced training computations on CIFAR-10 with all but 40, 250 and 4000 labels removed, compared to vanilla FixMatch, while attaining the same cited state-of-the-art error rate \citep{FixMatch}. Similar results are achieved for CIFAR-100, SVHN and STL-10. Finally, Fast MixMatch achieves between $2.6\times$ - $3.3\times$ reduced training computations in federated SSL tasks and online/streaming learning SSL tasks, which further demonstrate the generializbility of Fast MixMatch to different scenarios and tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑