Dictionary-enabled efficient training of ConvNets for image classification

被引:4
|
作者
Haider, Usman [1 ,2 ]
Hanif, Muhammad [1 ,2 ]
Rashid, Ahmar [1 ,2 ]
Hussain, Syed Fawad [1 ,3 ]
机构
[1] Ghulam Ishaq Khan Inst Engn Sci & Technol, Fac Comp Sci & Engn, Topi, Pakistan
[2] Ghulam Ishaq Khan Inst Engn Sci & Technol, Aerial Robot & Vis Lab, Topi, Pakistan
[3] Univ Birmingham, Sch Comp Sci, Birmingham, England
关键词
Sparse representation; Convolution neural networks; Deep learning; Dictionary learning; Image classification; SPARSE; REPRESENTATION; NETWORKS;
D O I
10.1016/j.imavis.2023.104718
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Convolutional networks (ConvNets) are computationally expensive but well known for their performance on image data. One way to reduce their complexity is to explore inherited data sparsity. However, since the gradi-ents involved in ConvNets require dynamic updates, applying data sparsity in the training step is not straightfor-ward. Dictionary-based learning methods can be useful since they encode the original data in a sparse form. This paper proposes a new dictionary-based training paradigm for ConvNets by exploiting redundancy in the training data while keeping the distinctive features intact. The ConvNet is then trained on the reduced, sparse dataset. The new approach significantly reduces the training time without compromising accuracy. To the best of our knowledge, this is the first implementation of ConvNet on dictionary-based sparse training data. The proposed method is validated on three publicly available datasets aeuro"MNIST, USPS, and MNIST FASHION. The experimental results show a significant reduction of 4.5 times in the overall computational burden of vanilla ConvNet for all the datasets. Whereas the accuracy is intact at 97.21% for MNIST, 96.81% for USPS, and 88.4% for FASHION datasets. These results are comparable to state-of-the-art algorithms, such as ResNet-{18,34,50}, trained on the full training dataset.& COPY; 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:10
相关论文
共 50 条
  • [21] Entropy based dictionary learning for image classification
    Abdi, Arash
    Rahmati, Mohammad
    Ebadzadeh, Mohammad M.
    PATTERN RECOGNITION, 2021, 110
  • [22] Texture classification with a dictionary of basic image features
    Crosier, Michael
    Griffin, Lewis D.
    2008 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOLS 1-12, 2008, : 2502 - 2508
  • [23] Image Classification via Hierarchical Dictionary Learning
    Sun, Peng
    Zhu, Songhao
    Ju, Xuewen
    Guo, Wenbo
    PROCEEDINGS OF THE 2019 31ST CHINESE CONTROL AND DECISION CONFERENCE (CCDC 2019), 2019, : 4630 - 4634
  • [24] WILDCAT: Weakly Supervised Learning of Deep ConvNets for Image Classification, Pointwise Localization and Segmentation
    Durand, Thibaut
    Mordan, Taylor
    Thome, Nicolas
    Cord, Matthieu
    30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 5957 - 5966
  • [25] Learning a Discriminative Dictionary with CNN for Image Classification
    Yu, Shuai
    Zhang, Tao
    Ma, Chao
    Zhou, Lei
    Yang, Jie
    He, Xiangjian
    NEURAL INFORMATION PROCESSING, ICONIP 2016, PT II, 2016, 9948 : 185 - 194
  • [26] Discriminative Structured Dictionary Learning for Image Classification
    王萍
    兰俊花
    臧玉卫
    宋占杰
    Transactions of Tianjin University, 2016, (02) : 158 - 163
  • [27] Discriminative structured dictionary learning for image classification
    Wang P.
    Lan J.
    Zang Y.
    Song Z.
    Transactions of Tianjin University, 2016, 22 (2) : 158 - 163
  • [28] STRUCTURED ANALYSIS DICTIONARY LEARNING FOR IMAGE CLASSIFICATION
    Tang, Wen
    Panahi, Ashkan
    Krim, Hamid
    Dai, Liyi
    2018 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2018, : 2181 - 2185
  • [29] Dictionary cache transformer for hyperspectral image classification
    Zhou, Heng
    Zhang, Xin
    Zhang, Chunlei
    Ma, Qiaoyu
    Jiang, Yanan
    APPLIED INTELLIGENCE, 2023, 53 (22) : 26725 - 26749
  • [30] Self-Training Enabled Efficient Classification Algorithm: An Application to Charging Pile Risk Assessment
    Wang, Wen
    Peng, Xiaofeng
    Yang, Ye
    Xiao, Chun
    Yang, Shuai
    Wang, Mingcai
    Wang, Lingfei
    Wang, Yanling
    Li, Lin
    Chang, Xiaolin
    IEEE ACCESS, 2022, 10 : 86953 - 86961