Enhancing Bee Mite Detection with YOLO: The Role of Data Augmentation and Stratified Sampling

초록

Beekeeping is facing a serious crisis due to climate change and diseases such as bee mites (Varroa destructor), which have led to declining populations, collapsing colonies, and reduced beekeeping productivity. Bee mites are small, reddish-brown in color, and difficult to distinguish from bees. Rapid bee mite detection techniques are essential for overcoming this crisis. This study developed a technology for recognizing bee mites and beekeeping objects in beecombs using the You Only Look Once (YOLO) object detection algorithm. The dataset was constructed by acquiring RGB images of beecombs containing mites. Regions of interest with a size of 640 x 640 pixels centered on the bee mites were extracted and labeled as seven classes: bee mites, bees, mite-infected bees, larvae, abnormal larvae, and cells. Image processing, data augmentation, and stratified data distribution methods were applied to enhance the object recognition performance. Four datasets were constructed using different augmentation and distribution strategies, including random and stratified sampling. The datasets were partitioned into training, testing, and validation sets in a 7:2:1 ratio, respectively. A YOLO-based model for the detection of bee mites and seven beekeeping-related objects was developed for each dataset. The F1 scores for the detection of bee mites and seven beekeeping-related objectives using the YOLO model based on original datasets were 94.1% and 91.9%, respectively. The model applied data augmentation, and stratified sampling achieved the highest performance, with F1 scores of 97.4% and 96.4% for the detection of bee mites and seven beekeeping-related objects, respectively. The results underscore the efficacy of using the YOLO architecture on RGB images of beecombs for simultaneously detecting bee mites and various beekeeping-related objects. This advanced mite detection method is expected to contribute significantly to the early identification of pests and disease outbreaks, offering a valuable tool for enhancing beekeeping practices.

키워드