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

一种结合卷积和注意力机制的混合框架,并提供可视化解释,用于水果病害识别

阅读:1

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.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。