HGSM-YOLO: A Small-Lesion-Oriented Lightweight YOLO11n Framework for Citrus Leaf Disease Detection
Accurate and rapid detection of citrus leaf diseases is important for early diagnosis, precision orchard management, and the reduction of economic losses in citrus production. Automatic detection remains difficult because early lesions are often small and irregular. Several disease categories also share similar visual appearances, and localization is easily affected by veins, shadows, and cluttered backgrounds. To address these task-specific challenges, we propose HGSM-YOLO, where HGSM denotes the coordinated use of heterogeneous convolution, a GSConv-based slim neck, and multi-scale dilated local attention. The framework is built on YOLO11n because its 2.59 M-parameter and 6.4 GFLOP design provides a stringent compact baseline for edge-oriented improvement. The method follows a hierarchical design: C3k2-HetConv preserves lesion edges and local morphology in the backbone; the GSConv-based slim neck reduces part of the feature fusion cost; and an MSDA module in the high-resolution P3 branch enhances the context of small lesions. Following model selection on the validation split, the final locked models were evaluated once on the held-out test split, with HGSM-YOLO reaching 77.5% precision, 66.8% recall, 71.7% F1-score, 70.8% [email protected], and 44.2% [email protected]:0.95, compared with 68.5%, 61.5%, 64.8%, 66.0%, and 40.2% for YOLO11n. A stratified outer five-fold cross-validation further yields 71.0% ± 1.4% [email protected] and 44.4% ± 1.1% [email protected]:0.95 for HGSM-YOLO, versus 65.9% ± 1.1% and 40.2% ± 0.9% for YOLO11n. On the independent 1871-image citrus-leaf-disease-2 dataset, retraining under the same protocol gives 94.4% [email protected] for HGSM-YOLO versus 92.2% for YOLO11n and 93.1% for the public Roboflow YOLOv11 reference model. The complete HGSM-YOLO architecture uses 7.2 GFLOPs, 2.82 M parameters, and runs at 90.9 FPS on the RTX 4090, compared with 6.4 GFLOPs, 2.59 M parameters, and 110.1 FPS for the baseline. Thus, the contribution provides a recall- and localization-oriented accuracy–efficiency trade-off rather than universal superiority in every individual metric.
Authors
- Xinwei Wang (ORCID: https://orcid.org/0000-0003-2791-0170)
- Rui Zheng
- Jing Zhao
- Feng Wang
Institutions
- Xijing University (CN)
Publication Details
- Journal
- Sensors
- Published
- 2026-08-24
- DOI
- https://doi.org/10.3390/s26175345
- Primary Topic
- Smart Agriculture and AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00