Tiny deep learning model for insect segmentation and counting on resource-constrained devices
Year: 2025
Type: article
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 1MB of RAM and 2MB 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 965 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 3.6 seconds and consumes 4.9J. In terms of segmentation, it achieved a Dice Similarity Coefficient (DSC) of 87% and an Intersection over Union (IoU) of 77% with a precision and recall of 86% and 89%, respectively. In terms of counting, it achieved a Mean Square Error (MSE) of 1.55, Mean Absolute Error (MAE) of 0.84 and R2 of 0.97.
Authors: Amin Kargar, Dimitrios Zorbas, Michael Gaffney, Brendan O’Flynn, Salvatore Tedesco
Language: English
Subfield: Computer Vision and Pattern Recognition
Field: Computer Science
Domain: Physical Sciences
Open Access status: hybrid
Estimated APC paid: $3,870
OpenAlex ID: W4409536083