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

作     者：马致颖 赵佰亭 冯铸 张华 MA Zhi-ying;ZHAO Bai-ting;FENG Zhu;ZHANG Hua

作者机构：安徽理工大学机械工程学院安徽淮南232001 安徽理工大学电气与信息工程学院安徽淮南232001 

出 版 物：《哈尔滨商业大学学报（自然科学版）》 (Journal of Harbin University of Commerce：Natural Sciences Edition)

年 卷 期：2021年第37卷第4期

页      码：415-421页

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

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

学科分类：08[工学] 080203[080203] 0802[工学-机械学] 

D　O　I：10.19492/j.cnki.1672-0946.2021.04.005

馆 藏 号：203104699...