A Hybrid Deep Learning Model with Bayesian Optimization Technique for Leaf Disease Classification

Detecting plant diseases on time is imperative to improve agricultural productivity and mitigate economic losses.  This study presents a novel artificial intelligence framework for classifying plant diseases, combining advanced deep learning technologies with a hyperparameter optimization strategy.  Specifically, we employed a hybrid architecture that concatenates two pre-trained models, MobileNetV2 and DenseNet201, enriched with custom layers developed by the researchers.  Bayesian Optimization was employed to enhance model performance, focusing on four critical hyperparameters: learning rate, number of units, optimizer, and dropout rate.  The methodology integrates data augmentation, transfer learning, and fine-tuning to enhance feature extraction and model robustness.  The results demonstrate remarkable performance, with training accuracy at 98.95%, validation accuracy at 99.83%, and testing accuracy at 99.51%.  These findings underscore the efficacy of sophisticated AI methodologies in automating the detection of plant diseases, indicating substantial progress in precision agriculture and contributing to sustainable food security.

  1. Rizzo D. M., Lichtveld M., Mazet J. A. K. et al.  Plant health and its effects on food safety and security in a One Health framework: four case studies.  One Health Outlook.  3 (1), 6 (2021).
  2. Amanullah Ed.  Agronomy – Climate Change and Food Security. IntechOpen (2020).
  3. Buja I., Sabella E., Monteduro A. G. et al.  Advances in Plant Disease Detection and Monitoring: From Traditional Assays to In-Field Diagnostics.  Sensors.  21 (6), 2129 (2021).
  4. Ahishakiye E., Van Gijzen M. B., Tumwiine J., Wario R., Obungoloch J.  A survey on deep learning in medical image reconstruction.  Intelligent Medicine.  1 (3), 118–127 (2021).
  5. Yang S., Zhu F., Ling X., Liu Q., Zhao P.  Intelligent Health Care: Applications of Deep Learning in Computational Medicine.  Frontiers in Genetics.  12, 607471 (2021).
  6. Vo H.-T., Ngoc H. T., Quach L.-D.  An Approach to Hyperparameter Tuning in Transfer Learning for Driver Drowsiness Detection Based on Bayesian Optimization and Random Search.  International Journal of Advanced Computer Science and Applications.  14 (4), (2023).
  7. Murataj J., Alkholidi A., Hamam H., Alimeti A.  A review of deep learning for self-driving cars: case study.  CIT Review Journal.  1, 16–26 (2023).
  8. Attri I., Awasthi L. K., Sharma T. P., Rathee P.  A review of deep learning techniques used in agriculture.  Ecological Informatics.  77, 102217 (2023).
  9. Saleem M. H., Potgieter J., Arif K. M.  Automation in Agriculture by Machine and Deep Learning Techniques: A Review of Recent Developments.  Precision Agriculture.  22 (6), 2053–2091 (2021).
  10. Vo H.-T., Thien N. N., Mui K. C.  Tomato Disease Recognition: Advancing Accuracy Through Xception and Bilinear Pooling Fusion.  International Journal of Advanced Computer Science and Applications.  14 (8), (2023).
  11. Vo H.-T., Quach L.-D., Ngoc H. T.  Ensemble of Deep Learning Models for Multi-plant Disease Classification in Smart Farming.  International Journal of Advanced Computer Science and Applications.  14 (5), (2023).
  12. Wu J., Chen X.-Y., Zhang H., Xiong L.-D., Lei H., Deng S.-H.  Hyperparameter Optimization for Machine Learning Models Based on Bayesian Optimization.  Journal of Electronic Science and Technology.  17 (1), 26–40 (2019).
  13. Shaheen A.  Weed Detection Using Lightweight Deep Learning Models with Transfer Learning and Hyperparameter Optimization.  PhD Thesis, Dakota State University (2024).
  14. Hossain R., Timmer D. D., Moya H.  Machine learning model optimization with hyper-parameter tuning approach.  2021 International Conference on Advanced Engineering, Technology and Applications (2021).
  15. Bhandari M., Shahi T. B., Neupane A., Walsh K. B.  BotanicX-AI: Identification of Tomato Leaf Diseases Using an Explanation-Driven Deep-Learning Model.  Journal of Imaging.  9 (2), 53 (2023).
  16. Ullah N., Marzougui M., Ahmad I., Chelloug S. A.  DeepLungNet: An Effective DL-Based Approach for Lung Disease Classification Using CRIs.  Electronics.  12 (8), 1860 (2023).
  17. Latif G., Abdelhamid S. E., Mallouhy R. E., Alghazo J., Kazimi Z. A.  Deep Learning Utilization in Agriculture: Detection of Rice Plant Diseases Using an Improved CNN Model.  Plants.  11 (17), 2230 (2022).
  18. Guerrero-Ibañez A., Reyes-Muñoz A.  Monitoring Tomato Leaf Disease through Convolutional Neural Networks.  Electronics.  12 (1), 229 (2023).
  19. Liu S., Bai H., Li F. et al.  An apple leaf disease identification model for safeguarding apple food safety.  Food Science and Technology.  43, e104322 (2023).
  20. Ottoni A. L. C., De Amorim R. M., Novo M. S., Costa D. B.  Tuning of data augmentation hyperparameters in deep learning to building construction image classification with small datasets.  International Journal of Machine Learning and Cybernetics.  14 (1), 171–186 (2023).
  21. Vo H.-T., Mui K. C., Thien N. N., Tien P. P., Le H. L.  Optimizing Grape Leaf Disease Identification Through Transfer Learning and Hyperparameter Tuning.  International Journal of Advanced Computer Science and Applications.  15 (2), (2024).
  22. Lu Q. T.  An Approach for Classification of Diseases on Leaves.  International Journal of Advanced Computer Science and Applications.  14 (10), (2023).
  23. Hajraoui N., Azrour M., El Allaoui A.  Classification of diseases in tomato leaves with Deep Transfer Learning.  Data and Metadata.  2, 181 (2023).
  24. Kumar R., Singh D., Chug A., Singh A. P.  Evaluation of Deep learning based Resnet-50 for Plant Disease Classification with Stability Analysis.  2022 6th International Conference on Intelligent Computing and Control Systems (ICICCS).  1280–1287 (2022).
  25. Hajraoui N., Azrour M., Farhaoui Y., El Allaoui A.  Plant Disease Classification Using Deep Learning and the Hyperband Strategy.  Intersection of Artificial Intelligence, Data Science, and Cutting-Edge Technologies: From Concepts to Applications in Smart Environment.  709–717 (2025).
  26. Zahra U., Khan M. A., Alhaisoni M., Alasiry A., Marzougui M., Masood A.  An Integrated Framework of Two-Stream Deep Learning Models Optimal Information Fusion for Fruits Disease Recognition.  IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.  17, 3038–3052 (2024).
  27. Wang Y., Wang H., Peng Z.  Rice Diseases Detection and Classification Using Attention-Based Neural Network and Bayesian Optimization.  Expert Systems with Applications.  178, 114770 (2021).