Few-shot learning based on enhanced pseudo-labels and graded pseudo-labeled data selection

Kang Wang,Xuesong Wang,Yuhu Cheng
DOI: https://doi.org/10.1007/s13042-022-01727-z
2022-01-01
International Journal of Machine Learning and Cybernetics
Abstract:Pseudo-labeled data is used to solve the data shortage in few-shot learning, in which the quality of pseudo-labels and pseudo-labeled data selection determine the classification performance. In order to obtain the enhanced pseudo-labels, we used diverse inputs to encourage the label network to learn invariant and robust representations, improving the generalization ability. Simultaneously, the depthwise over-parameterized convolutional layer and group residual connection with shared parameters accelerate the network training and overcome the time-consuming caused by diverse inputs. Then, the graded pseudo-labeled data selection is proposed to determine various quantities of pseudo-labeled data based on the label network’s performance level, which improves the classification accuracy and avoids the high consumption caused by using all the pseudo-labeled data. Finally, we solved the data shortage in food recognition with the proposed method. The experiments show that our method has better classification accuracy and generalization ability in few-shot benchmark datasets and food recognition with few samples.
What problem does this paper attempt to address?