Robust deep convolutional neural network against image distortions
Open Access
- 11 October 2021
- journal article
- research article
- Published by Now Publishers in APSIPA Transactions on Signal and Information Processing
- Vol. 10 (1)
- https://doi.org/10.1017/atsip.2021.14
Abstract
Many approaches have been proposed in the literature to enhance the robustness of Convolutional Neural Network (CNN)-based architectures against image distortions. Attempts to combat various types of distortions can be made by combining multiple expert networks, each trained by a certain type of distorted images, which however lead to a large model with high complexity. In this paper, we propose a CNN-based architecture with a pre-processing unit in which only undistorted data are used for training. The pre-processing unit employs discrete cosine transform (DCT) and discrete wavelets transform (DWT) to remove high-frequency components while capturing prominent high-frequency features in the undistorted data by means of random selection. We further utilize the singular value decomposition (SVD) to extract features before feeding the preprocessed data into the CNN for training. During testing, distorted images directly enter the CNN for classification without having to go through the hybrid module. Five different types of distortions are produced in the SVHN dataset and the CIFAR-10/100 datasets. Experimental results show that the proposed DCT-DWT-SVD module built upon the CNN architecture provides a classifier robust to input image distortions, outperforming the state-of-the-art approaches in terms of accuracy under different types of distortions.Keywords
This publication has 5 references indexed in Scilit:
- Selective Deep Convolutional Neural Network for Low Cost Distorted Image ClassificationIEEE Access, 2019
- Quality Robust Mixtures of Deep Neural NetworksIEEE Transactions on Image Processing, 2018
- ImageNet classification with deep convolutional neural networksCommunications of the ACM, 2017
- The JPEG still picture compression standardIEEE Transactions on Consumer Electronics, 1992
- Discrete Cosine TransformInternational Conference on Acoustics, Speech, and Signal Processing (ICASSP), 1974