4 个月前

通过简单的集成投影器去偏、校准和提高半监督学习性能

通过简单的集成投影器去偏、校准和提高半监督学习性能

摘要

近期关于半监督学习(SSL)的研究取得了显著成功。尽管这些方法表现出色,但当前最先进的技术往往通过增加网络组件和额外的训练程序来实现越来越复杂的设计。本文提出了一种名为集成投影器辅助半监督学习(EPASS)的简单方法,主要集中在改进所学嵌入以提升现有对比联合训练半监督学习框架的性能。与标准方法不同的是,标准方法中从单个投影器学到的嵌入存储在内存库中用于对比学习,而EPASS则将多个投影器的集成嵌入存储在内存库中。因此,EPASS提高了泛化能力,增强了特征表示,并提升了性能。例如,在仅使用100k/1%/10%标记数据的情况下,EPASS分别将SimMatch的Top-1错误率降低了39.47%/31.39%/24.70%,并在ImageNet数据集上将CoMatch的Top-1错误率降低至40.24%/32.64%/25.90%。这些改进在不同的方法、网络架构和数据集中均表现一致,证明了所提方法的一般有效性。代码可在https://github.com/beandkay/EPASS 获取。

代码仓库

beandkay/epass
官方
pytorch

基准测试

基准方法指标
semi-supervised-image-classification-on-1CoMatch + EPASS (ResNet-50)
Top 1 Accuracy: 67.4%
Top 5 Accuracy: 87.3
semi-supervised-image-classification-on-1SimMatch + EPASS (ResNet-50)
Top 1 Accuracy: 68.6%
Top 5 Accuracy: 87.6
semi-supervised-image-classification-on-2CoMatch + EPASS (ResNet-50)
Top 1 Accuracy: 74.1%
Top 5 Accuracy: 91.5
semi-supervised-image-classification-on-2SimMatch + EPASS (ResNet-50)
Top 1 Accuracy: 75.3%
Top 5 Accuracy: 92.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过简单的集成投影器去偏、校准和提高半监督学习性能 | 论文 | HyperAI超神经