A Comparison of Convolutional Neural Network (CNN) and Transfer Learning MobileNetV2 Performance on Spices Images Classification
Abstract
This research was conducted to analyze the performance of the CNN algorithm without transfer learning in classifying spice images and compare it with the CNN algorithm using transfer learning on the MobileNetV2 architecture. This comparison aims to evaluate both methods' accuracy, efficiency, and overall performance and analyze the impact of transfer learning on classification results in the context of spices. The dataset consists of 1500 spice images divided into 10 classes, with each class of 150 images. In the first experiment, CNN without transfer learning resulted in 93% accuracy performance. For the second experiment using MobileNetV2, there was an increase in accuracy, reaching a value of 99% for all spice classes. The results of this study confirm that MobileNetV2 architecture significantly improves the accuracy and performance of spice classification compared to CNN without transfer learning, which can be recommended for spice image classification.
Downloads
References
M. I. Al-Amin, F. N. Sidiq, D. R. Ramdania, N. Fajar, Y. A. Gerhana, and M. Harika, “Spices Image Classification Using Support Vector Machine,” in 2022 10th International Conference on Cyber and IT Service Management, CITSM 2022, Institute of Electrical and Electronics Engineers Inc., 2022, pp. 1–4. doi: 10.1109/CITSM56380.2022.9935856.
R. Widyani, M. Nilamcaya, and D. Iftitah, “The Benefits of Indonesian Spices for Livestock Health As A Legacy of Our Ancestors Local Wisdom,” IOP Conf Ser Earth Environ Sci, vol. 1020, no. 1, pp. 1–11, May 2022, doi: 10.1088/1755-1315/1020/1/012032.
R. Velázquez, A. Rodríguez, A. Hernández, R. Casquete, M. J. Benito, and A. Martín, “Spice and Herb Frauds: Types, Incidence, and Detection: The State of the Art,” Foods, vol. 12, no. 18, pp. 1–38, Sep. 2023, doi: 10.3390/foods12183373.
I. P. W. Prasetia and I. M. G. Sunarya, “Image Classification of Spices Basic Ingredients for Making Bumbu Bali Base Genep Based on Deep Learning,” Jurnal Nasional Pendidikan Teknik Informatika (JANAPATI), vol. 13, no. 1, pp. 79–90, Mar. 2024, doi: 10.23887/janapati.v13i1.67967.
Y. Zhang and X. Zheng, “Development of Image Processing Based on Deep Learning Algorithm,” in 2022 IEEE Asia-Pacific Conference on Image Processing, Electronics and Computers, IPEC 2022, Institute of Electrical and Electronics Engineers Inc., 2022, pp. 1226–1228. doi: 10.1109/IPEC54454.2022.9777479.
D. K. Ponmozhi and R. Jose, “A REVIEW ON SPICES CLASSIFICATION AND RECOGNITION,” Transtellar Publications, vol. 12, no. 1, pp. 25–32, 2022, [Online]. Available: www.tjprc.org
C. Li, J. Feng, L. Hu, J. Li, and H. Ma, “Review of Image Classification Method Based on Deep Transfer Learning,” in Proceedings - 2020 16th International Conference on Computational Intelligence and Security, CIS 2020, Institute of Electrical and Electronics Engineers Inc., Nov. 2020, pp. 104–108. doi: 10.1109/CIS52066.2020.00031.
A. Hosna, E. Merry, J. Gyalmo, Z. Alom, Z. Aung, and M. A. Azim, “Transfer learning: a friendly introduction,” J Big Data, vol. 9, no. 102, p. 1, Oct. 2022, doi: 10.1186/s40537-022-00652-w.
Parjito, F. Ulum, K. Muludi, Z. Abidin, R. M. Alma, and Permata, “Classification of Ornamental Plants with Convolutional Neural Networks and MobileNetV2 Approach,” in Proceedings - ISMODE 2022: 2nd International Seminar on Machine Learning, Optimization, and Data Science, Institute of Electrical and Electronics Engineers Inc., 2022, pp. 534–539. doi: 10.1109/ISMODE56940.2022.10180988.
D. Watni and S. Chawla, “Impact of various image formats supported by android smartphones on image steganography: a preliminary study,” Multimed Tools Appl, vol. 83, no. 11, pp. 33031–33050, Mar. 2023, doi: 10.1007/s11042-023-16343-4.
X. Pei et al., “Robustness of machine learning to color, size change, normalization, and image enhancement on micrograph datasets with large sample differences,” Mater Des, vol. 232, pp. 1–13, 2023, doi: 10.1016/j.matdes.2023.112086.
D. R. Nayak, N. Padhy, P. K. Mallick, M. Zymbler, and S. Kumar, “Brain Tumor Classification Using Dense Efficient-Net,” Axioms, vol. 11, no. 1, pp. 1–13, Jan. 2022, doi: 10.3390/axioms11010034.
A. AKILLI and H. ATIL, “Evaluation of Normalization Techniques on Neural Networks for the Prediction of 305-Day Milk Yield,” Turkish Journal of Agricultural Engineering Research, vol. 1, no. 2, pp. 354–367, 2020, doi: 10.46592/turkager.2020.v01i02.011.
E. A. Tjoa, I. P. Y. N. Suparta, R. Magdalena, and N. K. CP, “The use of CLAHE for improving an accuracy of CNN architecture for detecting pneumonia,” SHS Web of Conferences, vol. 139, pp. 1–7, 2022, doi: 10.1051/shsconf/202213903026.
K. Maharana, S. Mondal, and B. Nemade, “A review: Data pre-processing and data augmentation techniques,” Global Transitions Proceedings, vol. 3, no. 1, pp. 91–99, Jun. 2022, doi: 10.1016/j.gltp.2022.04.020.
H. Naveed, S. Anwar, M. Hayat, K. Javed, and A. Mian, “Survey: Image mixing and deleting for data augmentation,” Eng Appl Artif Intell, vol. 131, May 2024, doi: 10.1016/j.engappai.2023.107791.
J. Gupta, S. Pathak, and G. Kumar, “Deep Learning (CNN) and Transfer Learning: A Review,” J Phys Conf Ser, vol. 2273, no. 1, pp. 1–10, 2022, doi: 10.1088/1742-6596/2273/1/012029.
N. I. R. Yassin, “Fundus Images Classification of Diabetic Retinopathy using MobileNetV2,” International Journal of Computer Science and Mobile Computing, vol. 12, no. 5, pp. 54–63, May 2023, doi: 10.47760/ijcsmc.2023.v12i05.006.
R. Indraswari, R. Rokhana, and W. Herulambang, “Melanoma image classification based on MobileNetV2 network,” Procedia Comput Sci, vol. 197, pp. 198–207, 2022, doi: 10.1016/j.procs.2021.12.132.
Y. Wang, Z. Xiao, and G. Cao, “A convolutional neural network method based on Adam optimizer with power-exponential learning rate for bearing fault diagnosis,” Journal of Vibroengineering, vol. 24, no. 4, pp. 666–678, Feb. 2022, doi: 10.21595/jve.2022.22271.
J. Ghosh and S. Gupta, “ADAM Optimizer and CATEGORICAL CROSSENTROPY Loss Function-Based CNN Method for Diagnosing Colorectal Cancer,” 2023 International Conference on Computational Intelligence and Sustainable Engineering Solutions (CISES), pp. 470–474, 2023, doi: 10.1109/CISES58720.2023.10183491.
P. Das, S. Gupta, J. Patra, and B. Mondal, “ADAMAX optimizer and CATEGORICAL CROSSENTROPY loss function-based CNN method for diagnosing Lung cancer,” 7th International Conference on Trends in Electronics and Informatics, ICOEI 2023 - Proceedings, pp. 806–810, 2023, doi: 10.1109/ICOEI56765.2023.10126046.
A. J. E. Taiwo, E. A. Olajubu, and G. A. Aderounmu, “Development of Threat Hunting Model Using Machine Learning Algorithms for Cyber Attacks Mitigation,” Proceedings - 2022 International Conference on Computational Science and Computational Intelligence, CSCI 2022, pp. 1010–1015, 2022, doi: 10.1109/CSCI58124.2022.00179.
P. Fränti and R. Mariescu-Istodor, “Pattern Recognition Letters,” Pattern Recognit Lett, vol. 167, pp. 115–121, Mar. 2023, doi: 10.1016/j.patrec.2023.02.005.
N. G. Sorboni, J. Wang, and M. R. Najafi, “Fusion of Google Street View, LiDAR, and Orthophoto Classifications Using Ranking Classes Based on F1 Score for Building Land-Use Type Detection,” Remote Sens (Basel), vol. 16, no. 11, pp. 1–27, Jun. 2024, doi: 10.3390/rs16112011.
F. Najjar, R. T. Naha, M. M. Feridani, O. Dekhil, and K. Zhang, “Change Address Detection in Bitcoin using Hierarchical Clustering,” Proceedings - 2024 IEEE World Forum on Public Safety Technology, WFPST 2024, pp. 42–48, 2024, doi: 10.1109/WFPST58552.2024.00015.
G. K. Gupta and D. K. Sharma, “A Review of Overfitting Solutions in Smart Depression Detection Models,” 2022 9th International Conference on Computing for Sustainable Global Development (INDIACom), pp. 145–151, 2022, doi: 10.23919/INDIACom54597.2022.9763147.
Copyright (c) 2024 Khoirizqi Velarati, Christy Atika Sari, Eko Hari Rachmawanto
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License (Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) ) that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).