Diagonal-kernel convolutional neural networks for image classification

被引:22
|
作者
Li, Guoqing [1 ]
Shen, Xuzhao [1 ]
Li, Jiaojie [1 ]
Wang, Jiuyang [1 ]
机构
[1] Southeast Univ, Natl ASIC Res Ctr, Sch Elect Sci & Engn, Nanjing, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolutional neural networks; Diagonal kernels; Parameter efficiency; Image classification;
D O I
10.1016/j.dsp.2020.102898
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The recognition performance of convolutional neural networks has surpassed that of humans in many computer vision areas. However, there is a large number of parameter redundancy in deep neural networks, especially the weights of the convolutional kernels. In this work, we propose a simple Diagonal-kernel, in which a standard square kernel is replaced by a diagonal kernel and an anti-diagonal kernel. Diagonal-kernels with fewer parameters can have similar or larger local receptive fields than square kernels. The performance of the Diagonal-kernel is firstly evaluated on two benchmark image classification datasets, CIFAR, and ImageNet. The experimental results indicate that the Diagonal-kernel can effectively reduce parameters and computational cost while maintaining high accuracy. Furthermore, compared with Vector-kernel, Diagonal-kernel has larger local receptive fields and is more efficient. Then, we test the Diagonal-kernel for fine-grained image and imbalanced image dataset. The results show that Diagonal-kernel has larger accuracy loss for fine-grained than the coarse-grain image, but the loss is tolerable. The imbalanced data does not influence the performance of the Diagonal-kernel. The proposed Diagonal-kernel is mainly for traditional convolution but not for depthwise convolution because the number of weights for deep convolution is very small. (C) 2020 Elsevier Inc. All rights reserved.
引用
收藏
页数:8
相关论文
共 50 条
  • [1] IMAGE CLASSIFICATION USING CONVOLUTIONAL NEURAL NETWORKS AND KERNEL EXTREME LEARNING MACHINES
    Li, Zhuangzi
    Zhu, Xiaobin
    Wang, Lei
    Guo, Peiyu
    2018 25TH IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2018, : 3009 - 3013
  • [2] Convolutional Neural Networks based Pornographic Image Classification
    Zhou, KaiLong
    Zhou, Li
    Geng, Zhen
    Zhang, Jing
    Li, Xiao Guang
    2016 IEEE SECOND INTERNATIONAL CONFERENCE ON MULTIMEDIA BIG DATA (BIGMM), 2016, : 206 - 209
  • [3] Dealing with Robustness of Convolutional Neural Networks for Image Classification
    Arcaini, Paolo
    Bombarda, Andrea
    Bonfanti, Silvia
    Gargantini, Angelo
    2020 IEEE INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE TESTING (AITEST), 2020, : 7 - 14
  • [4] Design of Kernels in Convolutional Neural Networks for Image Classification
    Sun, Zhun
    Ozay, Mete
    Okatani, Takayuki
    COMPUTER VISION - ECCV 2016, PT VII, 2016, 9911 : 51 - 66
  • [5] Hierarchical Transfer Convolutional Neural Networks for Image Classification
    Dong, Xishuang
    Wu, Hsiang-Huang
    Yan, Yuzhong
    Qian, Lijun
    2019 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2019, : 2817 - 2825
  • [6] Evolving Deep Convolutional Neural Networks for Image Classification
    Sun, Yanan
    Xue, Bing
    Zhang, Mengjie
    Yen, Gary G.
    IEEE TRANSACTIONS ON EVOLUTIONARY COMPUTATION, 2020, 24 (02) : 394 - 407
  • [7] Convolutional neural networks for hyperspectral image classification
    Yu, Shiqi
    Jia, Sen
    Xu, Chunyan
    NEUROCOMPUTING, 2017, 219 : 88 - 98
  • [8] Hyperspectral Image Classification with Convolutional Neural Networks
    Slavkovikj, Viktor
    Verstockt, Steven
    De Neve, Wesley
    Van Hoecke, Sofie
    Van de Walle, Rik
    MM'15: PROCEEDINGS OF THE 2015 ACM MULTIMEDIA CONFERENCE, 2015, : 1159 - 1162
  • [9] Image Classification Using Convolutional Neural Networks
    Filippov, S. A.
    AUTOMATIC DOCUMENTATION AND MATHEMATICAL LINGUISTICS, 2024, 58 (SUPPL3) : S143 - S149
  • [10] Feature Correlation Loss in Convolutional Neural Networks for Image Classification
    Zhou, Jiahuan
    Xiao, Di
    Zhang, Mengyi
    PROCEEDINGS OF 2019 IEEE 3RD INFORMATION TECHNOLOGY, NETWORKING, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (ITNEC 2019), 2019, : 219 - 223