Improving Wheat Leaf Disease Classification: Evaluating Augmentation Strategies and CNN-Based Models With Limited Dataset

Abstract

Global food security is seriously threatened by wheat leaf disease, which makes effective and precise disease detection and classification techniques necessary. For efficient disease control and the best possible crop health, timely identification and precise classification are essential. However, the limited availability of datasets for wheat leaf diseases hinders the development of effective and robust classification models. This research emphasizes the importance of precise wheat leaf disease diagnosis for global food security. The existing methods face challenges with limited data and computational demands. The research explores the potential of deep learning for automated disease detection, considering these challenges. CycleGAN proved to be the most effective among various augmentation techniques, enhancing the performance of classifiers DenseNet121, ResNet50V2, DenseNet169, Xception, ResNet152V2, and MobileNetV2. ADASYN also significantly improved classification accuracy, with MobileNetV2 consistently outperforming across different augmentation methods. This technique excels in overcoming challenges posed by limited datasets and class imbalances. Using CycleGAN for data augmentation notably enhanced classifier performance, addressing the scarcity of real-world samples. Evaluation through confusion matrix analysis revealed a minimal number of misclassified images—possibly as low as 0 to 3 images over the test dataset. The exceptional 100% accuracy achieved by the MobileNetV2 model on both CycleGAN and ADASYN augmented datasets highlights the potential of these techniques to unlock new levels of accuracy in wheat disease classification. This augmentation technique fine-tuned the classifier, reducing errors and highlighting the crucial role of CycleGAN in enhancing the accuracy and precision of wheat disease classification models. The proposed method establishes CycleGAN’s effectiveness in augmenting wheat leaf disease classification and recognizes ADASYN’s potential. The developed technique shows promise for automated disease detection in agriculture, enhancing global food security. Future research may optimize computational efficiency and explore integrating emerging technologies such as edge computing.

Keywords

• Wheat leaf disease
• deep learning
• augmentation techniques
• CycleGAN
• SMOTE
• SMOTETomek