A Convolutional Autoencoder Topology for Classification in High-Dimensional Noisy Image Datasets

Sensors (Basel). 2021 Nov 20;21(22):7731. doi: 10.3390/s21227731.

Abstract

Deep convolutional neural networks have shown remarkable performance in the image classification domain. However, Deep Learning models are vulnerable to noise and redundant information encapsulated into the high-dimensional raw input images, leading to unstable and unreliable predictions. Autoencoders constitute an unsupervised dimensionality reduction technique, proven to filter out noise and redundant information and create robust and stable feature representations. In this work, in order to resolve the problem of DL models' vulnerability, we propose a convolutional autoencoder topological model for compressing and filtering out noise and redundant information from initial high dimensionality input images and then feeding this compressed output into convolutional neural networks. Our results reveal the efficiency of the proposed approach, leading to a significant performance improvement compared to Deep Learning models trained with the initial raw images.

Keywords: computer vision; convolutional autoencoders; convolutional neural networks; deep learning; dimensionality reduction; image classification.

MeSH terms

  • Neural Networks, Computer*