算法的整个流程还是比较清晰简单的,主要使用了Mixup这一数据增强,混合了标签数据和无标签数据。
流程步骤如下:
一、记batch size为B,标签样本为X,无标签样本为U,temperature为T,无标签样本数据增强的次数为K,在实验中,K设为2,这里的数据增强就是指普通的数据增强而不是mixup。
二、
对一个batch中的X进行一次数据增强,得到
对一个batch中的U进行K次数据增强,得到
三、对所有增强的无标签数据进行预测,并对它们的预测求平均,再通过temperature进行sharpen,得到猜测标签
。
四、
得到增强的有标签数据和它们对应的标签;
得到增强的无标签数据和它们对应的猜测标签。