Fashion-MNIST Classification Based on CNN Image Recognition

Abstract

Image classification is an important part of Artificial Intelligence (AI) and involves several tasks such as image normalization, image segmentation, feature extraction etc. Convolutional Neural Network (CNN) has been proved to be an effective network in image classification. According to this study, we had used a relatively small dataset named Fashion-MNIST (i.e. 70,000 images, 10 categories) to find out the model that can have a higher accuracy with limited training samples. We have trained three classic CNN models which are DenseNet121, MobileNet, ResNet 50 respectively. After that, we comprehensively measured the performance by several norms that comes from these three different models. Finally, we had a conclusion on which could be the most efficient one in this scenario based on the test results, in order to discovery which models are the most powerful one, and worth training. Human always deal with many types of images, so people need a powerful AI to help them to recognize and categorize these images. After this study, the DenseNet121 is the most powerful model in these three - DenseNet121, MobileNet, ResNet 50, the method to determine this result is that in the whole study we used a method called control variate method, we use the same amount of images, same amount of training times, then compared the final output of these three models, in the end we discovered that the DenseNet121 is the most powerful one.