Fusion-Net: Integration of Dimension Reduction and Deep Learning Neural Network for Image Classification
Statistics and Analytical Sciences
Building a deep network using original digital images requires learning many parameters which may reduce the accuracy rates. The images can be compressed by using dimension reduction methods and extracted reduced features can be feeding into a deep network for classification. Hence, in the training phase of the network, the number of parameters will be decreased. Principal Component Analysis is a well-known dimension reduction technique that leverage orthogonal linear transformation of the original data. In this paper, we propose a neural network-based framework, named Fusion-Net, which implements PCA on an image dataset (CIFAR-10) and then a neural network applies on the extract principal components. We also implemented logistic regression on the reduced dataset. Finally, we compare between results of using original features and reduced features. The experimental results show that Fusion-Net outperformed other methods.
Artificial Intelligence and Robotics Commons, Data Science Commons, Statistics and Probability Commons