首页 | 本学科首页   官方微博 | 高级检索  
     

基于轻量级特征融合卷积网络的图像分类算法
引用本文:陈鑫华,钱雪忠,宋威.基于轻量级特征融合卷积网络的图像分类算法[J].计算机工程,2021,47(11):268-275.
作者姓名:陈鑫华  钱雪忠  宋威
作者单位:江南大学人工智能与计算机学院,江苏无锡214122;江南大学物联网技术应用教育部工程研究中心,江苏无锡214122;江南大学 江苏省模式识别与计算智能工程实验室,江苏 无锡 214122
基金项目:国家自然科学基金(61673193);中国博士后科学基金(2017M621625);江苏省自然科学基金(BK20181341)。
摘    要:传统卷积神经网络存在卷积核单一、网络结构复杂和参数冗余的问题。提出一种轻量级特征融合卷积神经网络MS-FNet。在融合模块中采用多路结构以增加卷积神经网络的宽度,通过不同尺寸的卷积核对输入特征图进行处理,提高网络在同一层中提取不同特征的能力,并在每次卷积后采用批归一化、ReLU等方法去除冗余特征。此外,使用卷积层代替传统的全连接层,从而加快模型的训练速度,缓解因参数过多造成的过拟合现象。实验结果表明,MS-FNet可在降低错误率的同时,有效减少网络参数量。

关 键 词:深度学习  卷积神经网络  特征提取  特征融合  图像分类
收稿时间:2020-10-23
修稿时间:2020-12-02

Image Classification Algorithm Based on Lightweight Feature Fusion Convolutional Network
CHEN Xinhua,QIAN Xuezhong,SONG Wei.Image Classification Algorithm Based on Lightweight Feature Fusion Convolutional Network[J].Computer Engineering,2021,47(11):268-275.
Authors:CHEN Xinhua  QIAN Xuezhong  SONG Wei
Affiliation:1. School of Artificial Intelligence and Computer Science, Jiangnan University, Wuxi, Jiangsu 214122, China;2. Engineering Research Center of Internet of Things Technology Applications of Ministry of Education, Jiangnan University, Wuxi, Jiangsu 214122, China;3. Jiangsu Provincial Engineering Laboratory of Pattern Recognition and Computer Intelligence, Jiangnan University, Wuxi, Jiangsu 214122, China
Abstract:The traditional Convolutional Neural Networks (CNN) suffer from single convolutional kernels,complex network structure and redundant parameters.To address the problem,a lightweight CNN named MS-FNet is designed for feature fusion.The fusion module employs a multi-branch structure to increase the width of the CNN,and different sizes of convolutional kernels to process the input feature map,which improves the ability of the network to extract different features in the same layer.And the redundant features are removed after each convolution by using BN,ReLU,etc.Convolutional layers are used to replace the traditional fully connected layer,which not only accelerates the training speed of model but also alleviates overfitting problems caused by too many parameters.The experimental results show that MS-FNet greatly reduces the number of network parameters and the error rate.
Keywords:Deep Learning(DL)  Convolutional Neural Network(CNN)  feature extraction  feature fusion  image classification  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机工程》浏览原始摘要信息
点击此处可从《计算机工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号