arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.02580cs.CVcs.AI

精准农业中用于高效标注的主动学习:一个作物-杂草语义分割用例

Active learning for efficient annotation in precision agriculture: a use-case on crop-weed semantic segmentation

Wageningen University & Research(瓦赫宁根大学与研究中心);Exxact Robotics(埃克萨克特机器人公司);University of Bordeaux(波尔多大学)

Bart M. van Marrewijk, Charbel Dandjinou, Dan Jeric Arcega Rustia, Nicolas Franco Gonzalez, Boubacar Diallo, Jérôme Dias, Paul Melki, Pieter M. Blok

AI总结:

本文针对作物-杂草语义分割的高标注成本,比较 BALD、PowerBALD 与随机采样,发现 PowerBALD 略优但受类别不平衡和图像冗余影响,差异不显著。

摘要:

优化深度学习模型需要大量带标注图像,这一过程既耗时又昂贵。对于必须标注每个像素的语义分割模型而言尤其如此。缓解标注工作量的一种潜在策略是主动学习。主动学习有助于从大型未标注图像池中识别并选择信息量最高的图像。其基本前提是,与随机选择相比,这些被选中的图像能够更快提升模型性能,从而减少标注工作量。尽管主动学习已在 Cityscapes 等基准数据集上展现出良好结果,但其在农业领域的性能仍基本未被探索。本研究通过对三种基于主动学习的采集函数进行比较研究来弥补这一研究空白:Bayesian Active Learning by Disagreement(BALD,基于分歧的贝叶斯主动学习)、基于随机方法的 BALD(PowerBALD)以及 Random(随机)。这些采集函数在两个农业数据集上进行了测试:Sugarbeet 和 Corn-Weed,二者均包含三个语义类别:背景、作物和杂草。结果表明,主动学习(尤其是 PowerBALD)在两个数据集上都取得了高于 Random 采样的性能。但由于标准差相对较大,所观察到的差异很小;其部分原因在于图像冗余度高和类别不平衡。具体而言,两个数据集中超过 89% 的像素都属于背景类别。两个数据集上均未取得显著结果,这表明将主动学习应用于农业数据集仍需进一步研究,尤其是当这些数据集包含严重类别不平衡和冗余图像时。本文还提供了可能解决此类问题的建议和见解。

英文摘要:

Optimizing deep learning models requires large amounts of annotated images, a process that is both time-intensive and costly. Especially for semantic segmentation models in which every pixel must be annotated. A potential strategy to mitigate annotation effort is active learning. Active learning facilitates the identification and selection of the most informative images from a large unlabelled pool. The underlying premise is that these selected images can improve the model's performance faster than random selection to reduce annotation effort. While active learning has demonstrated promising results on benchmark datasets like Cityscapes, its performance in the agricultural domain remains largely unexplored. This study addresses this research gap by conducting a comparative study of three active learning-based acquisition functions: Bayesian Active Learning by Disagreement (BALD), stochastic-based BALD (PowerBALD), and Random. The acquisition functions were tested on two agricultural datasets: Sugarbeet and Corn-Weed, both containing three semantic classes: background, crop and weed. Our results indicated that active learning, especially PowerBALD, yields a higher performance than Random sampling on both datasets. But due to the relatively large standard deviations, the differences observed were minimal; this was partly caused by high image redundancy and imbalanced classes. Specifically, more than 89\% of the pixels belonged to the background class on both datasets. The absence of significant results on both datasets indicates that further research is required for applying active learning on agricultural datasets, especially if they contain a high-class imbalance and redundant images. Recommendations and insights are provided in this paper to potentially resolve such issues.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑