仅用10张目标域样本,就能显著提升植物病害诊断模型的泛化能力。
Robust Plant Disease Diagnosis with Few Target-Domain Samples
- 基于度量学习设计新框架,利用少量目标域样本增强模型适应性
- 每病种仅需10个样本即实现7.3点平均宏F1提升
- 适合实际部署中数据稀缺的农业病害诊断场景
基于深度学习的植物病害诊断系统虽已取得优异性能,但在不同拍摄条件下常出现准确率下降,暴露了模型泛化能力不足的问题。这一现象源于病害症状的细微差异与域间差距(如图像背景、光照等)。根本原因在于训练数据多样性不足,难以应对复杂任务。为此,本文提出一种名为目标感知度量学习与优先采样(TMPS)的简单而灵活的学习框架。该方法在仅有少量目标域标注样本的前提下,有效提升诊断鲁棒性。在包含223,073张叶片图像的大规模数据集上验证,涵盖21种病害及健康样本,覆盖三种作物。仅需每病种10个目标域样本,TMPS即超越联合源/目标数据训练模型,以及预训练后微调的模型,在平均宏F1上分别提升7.3和3.6点,并较基线与传统度量学习提升18.7和17.1点。
原文摘要 · Abstract (English)
Various deep learning-based systems have been proposed for accurate and convenient plant disease diagnosis, achieving impressive performance. However, recent studies show that these systems often fail to maintain diagnostic accuracy on images captured under different conditions from the training environment -- an essential criterion for model robustness. Many deep learning methods have shown high accuracy in plant disease diagnosis. However, they often struggle to generalize to images taken in conditions that differ from the training setting. This drop in performance stems from the subtle variability of disease symptoms and domain gaps -- differences in image context and environment. The root cause is the limited diversity of training data relative to task complexity, making even advanced models vulnerable in unseen domains. To tackle this challenge, we propose a simple yet highly adaptable learning framework called Target-Aware Metric Learning with Prioritized Sampling (TMPS), grounded in metric learning. TMPS operates under the assumption of access to a limited number of labeled samples from the target (deployment) domain and leverages these samples effectively to improve diagnostic robustness. We assess TMPS on a large-scale automated plant disease diagnostic task using a dataset comprising 223,073 leaf images sourced from 23 agricultural fields, spanning 21 diseases and healthy instances across three crop species. By incorporating just 10 target domain samples per disease into training, TMPS surpasses models trained using the same combined source and target samples, and those fine-tuned with these target samples after pre-training on source data. It achieves average macro F1 score improvements of 7.3 and 3.6 points, respectively, and a remarkable 18.7 and 17.1 point improvement over the baseline and conventional metric learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。