Classifying e-Commerce Product Categories Using Image-based Deep Learning
Keywords:
Deep Learning, Image Classification, Categorizing Products, E-commerceAbstract
This research aimed to create a deep learning model to classify categories of products that are being sold on an e-commerce platform by analyzing images of the products. Different techniques were comparatively assessed and their efficiencies were compared to arrive at the best model. The public image dataset of the Shopee e-commerce system, consisting of 38 product categories, with a total of 106,309 images, was used. As these images belong to actual product images in the Shopee e-commerce system, they are highly complex in terms of computer vision and image analytics. Six models were tested along with 2 different loss functions. The results revealed that EfficientNetB5 was the best model in terms of accuracy, considering the size of the model and the time used in classifying the images; the accuracy on the test set was noted to be 84% while the value was 92% on the additional test set. The inference time used was 0.068 seconds per image with the model size of 141.9 MB. When the model was tested for the task of image classification in comparison with the actual Shopee application, the EfficientNetB5 model was 31.5% more accurate. When compared to four human participants, although our model was 7.16% less accurate, the model was 25.5 times faster in classifying each image.
References
Electronic Transactions Development Agency, 2019, Value of e-Commerce: Survey in Thailand 2019 [Online], Available: http://blog.bru.ac.th/wp-content/uploads/bp-attachments/35694/เอกสารการแถลงผลการสำรวจมูลค่าพาณิชย์อิเล็กทรอนิกส์-ปี-2562.pdf. (In Thai) [1 August 2021]
Wunderman Thompson, 2021, “The Future Shopper Report 2021 [Online], Available: https://www.wundermanthompson.com/insight/the-future-shopper-report-2021. [1 August 2021]
Dan, C., 2014, “Consumer-To-Consumer (C2C) Electronic Commerce: The Recent Picture,” International Journal of Networks and Communications, 4 (2), pp. 29-32.
Herdian, Kusuma, G.P. and Suharjito, 2019, “Classification of C2C e-Commerce Product Images Using Deep Learning Algorithm,” International Journal of Advanced Computer Science and Applications, 10 (9), pp. 196-203.
Partalas, I. and Balikas, G., 2016, E-Commerce Product Classification: Our Participation At cDiscount 2015 Challenge [Online], Available: https://arxiv.org/abs/1606.02854. [1 August 2021]
Oyewole, S.A. and Olugbara, O.O., 2018, “Product Image Classification Using Eigen Colour Feature with Ensemble Machine Learning,” Egyptian Informatics Journal, 19 (2), pp. 83-100.
Jha, B.K., Sivasankari, G.G. and Venugopal, K.R., 2021, “E-Commerce Product Image Classification Using Transfer Learning," International Conference on Computing Methodologies and Communication (ICCMC), 8-10 April 2021, Erode, India, pp. 904-912. https://doi.org/ 10.1109/ICCMC51019.2021.9418371
Simonyan, K. and Zisserman, A., 2015, Very Deep Convolutional Networks for Large-Scale Image Recognition [Online], Available: https://arxiv.org/abs/1409.1556. [1 August 2021]
He, K., Zhang, X., Ren, S. and Sun, J., 2016, “Deep Residual Learning for Image Recognition,” IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 27-30 June 2016, Las Vegas, USA, pp. 770-778.
Tan, M. and Le, Q.V., 2019, EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks [Online], Available: https://arxiv.org/abs/1905.11946. [1 August 2021]
Li, F., Kant, S., Araki, S., Bangera, S. and Shukla, S.S., 2020, Neural Networks for Fashion Image Classification and Visual Search [Online], Available: https://arxiv.org/abs/2005.08170 [1 August 2021]
Rajnoha, M., Burget, R. and Povoda, L., 2018, “Image Background Noise Impact on Convolutional Neural Network Training,” International Congress on Ultra-Modern Telecommunications and Control Systems and Workshops (ICUMT), 5-9 November 2018, Moscow, Russia, pp. 1-4.
Dodge, S. and Karam, L., 2016, “Understanding How Image Quality Affects Deep Neural Networks,” International Conference on Quality of Multimedia Experience (QoMEX), 6-8 June 2016, Lisbon, Portugal, pp. 1-6.
Lin, T.Y., Goyal, P., Girshick, R., He, K. and Dollár, P., 2017, “Focal Loss for Dense Object Detection," IEEE International Conference on Computer Vision (ICCV), 22-29 October 2017, Venice, Italy, pp. 2999-3007.
Zhao, Y., Lin, F., Liu, S., Hu, Z., Li, H. and Bai, Y., 2019, “Constrained-Focal-Loss Based Deep Learning for Segmentation of Spores,” IEEE Access, 7, pp. 165029-165038.
Bitirim, Y., Bitirim, S., Ertugrul, D.C. and Toygar, O., 2020, “An Evaluation of Reverse Image Search Performance of Google,” IEEE Annual Computers, Software, and Applications Conference (COMPSAC), 13-17 July 2020, Madrid, Spain, pp. 1368-1372.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2023 King Mongkut's University of Technology Thonburi
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
Any form of contents contained in an article published in Science and Engineering Connect, including text, equations, formula, tables, figures and other forms of illustrations are copyrights of King Mongkut's University of Technology Thonburi. Reproduction of these contents in any format for commercial purpose requires a prior written consent of the Editor of the Journal.
