Overcoming Overfitting in CNN Models for Potato Disease Classification Using Data Augmentation
DOI:
https://doi.org/10.21512/emacsjournal.v6i3.11840Keywords:
Agricultural Disease Management, Convolutional Neural Networks (CNN), Data Augmentation, Deep Learning, Potato Disease ClassificationAbstract
Classification of diseases in potato plants is crucial for agriculture to ensure quality and yield. Potatoes, being staple foods worldwide, are vulnerable to diseases that cause significant production losses. Early and accurate disease identification is essential. This study evaluates the impact of data augmentation on reducing overfitting in deep learning models for potato disease classification. Various CNN architectures, including VGG16, VGG19, Xception, and InceptionV3, were compared in transfer learning and fine-tuning phases. The "Potato Disease Dataset", consisting of 451 images across seven classes, was used. The dataset was split into training, validation, and test sets, and augmentation increased the training set from 360 to 2160 images. The results indicate that models trained with augmented data exhibited improved performance in terms of accuracy, precision, recall, and F1-scores compared to those trained without augmentation. The learning curves show that data augmentation helps in reducing overfitting and enhancing model stability. Data augmentation is crucial for developing robust deep learning models for potato disease classification. Future work will explore advanced augmentation techniques and other architectures to enhance model performance.
Plum Analytics
References
Abdulsattar, N. S., & Hussain, M. N. (2022). Facial Expression Recognition using Transfer Learning and Fine-tuning Strategies: A Comparative Study. 2022 International Conference on Computer Science and Software Engineering (CSASE), 101–106. https://doi.org/10.1109/CSASE51777.2022.9759754
Charisma, R. A., & Dharma Adhinata, F. (2023). Transfer Learning With Densenet201 Architecture Model For Potato Leaf Disease Classification. 2023 International Conference on Computer Science, Information Technology and Engineering (ICCoSITE), 738–743. https://doi.org/10.1109/ICCoSITE57641.2023.10127772
Datt, R. M., & Kukreja, V. (2022). Phenological Stage Recognition Model for Apple Crops using transfer learning. 2022 2nd International Conference on Advance Computing and Innovative Technologies in Engineering (ICACITE), 1537–1542. https://doi.org/10.1109/ICACITE53722.2022.9823711
Hamza, K., Un Nisa, S., & Irshad, G. (2022). A Review on Potato Disease Detection and Classification by exploiting Deep Learning Techniques. J. Agri. Vet. Sci, 01(2), 2022–2079. https://doi.org/10.55627/Javs.01.2.0251
Kaggle. (n.d.). https://www.kaggle.com/datasets
Mishra, S., Singh, A., & Singh, V. (2021). Application of MobileNet-v1 for Potato Plant Disease Detection Using Transfer Learning. 2021 Workshop on Algorithm and Big Data, 14–19. https://doi.org/10.1145/3456389.3456403
Moawad, N., Zaki, H., El Moniem Essa, T. abed, & Said, M. (2023). Detection of Potato Tuber Diseases Using Machine Learning Models. 2023 International Conference on Artificial Intelligence Science and Applications in Industry and Society (CAISAIS), 1–7. https://doi.org/10.1109/CAISAIS59399.2023.10269994
Nazir, T., Iqbal, M. M., Jabbar, S., Hussain, A., & Albathan, M. (2023). EfficientPNet—An Optimized and Efficient Deep Learning Approach for Classifying Disease of Potato Plant Leaves. Agriculture, 13(4). https://doi.org/10.3390/agriculture13040841
Patil, M. A., & M, M. (2023). Potato Leaf Disease Identification using Hybrid Deep Learning Model. 2023 International Conference on Network, Multimedia and Information Technology (NMITCON), 1–9. https://doi.org/10.1109/NMITCON58196.2023.10276091
Patil, N., Ingole, K., & Rajani Mangala, T. (2020). Deep convolutional neural networks approach for classification of lung diseases using x-rays: Covid-19, pneumonia, and tuberculosis. International Journal of Performability Engineering, 16(9), 1332–1340. https://doi.org/10.23940/ijpe.20.09.p2.13321340
Riyanto, S., Sitanggang, I. S., Djatna, T., & Atikah, T. D. (n.d.). Comparative Analysis using Various Performance Metrics in Imbalanced Data for Multi-class Text Classification. In IJACSA) International Journal of Advanced Computer Science and Applications (Vol. 14, Issue 6). http://gcancer.org/pdr
Sharma, O., Rajgaurang, Mohapatra, S., Mohanty, J., Dhiman, P., & Bonkra, A. (2023). Predicting Agriculture Leaf Diseases (Potato): An Automated Approach using Hyper-parameter Tuning and Deep Learning. 2023 Third International Conference on Secure Cyber Computing and Communication (ICSCCC), 490–493. https://doi.org/10.1109/ICSCCC58608.2023.10176819
Shobanadevi, A., Shanthini, A., & Hsu, H.-C. (n.d.). Potato Leaf Disease Classification using Deep Learning : A Convolutional Neural Network Approach.
Sholihati, R. A., Sulistijono, I. A., Risnumawan, A., & Kusumawati, E. (2020). Potato Leaf Disease Classification Using Deep Learning Approach. 2020 International Electronics Symposium (IES), 392–397. https://doi.org/10.1109/IES50839.2020.9231784
Sinha, K., & Lalit, M. (2022). Comparative Analysis of Pre-Trained Deep Neural Networks for Vision-Based Security Systems on a Novel Dataset. Proceedings of the 2021 IEEE/ACM 8th International Conference on Big Data Computing, Applications and Technologies, 120–127. https://doi.org/10.1145/3492324.3494173
Sitarz, M. (2022). Extending F1 metric, probabilistic approach. https://doi.org/10.54364/AAIML.2023.1161
Thangaraj, R., P, P., Kaliappan, V. K., S, A., & P, I. (2020). Potato Leaf Disease Classification using Transfer Learning based Modified Xception Model. Innovations in Information and Communication Technology Series. https://api.semanticscholar.org/CorpusID:231783905
Yacouby, R., & Axman, D. (2020). Probabilistic Extension of Precision, Recall, and F1 Score for More Thorough Evaluation of Classification Models. In S. Eger, Y. Gao, M. Peyrard, W. Zhao, & E. Hovy (Eds.), Proceedings of the First Workshop on Evaluation and Comparison of NLP Systems (pp. 79–91). Association for Computational Linguistics. https://doi.org/10.18653/v1/2020.eval4nlp-1.9
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Engineering, MAthematics and Computer Science Journal (EMACS)
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:
a. Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License - Share Alike that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this journal.
b. 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 acknowledgment of its initial publication in this journal.
c. 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.
USER RIGHTS
All articles published Open Access will be immediately and permanently free for everyone to read and download. We are continuously working with our author communities to select the best choice of license options, currently being defined for this journal as follows: Creative Commons Attribution-Share Alike (CC BY-SA)
