结合批归一化的轻量化卷积神经网络分类算法

作者：张百川; 赵佰亭^*

来源：哈尔滨商业大学学报(自然科学版), 2021, 37(03): 300-306.

DOI：10.19492/j.cnki.1672-0946.2021.03.007

• 轻量化卷积神经网络
• 模型压缩
• 深度可分离卷积
• 全局平均池化层
• 非线性激活层
• 归一化层

传统的深度卷积神经网络结构复杂,参数量多.针对现有的轻量化卷积神经网络模型结构,提出一种改进的轻量化卷积神经网络BN-MobileNet.采用模型压缩的方法对原结构进行缩减,减少冗余参数,再对深度可分离卷积的结构进行改进,在relu非线性激活层后加入归一化层来对非线性激活层所输出的数据进行归一化处理.同时,使用全局平均池化层来代替全连接层,减小模型结构的复杂度和参数量.通过在标准的分类数据集cifar10和cfar100上进行实验,实验结果表明,本文提出的改进模型能够降低运行的时间和模型的大小,提高了检测的速度,在标准分类数据集上的分类准确率各有2.68%和3.16%的提升.

• 单位
  安徽理工大学