Dictionary-enabled efficient training of ConvNets for image classification

• Published in: Image and vision computing 2023-07, Vol.135, p.104718, Article 104718
• Main Authors: Haider, Usman, Hanif, Muhammad, Rashid, Ahmar, Hussain, Syed Fawad
• Format: Article
• Language: English
• Subjects: Convolution neural networks; Deep learning; Dictionary learning; Image classification; Sparse representation
• ISSN: 0262-8856
• DOI: 10.1016/j.imavis.2023.104718

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 gradients involved in ConvNets require dynamic updates, applying data sparsity in the training step is not straightforward. 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 –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. •ConvNets are powerful for images but computationally expensive.•Paper proposes dictionary-based training for ConvNets exploiting sparsity.•The proposed method preserves orthogonal features and reduces training time.•Results: 4.5x reduction in computational burden.•High accuracy: 97.21% MNIST, 96.81% USPS, 88.4% FASHION.