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 条
  • [31] Deep Convolutional Neural Networks for Hyperspectral Image Classification
    Hu, Wei
    Huang, Yangyu
    Wei, Li
    Zhang, Fan
    Li, Hengchao
    JOURNAL OF SENSORS, 2015, 2015
  • [32] Improving the Performance of Convolutional Neural Networks for Image Classification
    Giveki, Davar
    OPTICAL MEMORY AND NEURAL NETWORKS, 2021, 30 (01) : 51 - 66
  • [33] Location Property of Convolutional Neural Networks for Image Classification
    Liang, Cong
    Zhang, Haixia
    Yuan, Dongfeng
    Zhang, Minggao
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2021, 32 (09) : 3831 - 3845
  • [34] 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
  • [35] 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
  • [36] An Image Classification Scheme for Improved Convolutional Neural Networks
    He Weixin
    Cong Linhu
    Deng Jianqiu
    Zhou Haichao
    2019 4TH INTERNATIONAL CONFERENCE ON MECHANICAL, CONTROL AND COMPUTER ENGINEERING (ICMCCE 2019), 2019, : 614 - 617
  • [37] GROUP CONVOLUTIONAL NEURAL NETWORKS FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    Li, Xian
    Ding, Mingli
    Pizurica, Aleksandra
    2019 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2019, : 639 - 643
  • [38] Quaternion convolutional neural networks for hyperspectral image classification
    Zhou, Heng
    Zhang, Xin
    Zhang, Chunlei
    Ma, Qiaoyu
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 123
  • [39] Fruit Image Classification Using Convolutional Neural Networks
    Ashraf, Shawon
    Kadery, Ivan
    Chowdhury, Md Abdul Ahad
    Mahbub, Tahsin Zahin
    Rahman, Rashedur M.
    INTERNATIONAL JOURNAL OF SOFTWARE INNOVATION, 2019, 7 (04) : 51 - 70
  • [40] Fiber Image Classification Using Convolutional Neural Networks
    Wang, Xinxin
    Chen, Zhao
    Liu, Guohua
    Wan, Yan
    2017 4TH INTERNATIONAL CONFERENCE ON SYSTEMS AND INFORMATICS (ICSAI), 2017, : 1214 - 1218