Preprint Article Version 1 Preserved in Portico This version is not peer-reviewed

LBFNet: A Tomato Leaf Disease Identification Model based on Three-Channel Attention Mechanism and Quantitative Pruning

Version 1 : Received: 14 April 2023 / Approved: 14 April 2023 / Online: 14 April 2023 (10:58:14 CEST)

A peer-reviewed article of this Preprint also exists.

Chen, H.; Wang, Y.; Jiang, P.; Zhang, R.; Peng, J. LBFNet: A Tomato Leaf Disease Identification Model Based on Three-Channel Attention Mechanism and Quantitative Pruning. Appl. Sci. 2023, 13, 5589. Chen, H.; Wang, Y.; Jiang, P.; Zhang, R.; Peng, J. LBFNet: A Tomato Leaf Disease Identification Model Based on Three-Channel Attention Mechanism and Quantitative Pruning. Appl. Sci. 2023, 13, 5589.

Abstract

The current neural networks for tomato leaf disease recognition have problems such as large model parameters, long training time, and low model accuracy. To solve these problems, a lightweight convolutional neural network LBFNet is proposed in this paper.First, a lightweight convolutional neural network LBFNet is established as the base model. Secondly, a three-channel attention mechanism module is introduced to learn the disease features in tomato leaf disease images and reduce the interference of redundant features. Finally, a cascade module is introduced to increase the depth of the model, solve the gradient descent problem, and reduce the loss caused by increasing the depth of the model. The quantized pruning technique is also used to further compress the model parameters and optimize the model performance. The results show that the LBFNet model achieves 99.06% accuracy on the LBFtomato dataset, with a training time of 996s and a single classification accuracy of over 94%. Further training using the saved weight file after quantized pruning makes the model accuracy reach 97.66%. Compared with the base model, the model accuracy was improved by 28%, and the model parameters were reduced by 96.7% compared with the traditional Resnet50. It was found that LBFNet can quickly and accurately identify tomato leaf diseases in complex environments, providing effective assistance to agricultural producers.

Keywords

artificial intelligence; three-channel attention mechanism; Tomato Leaf Disease;convolution neural network; deep learning

Subject

Biology and Life Sciences, Agricultural Science and Agronomy

Comments (0)

We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.

Leave a public comment
Send a private comment to the author(s)
* All users must log in before leaving a comment
Views 0
Downloads 0
Comments 0


×
Alerts
Notify me about updates to this article or when a peer-reviewed version is published.
We use cookies on our website to ensure you get the best experience.
Read more about our cookies here.