YOLOv11-RDTNet: A Lightweight Model for Citrus Pest and Disease Identification Based on an Improved YOLOv11n

Abstract

Citrus pests and diseases severely impact fruit yield and quality. However, existing object detection models face limitations in complex backgrounds, target occlusion, and small target recognition, and they struggle to be efficiently deployed on resource-constrained devices. To address these issues, this study proposes a lightweight pest and disease detection model, YOLOv11-RDTNet, based on the improved YOLOv11n. This model integrates multi-scale features and attention mechanisms to enhance recognition performance in complex scenarios, while adopting a lightweight design to reduce computational costs and improve deployment adaptability. The model introduces three key enhancement features: First, shallow RFD (SRFD) and deep RFD (DRFD) downsampling modules replace traditional convolution modules, improving image feature extraction accuracy and robustness. Second, the Dynamic Group Shuffle Transformer (DGST) module replaces the original C3k2 module, reducing the model’s parameter count and computational demand, further enhancing efficiency and performance. Lastly, the lightweight Task Align Dynamic Detection Head (TADDH) replaces the original detection head, significantly reducing the parameter count and improving accuracy in small-object detection. After processing the collected images, we obtained 1382 images and constructed a dataset containing five types of citrus pests and diseases: anthracnose, canker, yellow vein disease, coal pollution disease, and leaf miner moth. We applied data augmentation on the dataset and conducted experimental validation. Experimental results showed that the YOLOv11-RDTNet model had a parameter count of 1.54 MB, an mAP50 of 87.0%, and a model size of 3.4 MB. Compared to the original YOLOv11 model, the YOLOv11-RDTNet model reduced the parameter count by 40.3%, improved mAP50 by 4.8%, and reduced the model size from 5.5 MB to 3.4 MB. This model not only improved detection accuracy and reduced computational load but also achieved a balance in performance, size, and speed, making it more suitable for deployment on mobile devices. Additionally, the research findings provided an effective tool for citrus pest and disease detection with small sample sizes, offering valuable insights for citrus pest and disease detection in agricultural practices.