Please login first
A compressed convolutional neural network model for rice yield detection at ripening stage using weight pruning
1 , * 2 , * 3 , * 4
1  Dept. of Computer Science Federal University Kashere Gombe, Nigeria
2  Dept. of Software Engr. Bayero University Kano, Nigeria
3  Software Engineering Lab University of Aizu-Japan
4  Dept. of Computer Science Bayero University Kano, Nigeria
Academic Editor: Alessandro Bruno

https://doi.org/10.3390/ASEC2023-15518 (registering DOI)
Abstract:

This paper proposes a compressed Convolutional Neural Network (CNN) model for rice yield detection using the weight pruning technique. The initial CNN model achieved an accuracy of 91% on a dataset comprising 3120 images of both yield and unyield rice crops. However, it had a large size of approximately 603MB, posing challenges in terms of deployment and storage. To address this issue, weight pruning was applied to compress the model. The compressed model achieved a significant reduction in size to 186MB, representing a reduction of approximately 69.15%, while maintaining a reasonable accuracy of 86%.

The experiment was conducted in three phases. First, a dataset of 3120 images of yield and unyield rice crops was collected from different farms in Kano metropolis of Nigeria and preprocessed by resizing them to 250x250 pixels. Secondly, a CNN model with 12 layers was designed and trained using the preprocessed dataset. The model achieved an accuracy of 86%. Finally, weight pruning was applied to the trained CNN model to reduce its size. The compressed model exhibited a size of 300MB and an accuracy of 86%.

The results of this study demonstrate the effectiveness of weight pruning as a viable technique for compressing CNN models without significantly compromising their accuracy. The compressed model, with its reduced size, is well-suited for deployment on resource-constrained devices for rice yield detection applications.

Keywords: Convolutional Neural Network, rice yield detection, weight pruning, image compression.

 
 
Top