卷积神经网络与ELM的图像分类算法研究

作者：马致颖; 赵佰亭; 冯铸; 张华

来源：哈尔滨商业大学学报(自然科学版), 2021, 37(04): 415-421.

DOI：10.19492/j.cnki.1672-0946.2021.04.005

• 图像分类
• 混合模型
• 深度学习
• 卷积神经网络
• 极限学习机
• 训练时间

卷积神经网络的全连接层作为一个经典的分类器,是根据传统的梯度下降法来实现训练的,泛化能力有限.针对这一问题,提出了一种将卷积神经网络和极限学习机相结合的混合模型应用于图像分类领域.卷积神经网络用于从输入图像中提取特征,特征映射最终会被编码成一维向量送入极限学习机中进行分类.给出了混合模型的详细设计,包括参数设计、结构分析以及迭代过程中反向传播算法的推导.实验结果表明,混合模型分类精度和快速性优于传统模型.在MNIST数据集上的分类精度达到了99.32%,GTSRB德国交通标志数据集上的分类精度达到99.35%.批尺寸相同时,混合模型的训练时间仅为卷积神经网络模型的9.9%～10.3%,同时测试速度是卷积神经网络模型的1.50～1.56倍,大幅缩短了时间.

• 单位
  安徽理工大学