Work

Training and Optimization of a Rice Disease Detection Model Based on Ensemble Learning

Year: 2025

Type: article

Abstract: Accurate and reliable detection of rice diseases and pests is crucial for ensuring food security. However, traditional deep learning methods often suffer from high rates of missed and false detections when dealing with complex field environments, especially in the presence of tiny disease spots, due to insufficient feature extraction capabilities. To address this issue, this study proposes a high-precision rice disease detection method based on ensemble learning and conducts experiments on a self-built dataset of 12,572 images containing five types of diseases and one type of pest. The ensemble learning model is optimized and constructed through a phased approach: First, using YOLOv8s as the baseline, transfer learning is performed with the agriculture-related dataset PlantDoc. Subsequently, a P2 small-object detection head, an EMA mechanism, and the Focal Loss function are introduced to build an optimized single model, which achieves an mAP_0.5 of 0.899, an absolute improvement of 5.5% compared to the baseline YOLOv8s. Then, three high-performance YOLO object detection models, including the improved model mentioned above, are selected, and the Weighted Box Fusion technique is used to integrate their prediction results to construct the final Ensemble-WBF model. Finally, the AP_0.5 and AR_0.5:0.95 of the model reach 0.922 and 0.648, respectively, with absolute improvements of 2.2% and 3.2% compared to the improved single model, further reducing the false and missed detection rates. The experimental results show that the ensemble learning method proposed in this study can effectively overcome the interference of complex backgrounds, significantly improve the detection accuracy and robustness for tiny and similar diseases, and reduce the missed detection rate, providing an efficient technical solution for the accurate and automated monitoring of rice diseases in real agricultural scenarios.

Source: Agriculture

DOI: 10.3390/agriculture15212283

Authors: Jihong Sun, Peng Tian, Jiawei Zhao, Haokai Zhang, Ye Qian

Institutions: Kunming University, Yunnan Agricultural University, China Agricultural University

Language: English

Topic: Smart Agriculture and AI

Subfield: Plant Science

Field: Agricultural and Biological Sciences

Domain: Life Sciences

SDG: Zero hunger

Open Access status: gold

Estimated APC paid: $3,137