arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.03390cs.CVcs.LGeess.IV

面向多类别杂草检测的半监督学习框架性能评估

Performance Evaluation of Semi-supervised Learning Frameworks for Multi-Class Weed Detection

Michigan State University(密歇根州立大学);University of Virginia(弗吉尼亚大学);Nanyang Technological University(南洋理工大学)

Jiajia Li, Dong Chen, Xunyuan Yin, Zhaojian Li

AI总结:

该研究针对杂草检测监督学习标注成本高的问题,评估了带改进伪标签模块与集成学生网络的半监督师生框架,结合FCOS、Faster-RCNN实现了低标注量下的高检测精度并开源代码。

摘要:

有效的杂草防控在优化作物产量、提升农产品品质方面发挥着关键作用。然而,依赖除草剂施用不仅会对环境构成严重威胁,还会促使抗药性杂草的出现。幸运的是,由机器学习(ML)和深度学习(DL)赋能的精准杂草管理领域的最新进展提供了一种可持续的替代方案。尽管已取得长足进展,但现有算法主要基于监督学习方法开发,这类方法通常需要带有人工标注的大规模数据集,既耗时又费力。因此,标签高效学习方法,尤其是半监督学习,在更广泛的计算机视觉领域受到了越来越多的关注,并展现出了可观的性能。这类方法旨在利用少量标注数据样本与大量无标注样本,训练出性能可与基于大量标注数据训练的监督学习模型相媲美的高性能模型。在本研究中,我们评估了一种半监督学习框架在多类别杂草检测任务中的有效性,采用了两种知名的目标检测框架,即FCOS和Faster-RCNN。具体而言,我们评估了一种带有改进伪标签生成模块的通用师生框架,用于为无标注数据生成可靠的伪标签。为增强泛化能力,我们采用了集成学生网络来辅助训练过程。实验结果表明,在仅使用10%标注数据的情况下,所提出的方法在CottenWeedDet3和CottonWeedDet12数据集上分别能达到监督方法约76%和96%的检测精度。我们提供了源代码访问渠道,为杂草检测及更多领域的半监督学习研究贡献了宝贵资源。

英文摘要:

Effective weed control plays a crucial role in optimizing crop yield and enhancing agricultural product quality. However, the reliance on herbicide application not only poses a critical threat to the environment but also promotes the emergence of resistant weeds. Fortunately, recent advances in precision weed management enabled by ML and DL provide a sustainable alternative. Despite great progress, existing algorithms are mainly developed based on supervised learning approaches, which typically demand large-scale datasets with manual-labeled annotations, which is time-consuming and labor-intensive. As such, label-efficient learning methods, especially semi-supervised learning, have gained increased attention in the broader domain of computer vision and have demonstrated promising performance. These methods aim to utilize a small number of labeled data samples along with a great number of unlabeled samples to develop high-performing models comparable to the supervised learning counterpart trained on a large amount of labeled data samples. In this study, we assess the effectiveness of a semi-supervised learning framework for multi-class weed detection, employing two well-known object detection frameworks, namely FCOS and Faster-RCNN. Specifically, we evaluate a generalized student-teacher framework with an improved pseudo-label generation module to produce reliable pseudo-labels for the unlabeled data. To enhance generalization, an ensemble student network is employed to facilitate the training process. Experimental results show that the proposed approach is able to achieve approximately 76\% and 96\% detection accuracy as the supervised methods with only 10\% of labeled data in CottenWeedDet3 and CottonWeedDet12, respectively. We offer access to the source code, contributing a valuable resource for ongoing semi-supervised learning research in weed detection and beyond.

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

↑