基于多样化结构的轻量型卷积神经网络设计

作者：魏书伟; 曾上游; 潘兵; 王新娇

来源：现代电子技术, 2020, 43(12): 50-54.

DOI：10.16652/j.issn.1004-373x.2020.12.013

• 卷积神经网络
• 多样化结构
• 残差学习
• 图像预处理
• 图像分类
• 测试分析

传统的卷积神经网络的网络结构单一或模块单一,网络提取特征时都是以同一种模式持续进行,导致分类精度不够理想,且模型参数和计算量较大。针对这一问题,提出一种使用多样化模块的轻量型卷积神经网络结构Diverse-net,使用Reduce Module和Slice Module两种卷积模块,通过交替叠加使用两种不同的卷积模块来代替传统网络的单卷积核结构,增加网络的深度和宽度,并在不同模块之间加入残差学习。使用文中搭建的卷积神经网络Diverse-net,与其他网络模型在GTSRB和101food数据集上进行测试来对比网络性能与模型大小,Diverse-net网络模型大小减少至20.8 MB,在数据集GTSRB上识别率可达到98.72%;在数据集101food上识别率可达到68.09%。实验结果表明,所设计的卷积神经网络Diversenet在图像分类方面性能更优,且网络的模型较小。

• 单位
  广西师范大学; 电子工程学院