A hybrid convolution and attention-based framework with visual explanation for fruit disease identification.

Abstract

The objective of this study is to create a highly accurate and interpretable deep learning (DL) model for the multi-class classification of fruit using convolutional and transformer architectures. The classification performance can be enhanced by making sure that the used technique is explainable and interpretable. This research data was obtained from Kaggle which contains images of banana, grape, lemon, mango, and strawberry fruit classes. The total data was divided into 70:15:15 for training, validating and testing. To ensure consistent size and quality, all images were pre-processed before use. This study considered four pretrained models namely RegNetY-B3-GE, DarkNet53-SCSE, BEiT, and PVTv2 for performance assessment. We proposed a lightweight hybrid (convolution plus attention-based) CoAT-AgriLite model for fruit disease classification which extracts local lesion features and global context. Transferring training and data augmentation technique was utilized during training for better performance. To ensure interpretability of model decisions, Gradient-weighted Class Activation Mapping (Grad-CAM) which captures the discriminative regions from the input images for model predictions. Among all evaluated models, the proposed model achieved the highest classification accuracy of 99.37% on the testing dataset. Comparative results demonstrated that the proposed model outperformed other pretrained models in terms of precision, recall, and F1-score, confirming its robustness and effectiveness in real-world agricultural classification tasks. The experimental findings validate that the proposed model not only achieves superior classification accuracy but also provides interpretability through Grad-CAM visualizations. This hybrid framework offers a promising solution for intelligent and transparent fruit classification systems, with potential applications in precision agriculture and automated sorting systems.

Plant phenotyping relevance

果実病害を画像から分類する深層学習フレームワークを開発・比較し、病斑特徴の抽出とGrad-CAMによる説明性を評価しており、植物の病害状態の画像計測が中心である。

Code and data availability

The paper uses a Kaggle-sourced fruit image dataset and proposes the CoAT-AgriLite model, but no block provides a public URL, deposit, or availability statement for the dataset, code, or trained model. The only URLs present are license and affiliation links, none of which are paper-specific assets.

No evidence-backed public reproduction asset is currently recorded.