PlantClassiNet: A Dual-Modal Fine-Tuning Framework for CNN-Based Plant Disease Classification

Abstract

Although Convolutional Neural Networks (CNNs) have delivered state-of-the-art accuracy in plant disease classification, their deployment is still hindered by data scarcity, computational cost, and architectural heterogeneity. Transfer learning from large-scale pre-trained datasets alleviates these issues, yet generic feature extraction suffers from domain shift, while indiscriminate fine-tuning risks over-fitting and elevated training budgets. To address the identified limitations, PlantClassiNet is implemented as a unified framework. This framework facilitates systematic comparative analysis of six CNN architectures, AlexNet, ResNet50, InceptionV3, MobileNetV3Small, DenseNet121 and EfficientNetB0, across three publicly available datasets: PlantVillage, PlantLeaves and Eggplant. Two alternative fine-tuning approaches are proposed: Adaptive Fine-tuning (AdapFitu), which adaptively determines the depth of unfrozen layers, learning rates, and reinitializes selected layers, and a fixed-parameter baseline, which trains only the newly added classifier while keeping the convolutional backbone frozen and unfreezes a fixed number of network layers for retraining. Extensive experiments demonstrate that large models AlexNet, ResNet50, and Inceptionv3 achieve test accuracy exceeding 98.74% on the sizable PlantVillage dataset, whereas lightweight counterparts MobileNetV3Small, DenseNet121, and EfficientNetB0 achieve high accuracy of 99.79% ± 0.21% on the smaller Eggplant collection after fine-tuning.