Self-training with Noisy Student improves ImageNet classification
Authors: Qizhe Xie, Eduard Hovy, Minh-Thang Luong, Quoc V. Le
AI-generated Key Points
- Authors: Qizhe Xie, Eduard Hovy, Minh-Thang Luong, Quoc V. Le
- Self-training method significantly enhances ImageNet classification accuracy
- Achieves 87.4% top-1 accuracy, surpassing state-of-the-art model by 1.0%
- ImageNet-A top-1 accuracy increased from 16.6% to 74.2%
- Mean corruption error on ImageNet-C reduced from 45.7 to 31.2
- Mean flip rate on ImageNet-P decreased from 27.8 to 16.1
- Methodology involves training an EfficientNet teacher model initially on labeled images and using it to generate pseudo labels for unlabeled images
- Larger EfficientNet student model is trained using both labeled and pseudo-labeled data in combination
- Iterative process continues with student model becoming the new teacher in subsequent rounds of training
- No noise introduced during pseudo label generation; noise deliberately injected during student learning phase for improved performance based on noisy labels
Abstract: We present a simple self-training method that achieves 87.4% top-1 accuracy on ImageNet, which is 1.0% better than the state-of-the-art model that requires 3.5B weakly labeled Instagram images. On robustness test sets, it improves ImageNet-A top-1 accuracy from 16.6% to 74.2%, reduces ImageNet-C mean corruption error from 45.7 to 31.2, and reduces ImageNet-P mean flip rate from 27.8 to 16.1. To achieve this result, we first train an EfficientNet model on labeled ImageNet images and use it as a teacher to generate pseudo labels on 300M unlabeled images. We then train a larger EfficientNet as a student model on the combination of labeled and pseudo labeled images. We iterate this process by putting back the student as the teacher. During the generation of the pseudo labels, the teacher is not noised so that the pseudo labels are as good as possible. But during the learning of the student, we inject noise such as data augmentation, dropout, stochastic depth to the student so that the noised student is forced to learn harder from the pseudo labels.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.