16 天前
Dash:基于动态阈值的半监督学习
Yi Xu, Lei Shang, Jinxing Ye, Qi Qian, Yu-Feng Li, Baigui Sun, Hao Li, Rong Jin

摘要
尽管半监督学习(Semi-Supervised Learning, SSL)因其在利用未标记数据方面的成功而在众多机器学习任务中受到广泛关注,但现有的SSL算法在训练过程中通常要么使用全部未标记样本,要么仅采用具有固定高置信度预测结果的未标记样本。然而,这种策略可能导致过多正确的或错误的伪标签样本被错误地剔除或保留。针对这一问题,本文提出了一种简单而强大的新框架——Dash。其核心思想是在应用现有SSL方法时,从未标记数据中动态选择一个子集进行训练,从而仅保留那些其伪标签与已标记数据相关联的未标记样本用于模型训练。该选择过程在每次参数更新迭代中执行,仅保留损失值低于动态调整阈值的样本。所提出的Dash方法在未标记数据选择方面展现出良好的自适应性,并具备理论保障。具体而言,本文从非凸优化的角度理论推导了Dash方法的收敛速率。最后,我们在多个基准数据集上通过实验验证了该方法相较于当前最先进方法的优越性能,充分证明了其有效性。