06 December, 2021
CNN-based Weighted Ensemble Technique for ImageNet Classification -IEMEK Journal of Embedded Systems and Applications | Korea Science
The ImageNet dataset is a large scale dataset and contains various natural scene images. In this paper, we propose a convolutional neural network (CNN)-based weighted ensemble technique for the ImageNet classification task. First, in order to fuse several models, our technique uses weights for each model, unlike the existing average-based ensemble technique. Then we propose an algorithm that automatically finds the coefficients used in later ensemble process. Our algorithm sequentially selects the model with the best performance of the validation set, and then obtains a weight that improves performance when combined with existing selected models. We applied the proposed algorithm to a total of 13 heterogeneous models, and as a result, 5 models were selected. These selected models were combined with weights, and we achieved 3.297% Top-5 error rate on the ImageNet test dataset.
Semi-supervised Learning for Instrument Detection with a Class Imbalanced Dataset
Memory-Efficient Models for Scene Text Recognition via Neural Architecture Search