Convolutional Rebalancing Network for the Classification of Large Imbalanced Rice Pest and Disease Datasets in the Field

文献类型： 外文期刊

第一作者： Yang, Guofeng

作者： Yang, Guofeng;Chen, Guipeng;Li, Cong;Fu, Jiangfan;Guo, Yang;Liang, Hua;Yang, Guofeng;Chen, Guipeng;Li, Cong;Fu, Jiangfan;Guo, Yang;Liang, Hua

作者机构：

关键词： imbalanced dataset; convolutional neural network; image classification; feature fusion; rice pests and diseases

期刊名称：FRONTIERS IN PLANT SCIENCE （ 影响因子：5.754； 五年影响因子：6.612 ）

ISSN： 1664-462X

年卷期： 2021 年 12 卷

页码：

收录情况： SCI

摘要： The accurate classification of crop pests and diseases is essential for their prevention and control. However, datasets of pest and disease images collected in the field usually exhibit long-tailed distributions with heavy category imbalance, posing great challenges for a deep recognition and classification model. This paper proposes a novel convolutional rebalancing network to classify rice pests and diseases from image datasets collected in the field. To improve the classification performance, the proposed network includes a convolutional rebalancing module, an image augmentation module, and a feature fusion module. In the convolutional rebalancing module, instance-balanced sampling is used to extract features of the images in the rice pest and disease dataset, while reversed sampling is used to improve feature extraction of the categories with fewer images in the dataset. Building on the convolutional rebalancing module, we design an image augmentation module to augment the training data effectively. To further enhance the classification performance, a feature fusion module fuses the image features learned by the convolutional rebalancing module and ensures that the feature extraction of the imbalanced dataset is more comprehensive. Extensive experiments in the large-scale imbalanced dataset of rice pests and diseases (18,391 images), publicly available plant image datasets (Flavia, Swedish Leaf, and UCI Leaf) and pest image datasets (SMALL and IP102) verify the robustness of the proposed network, and the results demonstrate its superior performance over state-of-the-art methods, with an accuracy of 97.58% on rice pest and disease image dataset. We conclude that the proposed network can provide an important tool for the intelligent control of rice pests and diseases in the field.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序