Few-shot disease recognition algorithm based on supervised contrastive learning

文献类型： 外文期刊

第一作者： Mu, Jiawei

作者： Mu, Jiawei;Feng, Quan;Yang, Junqi;Yang, Sen;Zhang, Jianhua;Zhang, Jianhua

作者机构：

关键词： plant disease recognition; few-shot learning; meta-learning; supervised contrastive learning; nearest-centroid classification

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

ISSN： 1664-462X

年卷期： 2024 年 15 卷

页码：

收录情况： SCI

摘要： Diseases cause crop yield reduction and quality decline, which has a great impact on agricultural production. Plant disease recognition based on computer vision can help farmers quickly and accurately recognize diseases. However, the occurrence of diseases is random and the collection cost is very high. In many cases, the number of disease samples that can be used to train the disease classifier is small. To address this problem, we propose a few-shot disease recognition algorithm that uses supervised contrastive learning. Our algorithm is divided into two phases: supervised contrastive learning and meta-learning. In the first phase, we use a supervised contrastive learning algorithm to train an encoder with strong generalization capabilities using a large number of samples. In the second phase, we treat this encoder as an extractor of plant disease features and adopt the meta-learning training mechanism to accomplish the few-shot disease recognition tasks by training a nearest-centroid classifier based on distance metrics. The experimental results indicate that the proposed method outperforms the other nine popular few-shot learning algorithms as a comparison in the disease recognition accuracy over the public plant disease dataset PlantVillage. In few-shot potato leaf disease recognition tasks in natural scenarios, the accuracy of the model reaches the accuracy of 79.51% with only 30 training images. The experiment also revealed that, in the contrastive learning phase, the combination of different image augmentation operations has a greater impact on model. Furthermore, the introduction of label information in supervised contrastive learning enables our algorithm to still obtain high accuracy in few-shot disease recognition tasks with smaller batch size, thus allowing us to complete the training with less GPU resource compared to traditional contrastive learning.

分类号：

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

微信小程序