Multiple Diseases and Pests Detection Based on Federated Learning and Improved Faster R-CNN

Abstract

Traditional disease and pest detection technology employ cloud based deep learning, which facing the pressures such as high data storage and communication costs, unbalanced and insufficient data from orchards, diversity of pests and diseases, and complex detection environments. In this paper, we propose a multiple pest detection technique based on Federated Learning (FL) and improved Faster Region Convolutional Neural Network (R-CNN). As the new distributed computing model, FL can derive a shared model integrating the advantages of data from all parties without uploading local data, and also reduces the communication cost. A restriction M is added to the FL algorithm to ensure the convergence of the model and improve the training speed. According to the original Faster R-CNN network, ResNet-101 is used instead of VGG-16 to construct the base convolutional layer to maintain the original structure of small-sized targets and improve the detection speed. Then, the multi-size fusion of feature maps from different convolutional layers is performed to improve the detection accuracy of multi-size multiple pests and diseases. Finally, a Soft-NMS algorithm is proposed to solve the apple obscured problem after the RPN network. The experimental results show that the improved Faster R-CNN can achieve an average accuracy of 90.27% on multiple pest detection, and the detection time is only 0.05 seconds per image. After using FL, the mAP of the model reached 89.34% and the model training speed was improved by 59%.