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

推进绿色AI:用于水稻叶片病害识别的高效且准确的轻量级CNN

Advancing Green AI: Efficient and Accurate Lightweight CNNs for Rice Leaf Disease Identification

Universitas Syiah Kuala(Syiah Kuala大学);University of Malaya(马来亚大学)

Khairun Saddami, Yudha Nurdin, Mutia Zahramita, Muhammad Shahreeza Safiruz

AI总结:

针对水稻叶片病害识别,本文比较ShuffleNet、MobileNetV2和EfficientNet-B0三种轻量级CNN,并通过新增全连接层、dropout与早停优化,使EfficientNet-B0达到99.8%准确率。

摘要:

水稻作为全球一半以上人口的主要食物来源,发挥着至关重要的作用,其生产对全球粮食安全至关重要。然而,水稻种植经常受到各种病害的影响,这些病害会严重降低产量和品质。因此,有必要及早、准确地检测水稻病害,以防止其传播并最大限度地减少作物损失。在本研究中,我们探索了三种兼容移动端的CNN架构用于水稻叶片病害分类,即ShuffleNet、MobileNetV2和EfficientNet-B0。选择这些模型是因为它们与移动设备兼容,并且与其他CNN模型相比需要更少的计算能力和内存。为提升这三个模型的性能,我们添加了两个全连接层,并在二者之间设置一个dropout层。我们使用早停创建来防止模型过拟合。研究结果表明,EfficientNet-B0模型取得了最佳性能,准确率达到99.8%。同时,MobileNetV2和ShuffleNet的准确率分别仅为84.21%和66.51%。本研究表明,将EfficientNet-B0与所提出的层和早停相结合,可以生成一个高精度模型。

英文摘要:

Rice plays a vital role as a primary food source for over half of the world's population, and its production is critical for global food security. Nevertheless, rice cultivation is frequently affected by various diseases that can severely decrease yield and quality. Therefore, early and accurate detection of rice diseases is necessary to prevent their spread and minimize crop losses. In this research, we explore three mobile-compatible CNN architectures, namely ShuffleNet, MobileNetV2, and EfficientNet-B0, for rice leaf disease classification. These models are selected due to their compatibility with mobile devices, as they demand less computational power and memory compared to other CNN models. To enhance the performance of the three models, we added two fully connected layers separated by a dropout layer. We used early stop creation to prevent the model from being overfiting. The results of the study showed that the best performance was achieved by the EfficientNet-B0 model with an accuracy of 99.8%. Meanwhile, MobileNetV2 and ShuffleNet only achieved accuracies of 84.21% and 66.51%, respectively. This study shows that EfficientNet-B0 when combined with the proposed layer and early stop, can produce a high-accuracy model. Keywords: rice leaf detection; green AI; smart agriculture; EfficientNet;

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

↑