针对田间植物病害少样本分类的双阶段参数高效调优方法

《Computers and Electronics in Agriculture》:Dual-stage parameter-efficient tuning for in-field plant disease few-shot classification

【字体: 大 中 小 】 时间:2026年04月19日 来源:Computers and Electronics in Agriculture 8.9

编辑推荐:

  针对田间植物疾病数据稀缺且视觉复杂的问题,提出两阶段参数高效自适应方法。离线阶段训练轻量级适配器提取领域知识,在线阶段快速适应少量样本。实验表明,在PlantDoc等数据集上,5-way 1-shot和5-way 5-shot准确率分别达76.13%和89.00%,优于基线方法12.01%和7.68%。

  
Saeed Khankalantary | Mohammad Reza Eskandari
伊朗德黑兰K. N. Toosi技术大学电气工程系

摘要

在本文中,我们解决了对现场植物疾病数据进行分类的挑战,这些数据既稀缺又在视觉上非常复杂。现有方法通常分别处理这些问题,但当在实际环境中的边缘设备上部署时,它们的性能往往会下降。为了克服这两个问题,我们提出了一种两阶段、参数效率高的适应流程,该流程在多个目标领域都具有很强的经验鲁棒性。在离线领域适应阶段,模型适配器会在复杂的植物疾病和领域相关数据集上进行广泛训练,以获取特定领域的知识。随后,在测试阶段,新的在线适配器通过快速的特定任务少量样本测试对新类别进行分类。在具有挑战性的现场数据集(PlantDoc、PlantSeg、PlantWild)上,我们的模型平均准确率达到了76.13%(5路1次样本)和89.00%(5路5次样本),分别比基础迁移学习流程提高了12.01%和7.68%。由于其适应策略、模块化设计和参数效率,我们的方法在数据量少的情况下表现出很强的鲁棒性,并且在评估的数据集中对新的疾病类型也适用;在升级骨干架构时不需要昂贵的重新训练,并且对现场边缘设备的部署几乎没有内存和计算开销。

引言

早期研究表明,深度卷积神经网络在实验室收集的植物图像上的表现明显优于传统的机器学习算法,如支持向量机和k最近邻算法。这一改进主要得益于PlantVillage数据集(Hughes和Salathe,2015年)的可用性,该数据集是一个包含大量标记叶片图像的广泛实验室策划的集合,为这些监督式深度模型提供了所需的样本量。大多数开创性研究使用了标准的迁移学习CNN架构,并将其适配到这个数据集上,报告了许多疾病类别的高准确率,从而确立了监督式深度学习流程作为控制环境中的主导方法(Fuentes等人,2017年;Long等人,2018年;Ma等人,2018年;Zhang等人,2018年;Akshai和Anitha,2021年;Dhiman等人,2023年)。这些结果激发了后续研究,这些研究改进了网络架构和数据处理流程,以实现更快、更准确的检测(Ferentinos,2018年;Saleem等人,2019年;Hasan等人,2020年;Rahman等人,2020年;Xiong等人,2021年;Yu等人,2022年;Haque和Sohel,2022年;Ahmad等人,2023年;Alaeddine和Jihene,2023年;Alqahtani等人,2023年;Dai等人,2023年;Pan等人,2023年)。
然而,现场部署要复杂得多。例如,在现实世界的侦察图像中,叶片在光线变化极大的情况下出现,存在遮挡,并且处于多种植物场景中,背景也不明确。在实验室图像上训练的模型在野外往往表现不佳。边缘部署面临的另一个挑战是数据稀缺。获取和注释大型现场数据集成本高昂且往往不可行。它们需要专家意见和特定的实验室设备来区分某些疾病类型,而且有些疾病较为罕见,或者可能只是作为新的入侵事件出现。尽管如此,越来越多的研究已经收集了“现场”数据集,并使用迁移学习方法进行了特定领域的评估,但与实验室测试平台相比,封闭集准确率大幅下降(Singh等人,2020年;Wei等人,2024a;Wei等人,2024b;Yang等人,2024年)。与PlantVillage相比,这些数据集仍然较小,并且没有解决植物疾病图像的基本数据量问题,特别是在部署到大型模型(如十亿参数的视觉变换器(ViT)网络)时,这些模型通常需要相对大量的训练数据。尽管已经采用了一些数据增强和合成数据技术来提高鲁棒性,但这些方法对于现场图像来说往往仍然不够。这主要是由于疾病症状与多样化场景内容的相互作用以及难以合成捕获的伪影(Arun Pandian等人,2019年;Cap等人,2022年;Enkvetchakul和Surinta,2022年;Nagaraju等人,2022年;Haruna等人,2023年)。至关重要的是,这种严重的领域差异(从实验室到现场)和极端的数据稀缺性将问题置于典型的少量样本学习和领域适应之间。为此,我们必须从少量示例中泛化到新类别并纠正系统性的、数据集级别的视觉偏差。这一观察直接激发了我们两阶段的设计:(i)一个离线阶段,从汇总的源数据和领域相关数据中构建轻量级的领域相关先验;(ii)一个非常快速的在线 episodic 适配器,在测试时将模型专门化到本地支持集。
少量样本学习(FSL)方法被引入来减少对大型标注数据集的需求,通过元训练相关任务来从少量示例中学习新类别(Parnami和Lee,2022年)。在此基础上,跨领域少量样本学习(CDFSL)通过元训练跨异构源领域(例如,鸟类、汽车、医学扫描)来扩展FSL。在CDFSL中,目标是在元训练期间让模型从多样化领域学习可泛化的表示,以便它们能够用有限的数据适应新的、之前未见过的领域。这一框架使模型能够获得领域无关的知识,从而改善对训练期间未见过的目标领域的适应能力。CDFSL在Meta-Dataset(Triantafillou等人,2020年)等基准测试以及其他类似复杂领域(如医学图像、语言翻译和手写)上取得了显著的改进。然而,尽管显然需要从实验室到现场进行泛化,但其在植物疾病识别中的应用仍然有限。传统上,在典型的FSL中,新类别与训练类别具有相同的视觉领域(相似的传感器、背景和对象类型),因此任务和领域紧密对齐,主要挑战是在提供少量示例的情况下进行类别分离。相比之下,CDFSL通常假设新类别来自不同的领域(不同的图像统计特征、背景、传感器或模态),因此模型必须同时处理类别泛化和领域转换(Li和Yang,2021年)。
在植物病理学中,尽管减少了数据需求,但大多数先前的FSL工作假设使用的是干净的、实验室风格的图像,这限制了其在现实环境中的鲁棒性和对新疾病出现的适应性。植物疾病识别的任务介于FSL和CDFSL之间(Argüeso等人,2020年;Chen等人,2021年;Li等人,2021年;Li等人,2022年;Sun等人,2024年)。在植物组织上识别疾病症状的任务在很大程度上是稳定的,标签语义(患病与健康或特定病变类别)在数据集之间保持一致,因此从这个意义上说,这个问题类似于FSL。然而,这些标签的外观高度依赖于领域。宿主形态(叶片形状、叶脉)、症状表现(斑点大小、颜色、纹理)、成像条件(现场杂乱、光线、遮挡)以及特定物种的症状学在不同数据集之间可能会有很大变化。实际上,这意味着在实验室照片中学会了一种锈病病变“看起来”是什么样的模型可能仍然难以在另一种作物或现场图像中识别锈病,因为叶片、土壤和阴影引入了新的混淆因素。简而言之,任务建议使用少量样本方法,而数据分布迫使我们结合领域适应技术。由于植物疾病识别需要少量样本泛化和领域校正,我们将任务定义为领域自适应少量样本学习(DA-FSL)(Zhao等人,2021年),并采用了两阶段离线到在线的适配器策略。
从头开始为每个新领域训练一个高容量的视觉模型在时间、计算资源和标注数据方面代价高昂,因此在资源丰富的实验室之外很少实用。常见的替代方法是从预训练的骨干模型开始,并将其适配到目标领域,如迁移学习(Atila等人,2021年;Hassan等人,2021年;N等人,2021年;Xu等人,2022b;Paymode和Malode,2022年)。适应策略通常会在容量、内存、时间成本和鲁棒性之间进行权衡。完全微调会更新所有模型权重,并且在数据充足的情况下可以实现最高的准确率。然而,这在计算上非常耗时,并且在标签稀缺时容易过拟合和灾难性遗忘。另一方面,仅头部调整或线性探针(Alain和Bengio,2017年)(冻结骨干模型并仅训练最终分类器)快速且稳定。然而,它缺乏纠正重大领域转换的能力,并且在少量样本泛化方面的表现不佳。另一种方法是持续的自我监督预训练(Caccia和Pineau,2022年)(MIM / DINO / MAE),当目标视觉与骨干模型的训练来源不同时,这可以显著提高领域迁移能力。然而,这也通常需要解冻和更新骨干模型的大部分权重,使其与完全微调一样昂贵。为了解决无需完全微调的领域适应问题同时保持高适应性,提出了参数高效的微调(PEFT)方法(Xu等人,2023年)。这些方法,如Houlsby瓶颈适配器(Houlsby等人,2019年)、LoRA低秩更新(Hu等人,2022年)、SSF缩放和移位层(Lialin等人,2024年)、BitFit(Ben-Zaken、Ravfogel和Goldberg,2022年)和软提示调整(Tian等人,2023年),只训练少量额外的参数,从而保留了预训练的骨干模型,并允许进行领域特定的校正,同时占用很小的内存和存储空间。
在这项工作中,我们提出了一种实用的两阶段流程,明确针对植物疾病迁移问题的混合性质。首先,在离线阶段,我们在汇总的目标和领域相关训练数据上拟合一系列小型、参数效率高的适配器,以使冻结的ViT骨干模型暴露于现场变异性,并提取领域相关的特征先验。其次,在在线阶段,我们通过将紧凑的在线适配器插入后续层并仅调整这些模块(保持骨干和离线适配器冻结)来对小型支持集进行轻量级的 episodic 适应,以提取少量样本分类所需的精细、特定于类别的信息。这种分离使我们能够提前捕获广泛的领域因素,同时允许在测试时快速、低成本地进行专门化,通常每次调整的数量约为几十万个参数(约占整个骨干模型的0.5%)。然后,我们在四个与植物相关的领域(VGG Flowers、Danish Fungi、Pests-102、PlantVillage)(Nilsback和Zisserman,2008年;Hughes和Salathe,2015年;Wu等人,2019年;Picek等人,2022年)和三个目标复杂的现场领域(PlantDoc、PlantWild、PlantSeg v2)(Singh等人,2020年;Wei等人,2024a;Wei等人,2024b)上评估了我们提出的模型。
在本文中,我们:
• 进行了消融研究,以测量每个设计选择和超参数(如骨干预训练源、离线适应数据、训练策略、增强流程和适配器变体)对复杂植物疾病数据分类准确性的影响,以及结果在领域内和领域相关评估之间的差异。
• 在5路1次样本和5路5次样本设置中,与基础骨干模型相比,准确率提高了多达12.01%(从64.12%提高到76.13%)和7.65%(从81.35%提高到89.00%),证明了在领域内知识和参数高效测试时调整如何提高性能。
在接下来的部分中,我们首先讨论第2节中先前的研究人员探索的与植物疾病、PEFT适配器、深度学习和少量样本模型相关的工作。然后我们在第3节描述我们的模型设计,接着是第4节的实验设置、数据集和消融结果。第5节作为结论。

章节片段

相关工作

在视觉任务中,关于少量样本学习和跨领域少量样本学习的开创性工作主要依赖于卷积骨干模型,如ResNets(He等人,2016年),这些模型通常包含数千万个参数。随着变换器的引入(Vaswani等人,2023年),模型规模大幅增长,现在通常从数亿个参数到超过十亿个参数不等。变换器最初在自然语言处理中流行起来,并显示出更高的

模型

我们将最终任务形式化为N路K次样本分类,这要求模型在测试时仅根据每个类别nk个标记的支持图像,就能可靠地区分n个新类别。为了解决这个问题,我们遵循了原型网络(Snell、Swersky和Zemel,2017年)的现有工作,并提出了一个三阶段设计。我们从预训练的视觉骨干模型开始,通过离线监督适应注入轻量级的参数高效(PEFT)适配器来捕获领域特定的先验

实验和结果

在本节中,我们首先对在线适配器及其相关超参数进行了一组消融研究。然后,利用这些结果,我们对离线适配器进行了第二组消融研究,重点关注骨干训练数据集、离线适配器训练数据集、数据增强策略和训练策略。最终,我们使用所有这些实验的结果来评估最终模型作为我们的最终结果。

结论

在本文中,我们设计了一个两阶段的领域和任务适应模型,使用了众所周知的流行PEFT适配器方法,并在复杂的现场数据上评估了每个设计决策。我们证明了在线适配器可以在测试期间轻微改善适应能力,而解决领域适应问题的离线适配器对准确率改进的影响更大。我们证明了使用更多样化的训练骨干模型作为基础模型可以获得略好的整体结果。

CRediT作者贡献声明

Saeed Khankalantary:写作——审稿与编辑、验证、监督、项目管理。Mohammad Reza Eskandari:写作——原始草稿、可视化、软件、资源、方法论、调查、形式分析、数据整理、概念化。

利益冲突声明

作者声明他们没有已知的竞争性财务利益或个人关系可能会影响本文报告的工作。
相关新闻
生物通微信公众号
微信
新浪微博
  • 搜索
  • 国际
  • 国内
  • 人物
  • 产业
  • 热点
  • 科普

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号