A Dual-Branch Model Integrating CNN and Swin Transformer for Efficient Apple Leaf Disease Classification
Haiping Si,
Mingchun Li,
Weixia Li,
Guipei Zhang,
Ming Wang,
Feitao Li and
Yanling Li ()
Additional contact information
Haiping Si: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Mingchun Li: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Weixia Li: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Guipei Zhang: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Ming Wang: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Feitao Li: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Yanling Li: College of Information and Management Science, Henan Agricultural University, Zhengzhou 450046, China
Agriculture, 2024, vol. 14, issue 1, 1-20
Abstract:
Apples, as the fourth-largest globally produced fruit, play a crucial role in modern agriculture. However, accurately identifying apple diseases remains a significant challenge as failure in this regard leads to economic losses and poses threats to food safety. With the rapid development of artificial intelligence, advanced deep learning methods such as convolutional neural networks (CNNs) and Transformer-based technologies have made notable achievements in the agricultural field. In this study, we propose a dual-branch model named DBCoST, integrating CNN and Swin Transformer. CNNs focus on extracting local information, while Transformers are known for their ability to capture global information. The model aims to fully leverage the advantages of both in extracting local and global information. Additionally, we introduce the feature fusion module (FFM), which comprises a residual module and an enhanced Squeeze-and-Excitation (SE) attention mechanism, for more effective fusion and retention of both local and global information. In the natural environment, there are various sources of noise, such as the overlapping of apple branches and leaves, as well as the presence of fruits, which increase the complexity of accurately identifying diseases on apple leaves. This unique challenge provides a robust experimental foundation for validating the performance of our model. We comprehensively evaluate our model by conducting comparative experiments with other classification models under identical conditions. The experimental results demonstrate that our model outperforms other models across various metrics, including accuracy, recall, precision, and F1 score, achieving values of 97.32%, 97.33%, 97.40%, and 97.36%, respectively. Furthermore, detailed comparisons of our model’s accuracy across different diseases reveal accuracy rates exceeding 96% for each disease. In summary, our model performs better overall, achieving balanced accuracy across different apple leaf diseases.
Keywords: dual branch; feature fusion; image classification; CNN; Transformer (search for similar items in EconPapers)
JEL-codes: Q1 Q10 Q11 Q12 Q13 Q14 Q15 Q16 Q17 Q18 (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:
Downloads: (external link)
https://www.mdpi.com/2077-0472/14/1/142/pdf (application/pdf)
https://www.mdpi.com/2077-0472/14/1/142/ (text/html)
Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.
Export reference: BibTeX
RIS (EndNote, ProCite, RefMan)
HTML/Text
Persistent link: https://EconPapers.repec.org/RePEc:gam:jagris:v:14:y:2024:i:1:p:142-:d:1321695
Access Statistics for this article
Agriculture is currently edited by Ms. Leda Xuan
More articles in Agriculture from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().