Tiny deep learning model for insect segmentation and counting on resource-constrained devices
Loading...
Files
Published Version
Date
2025
Authors
Kargar, Amin
Zorbas, Dimitrios
Gaffney, Michael
O'Flynn, Brendan
Tedesco, Salvatore
Journal Title
Journal ISSN
Volume Title
Publisher
Elsevier B.V.
Published Version
Abstract
Automated insect monitoring is essential for early detection of insect pest infestations in orchards. It helps growers make an informed decision to control the insect pest population in their fields to avoid crop losses and improve crop quality. This study proposed a tiny deep-learning model for insect segmentation and counting suitable for battery-powered microcontroller (MCU)-based edge devices. For this aim, the critical layers in terms of peak memory usage in a U-Net-inspired model were recognized and then optimized to meet resources constraints on an MCU with 1 MB of RAM and 2 MB of flash storage. We then introduced an image dataset for the insect of interest, Halyomorpha halys, and the dataset-splitting strategy for the model training. The proposed model was investigated from different aspects to evaluate its performance and the feasibility of its implementation on battery-powered MCU-based edge devices. The proposed deep learning model only needs approximately 900 KB of RAM and 964 KB of storage to perform its computations and store its parameters, respectively, making it useful on edge, resource constrained systems. Moreover, each inference on an MCU-based board requires 2.6 s and consumes 4.9 J. In terms of segmentation, it achieved a Dice Similarity Coefficient (DSC) of 85% and an Intersection over Union (IoU) of 73% with a precision and recall of 83% and 86%, respectively. In terms of counting, it achieved a Mean Square Error (MSE) of 1.32, Mean Absolute Error (MAE) of 0.78 and R2 of 0.97.
Description
Keywords
Edge computing , Halyomorpha halys , Image segmentation , Insect monitoring , Microcontrollers , On-device deep learning , Tiny deep learning
Citation
Kargar, A., Zorbas, D., Gaffney, M., O’Flynn, B. and Tedesco, S. (2025) 'Tiny deep learning model for insect segmentation and counting on resource-constrained devices', Computers and Electronics in Agriculture, 236, p.110378. DOI: 10.1016/j.compag.2025.110378