基于轻量级特征融合卷积网络的图像分类算法

doi:10.19678/j.issn.1000-3428.0059815

计算机工程 ›› 2021, Vol. 47 ›› Issue (11): 268-275. doi: 10.19678/j.issn.1000-3428.0059815

基于轻量级特征融合卷积网络的图像分类算法

陈鑫华¹, 钱雪忠², 宋威³

1. 江南大学人工智能与计算机学院, 江苏无锡 214122;
2. 江南大学物联网技术应用教育部工程研究中心, 江苏无锡 214122;
3. 江南大学江苏省模式识别与计算智能工程实验室, 江苏无锡 214122

收稿日期:2020-10-23 修回日期:2020-12-02 发布日期:2020-12-09
作者简介:陈鑫华(1996-),男,硕士研究生,主研方向为深度学习、图像识别;钱雪忠,副教授、硕士;宋威,教授、博士。
基金资助:
国家自然科学基金（61673193）；中国博士后科学基金（2017M621625）；江苏省自然科学基金（BK20181341）。

Image Classification Algorithm Based on Lightweight Feature Fusion Convolutional Network

CHEN Xinhua¹, QIAN Xuezhong², SONG Wei³

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

Received:2020-10-23 Revised:2020-12-02 Published:2020-12-09

摘要/Abstract

摘要： 传统卷积神经网络存在卷积核单一、网络结构复杂和参数冗余的问题。提出一种轻量级特征融合卷积神经网络MS-FNet。在融合模块中采用多路结构以增加卷积神经网络的宽度，通过不同尺寸的卷积核对输入特征图进行处理，提高网络在同一层中提取不同特征的能力，并在每次卷积后采用批归一化、ReLU等方法去除冗余特征。此外，使用卷积层代替传统的全连接层，从而加快模型的训练速度，缓解因参数过多造成的过拟合现象。实验结果表明，MS-FNet可在降低错误率的同时，有效减少网络参数量。

关键词: 深度学习, 卷积神经网络, 特征提取, 特征融合, 图像分类

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.

Key words: Deep Learning(DL), Convolutional Neural Network(CNN), feature extraction, feature fusion, image classification

中图分类号:

TP391.4

陈鑫华, 钱雪忠, 宋威. 基于轻量级特征融合卷积网络的图像分类算法[J]. 计算机工程, 2021, 47(11): 268-275.

CHEN Xinhua, QIAN Xuezhong, SONG Wei. Image Classification Algorithm Based on Lightweight Feature Fusion Convolutional Network[J]. Computer Engineering, 2021, 47(11): 268-275.

https://www.ecice06.com/CN/Y2021/V47/I11/268

图/表 15

20211113145515

20211113145517

20211113145520

20211113145523

20211113145527

20211113145530

20211113145533

20211113145536

20211113145539

20211113145543

20211113145546

20211113145548

20211113145553

20211113145557

20211113145600

参考文献

[1] 张顺,龚怡宏,王进军.深度卷积神经网络的发展及其在计算机视觉领域的应用[J].计算机学报,2019,42(3):453-482. ZHANG S,GONG Y H,WANG J J.The development of deep convolution neural network and its application of computer vision[J].Chinese Journal of Computers,2019,42(3):453-482.(in Chinese)
[2] HU Y,SUN S,LI J,et al.A novel channel pruning method for deep neural network compression[EB/OL].[2020-09-15].http://export.arxiv.org/pdf/1805.11394.
[3] 靳丽蕾,杨文柱,王思乐,等.一种用于卷积神经网络压缩的混合剪枝方法[J].小型微型计算机系统,2018,39(12):2596-2601. JIN L L,YANG W Z,WANG S L,et al.Mixed pruning method for convolutional neural network compression[J].Journal of Chinese Mini-Micro Computer Systems,2018,39(12):2596-2601.(in Chinese)
[4] IANDOLA F,HAN S,MOSKEWICZ M W,et al.SqueezeNet:AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size[EB/OL].[2020-09-20].https://arxiv.org/abs/1602.07360v1.
[5] ZHANG X,ZHOU X,LIN M,et al.ShuffleNet:an extremely efficient convolutional neural network for mobile devices[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C.,USA:IEEE Press,2018:6848-6856.
[6] MEHTA S,RASTEGARI M,CASPI A,et al.ESPNet:efficient spatial pyramid of dilated convolutions for semantic segmentation[C]//Proceedings of European Conference on Computer Vision.Berlin,Germany:Springer,2018:552-568.
[7] SZEGEDY C,LIU W,JIA Y,et al.Going deeper with convolutions[C]//Proceedings of 2015 IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C.,USA:IEEE Press,2015:1-9.
[8] HE K,ZHANG X,REN S,et al.Deep residual learning for image recognition[C]//Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C.,USA:IEEE Press,2016:770-778.
[9] HUANG G,LIU Z,MAATEN L V D,et al.Densely connected convolutional networks[C]//Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C.,USA:IEEE Press,2017:2261-2269.
[10] ZHANG Z,WANG X,JUNG C.DCSR:dilated convolutions for single image super-resolution[J].IEEE Transactions on Image Processing,2019,28(4):1625-1635.
[11] SZEGEDY C,VANHOUCKE V,IOFFE S,et al.Rethinking the inception architecture for computer vision[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C.,USA:IEEE Press,2016:2818-2826.
[12] ARORA S,BHASKARA A,GE R,et al.Provable bounds for learning some deep representations[EB/OL].[2020-09-21].https://arxiv.org/abs/1310.6343v1.
[13] SIMONYAN K,ZISSERMAN A.Very deep convolutional networks for large-scale image recognition[EB/OL].[2020-09-18].https://arxiv.org/abs/1409.1556.
[14] LIN M,CHEN Q,YAN S,et al.Network in network[EB/OL].[2020-09-22].https://arxiv.org/abs/1312.4400.
[15] 高云龙,吴川,朱明.基于改进卷积神经网络的短文本分类模型[J].吉林大学学报(理学版),2020,58(4):923-930. GAO Y L,WU C,ZHU M.Short text classification model based on improved convolutional neural network[J].Journal of Jilin University(Science Edition),2020,58(4):923-930.(in Chinese)
[16] ROY A G,NAVAB N,WACHINGER C.Recalibrating fully convolutional networks with spatial and channel "squeeze and excitation" blocks[J].IEEE Transactions on Medical Imaging,2019,38(2):540-549.
[17] LI Y,WANG N,SHI J,et al.Adaptive batch normalization for practical domain adaptation[J].Pattern Recognition,2018,80(4):109-117.
[18] ABADI M,BARHAM P,CHEN J,et al.TensorFlow:a system for large-scale machine learning[C]//Proceedings of Symposium on Operating Systems Design and Implementation.New York,USA:ACM Press,2016:265-283.
[19] LEE C,XIE S,GALLAGHER P W,et al.Deeply-supervised nets[C]//Proceedings of International Conference on Artificial Intelligence and Statistics.New York,USA:ACM Press,2015:562-570.
[20] 杨萌林,张文生.分类激活图增强的图像分类算法[J].计算机科学与探索,2020,14(1):149-158. YANG M L,ZHANG W S.Image classification algorithm based on classification activation map enhancement[J].Journal of Frontiers of Computer Science and Technology,2020,14(1):149-158.(in Chinese)
[21] ZAGORUYKO S,KOMODAKIS N.Wide residual networks[EB/OL].[2020-09-20].https://arxiv.org/abs/1605.07146v4.
[22] LARSSON G,MAIRE M,SHAKHNAROVICH G,et al.FractalNet:ultra-deep neural networks without residuals[EB/OL].[2020-09-21].https://arxiv.org/abs/1605.07648v1.
[23] WU Y.Deep convolutional neural network based on densely connected squeeze-and-excitation blocks[J].AIP Advances,2019,9(6):1-10.
[24] HOU S,LIU X,WANG Z.Dualnet:learn complementary features for image recognition[C]//Proceedings of IEEE International Conference on Computer Vision.Washington D.C.,USA:IEEE Press,2017:502-510.
[25] VELICKOVIC P,WANG D,LANEY N D,et al.X-CNN:cross-modal convolutional neural networks for sparse datasets[C]//Proceedings of IEEE Symposium Series on Computational Intelligence.Washington D.C.,USA:IEEE Press,2016:1-8.
[26] KILINC O,UYSAL I.GAR:an efficient and scalable graph-based activity regularization for semi-supervised learning[J].Neurocomputing,2018,296(6):46-54.
[27] 付晓,沈远彤,李宏伟,等.基于半监督编码生成对抗网络的图像分类模型[J].自动化学报,2020,46(3):531-539. FU X,SHEN Y T,LI H W,et al.A semi-supervised encoder generative adversarial networks model for image classification[J].Acta Automatica Sinica,2020,46(3):531-539.(in Chinese)

选择文件类型/文献管理软件名称

选择包含的内容

基于轻量级特征融合卷积网络的图像分类算法

Image Classification Algorithm Based on Lightweight Feature Fusion Convolutional Network

RichHTML

PDF

可视化

被引次数

摘要/Abstract

引用本文

使用本文

图/表 15

参考文献

相关文章 15

编辑推荐

Metrics

本文评价

[1]	谭巨全, 王然. 特征融合下田径录像3D人体动作DTW捕捉算法[J]. 计算机工程, 2024, 50(7): 71-78.
[2]	耿丽丽, 牛保宁. 基于通道相似度熵的卷积神经网络裁剪[J]. 计算机工程, 2024, 50(7): 133-143.
[3]	张洋, 刘畅, 李少青. 基于可控制性度量的图神经网络门级硬件木马检测方法[J]. 计算机工程, 2024, 50(7): 164-173.
[4]	牛瑞婷, 严天峰, 高锐, 王映植. 低信噪比下基于深度学习TCNN-MobileNet的调制识别[J]. 计算机工程, 2024, 50(7): 204-215.
[5]	张溢文, 蔡满春, 陈咏豪, 朱懿, 姚利峰. 融合空间特征的多尺度深度伪造检测方法[J]. 计算机工程, 2024, 50(7): 240-250.
[6]	逯焕宇, 张永宏, 马光义, 谢东林, 田伟. 基于半监督对抗学习的遥感图像水体提取[J]. 计算机工程, 2024, 50(7): 251-263.
[7]	肖慈, 徐杨, 张永丹, 冯明文, 黄易仟. 结合注意力和低光增强的夜间语义分割[J]. 计算机工程, 2024, 50(7): 271-281.
[8]	张诗婧, 莫绪涛, 赵行, 董杨林. 基于球面折反射成像和YOLOv7的内螺纹缺陷检测[J]. 计算机工程, 2024, 50(7): 282-292.
[9]	王晋涛, 秦昂, 张元, 陈一飞, 王廷凤, 谢承霖, 邹刚. 基于注意力增强与特征融合的中文医学实体识别[J]. 计算机工程, 2024, 50(7): 324-332.
[10]	徐明亮, 李芳媛, 马浩然, 何飞. 大规模神经记录的峰电位聚类算法(特邀)[J]. 计算机工程, 2024, 50(6): 1-34.
[11]	魏琢艺, 罗迈, 李文兵, 曾远松, 余伟江, 杨跃东. 基于多源域适应的单细胞智能分类[J]. 计算机工程, 2024, 50(6): 48-55.
[12]	李亚康, 陈刚. 小角中子散射物理模型自动化筛选[J]. 计算机工程, 2024, 50(6): 56-64.
[13]	李子杰, 周菊香, 韩晓瑜, 甘健侯, 鹿泽光, 王俊. 序列特征与学习过程融合的知识追踪模型[J]. 计算机工程, 2024, 50(6): 77-85.
[14]	胡庆. 多尺度融合与双输出U-Net网络的行人重识别[J]. 计算机工程, 2024, 50(6): 102-109.
[15]	李永飞, 李铭洋, 常鑫, 曹可欣. 基于可解释性深度学习的物联网水质监测数据异常检测[J]. 计算机工程, 2024, 50(6): 179-187.

模态框（Modal）标题

选择文件类型/文献管理软件名称

选择包含的内容

基于轻量级特征融合卷积网络的图像分类算法

Image Classification Algorithm Based on Lightweight Feature Fusion Convolutional Network

RichHTML

PDF

可视化

被引次数

摘要/Abstract

引用本文

使用本文

图/表 15

参考文献

相关文章 15

编辑推荐

Metrics

本文评价