基于动态剪枝神经网络的杂草检测算法研究
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

陕西省重点研发计划项目(2021GY-022)和西安市科技计划项目(2019216514GXRC001CG002-GXYD1.7)


Weed Detection Algorithm Based on Dynamic Pruning Neural Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对卷积神经网络模型巨大的参数量和计算量导致其实际应用时难度较大的问题,提出了一种基于注意力机制与动态稀疏约束的模型压缩方法。该算法首先借助SENet(Squeeze and excitation networks,SENet)模块(可称为SE模块)评估出网络中各个通道的重要性,并施加稀疏正则化;然后提出一种网络稀疏度的自适应惩罚权重设计方法,根据模型学习效果,动态调整权重,将其添加到最终的训练目标上,实现模型动态压缩。最后,通过实验验证所提出的模型压缩方法,在经典的多分类数据集CIFAR-10上进行实验,证明了本文所提出的基于注意力机制与动态稀疏约束的模型压缩方法可降低网络的冗余度,使网络模型参数量减少43.97%,计算量减少82.94%,而分类准确率只比原始VGG16模型下降0.04个百分点。随后又将提出的模型压缩方法应用到杂草检测任务中,在甜菜与杂草数据集上进行实验,实验结果表明,剪枝模型相较于未剪枝模型的模型参数量减少41.26%,计算量减少45.77%,而平均检测精度均值只减少0.91个百分点,证明了该方法在杂草检测方面效果较好。

    Abstract:

    To address the problem that the convolutional neural network models are difficult to be applied in practice due to their vast number of parameters and computation, a model compression method based on attention mechanism and dynamic sparse constraint was proposed. Firstly, the importance of each channel in the network was evaluated with the help of the squeeze and excitation networks (SENet) module, and sparse regularization was applied; then an adaptive penalty weight design method for network sparsity was proposed. According to the learning effect of the model, the weight was dynamically adjusted and added to the final training target to realize the dynamic compression of the model. Finally, the proposed model compression method was verified by experiments on the classic multi-classification dataset CIFAR-10. It was proved that the proposed model compression method based on attention mechanism and dynamic sparse constraint can reduce the network redundancy, resulting in a 43.97% reduction in the amount of network model parameters and an 82.94% reduction in the amount of computation, while the classification accuracy was only 0.04 percentage points lower than that of the original VGG16 model. Then the proposed model compression method was applied to the weed detection task, and the experiment was carried out on the sugar beet and weed datasets. The experimental results showed that compared with the unpruned model, the pruned model reduced the model parameters by 41.26%, the calculation amount by 45.77%, and the average detection accuracy by only 0.91 percentage points, which proved that this method could also have a good effect on the weed detection task.

    参考文献
    相似文献
    引证文献
引用本文

亢洁,刘港,王勍,夏宇,郭国法,刘文波.基于动态剪枝神经网络的杂草检测算法研究[J].农业机械学报,2023,54(4):268-275. KANG Jie, LIU Gang, WANG Qing, XIA Yu, GUO Guofa, LIU Wenbo. Weed Detection Algorithm Based on Dynamic Pruning Neural Network[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(4):268-275.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-07-01
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2022-12-22
  • 出版日期: