Efficient Few-Shot Classification Via Contrastive Pretraining on Web Data.

Zhuoling Li,Haohan Wang,Tymoteusz Swistek,En Yu,Haoqian Wang
DOI: https://doi.org/10.1109/tai.2022.3169463
2023-01-01
IEEE Transactions on Artificial Intelligence
Abstract:Few-shot classification is challenging due to the limited data and labels. The existing algorithms usually resolve this problem by pretraining models with a considerable amount of annotated data, which share knowledge with the target domain. Nevertheless, large quantities of homogenous data samples are not always available. To tackle this obstacle, we develop a few-shot learning framework that prepares data automatically and still produces well-behaved models. This framework is implemented through conducting contrastive learning on unlabeled web images. Instead of requiring manually annotated data, this framework trains models via constructing pseudolabels. Additionally, since online data are virtually limitless and continue to be generated, the model can, thus, be empowered to constantly obtain up-to-date knowledge from the Internet. Furthermore, we observe that the generalization ability of learned representation is crucial for self-supervised learning. To present its importance, a naive yet efficient normalization strategy is proposed. Consequentially, this strategy boosts the accuracy of trained models significantly. We demonstrate the superiority of the proposed framework with experiments on miniImageNet, tieredImageNet, and Omniglot. The results indicate that our method has surpassed previous unsupervised counterparts by large margins and obtained performance comparable with some supervised ones.
What problem does this paper attempt to address?