ConvGeM-next: a deep learning framework for plant disease detection

Plant diseases pose a major challenge to sustainable agriculture, particularly in regions that heavily depend on farming. Early and accurate identification of plant diseases is crucial for ensuring food production and minimizing crop losses. The rapid advancement of deep learning, particularly in convolutional neural networks (CNNs), has significantly enhanced plant disease classification performance. However, many models often struggle to generalize effectively in real-world scenarios due to challenges such as low-intensity visuals, low contrast between the background and foreground of the suspected sample, noise, and chrominance variation.

To address the challenges mentioned above, we introduce ConvGem-NeXt, an end-to-end deep learning architecture specifically designed for fine-grained plant disease classification, built on the ConvNeXt baseline model featuring enhanced generalization capabilities. More precisely, our method incorporates a learnable Generalized Mean pooling layer and ReLU activation in the ConvNeXt model to enhance spatial feature representation, and a custom classifier head that integrates batch normalization, ReLU activation, and dropout to mitigate overfitting and improve classification accuracy.

We tested the presented model on two large-scale and diverse databases, PlantVillage and the PlantDoc. The model achieved 99.65% accuracy on the PlantVillage dataset and 94.69% accuracy on the real-world PlantDoc dataset, demonstrating the efficacy of our method for reliably classifying plant diseases.

This work contributes to the rapidly growing field of agricultural automation by providing a reliable framework for timely disease diagnosis and supporting the enhancement of crop productivity.