Convolutional Neural Network Features Comparison Between Back-Propagation and Extreme Learning Machine

Recently deep learning based architectures have been widely deployed in many problems of artificial intelligence. Among deep learning models, Convolutional Neural Networks (CNN) have been reported in numerous successful applications such as object recognition, and natural language processing. The convolutional neural networks are trained by back-propagating the classification error using the Back-Propagation (BP) algorithm, which requires a large amount of data and slows the training process. To overcome these difficulties, a new fast and accurate approach based on Extreme Learning Machine (ELM) to train any convolutional neural network has been proposed. The developed framework (ELM-CNN) is based on the concept of auto-encoding to learn the convolutional filters with biases, by reconstructing the normalized input and the intercept term. In this paper, systematic comparison with traditional back-propagation based training method (BP-CNN) has been made with respect to two aspects qualitative and quantitative. The experimental results on the popular MNIST dataset show that the ELM-CNN algorithm achieves competitive results in terms of generalization performance and up to 16 times faster than the back-propagation based training of CNN.

[1]  Hod Lipson,et al.  Understanding Neural Networks Through Deep Visualization , 2015, ArXiv.

[2]  Hongming Zhou,et al.  Extreme Learning Machine for Regression and Multiclass Classification , 2012, IEEE Transactions on Systems, Man, and Cybernetics, Part B (Cybernetics).

[3]  Qing Fei,et al.  Convolutional Neural Network Based on Extreme Learning Machine for Maritime Ships Recognition in Infrared Images , 2018, Sensors.

[4]  Se-Young Oh,et al.  Fast training of convolutional neural network classifiers through extreme learning machines , 2016, 2016 International Joint Conference on Neural Networks (IJCNN).

[5]  Michael S. Bernstein,et al.  ImageNet Large Scale Visual Recognition Challenge , 2014, International Journal of Computer Vision.

[6]  Minho Lee,et al.  Fast learning method for convolutional neural networks using extreme learning machine and its application to lane detection , 2017, Neural Networks.

[7]  Hongming Zhou,et al.  Extreme Learning Machines [Trends & Controversies] , 2013 .

[8]  Joan Bruna,et al.  Intriguing properties of neural networks , 2013, ICLR.

[9]  Chee Kheong Siew,et al.  Universal Approximation using Incremental Constructive Feedforward Networks with Random Hidden Nodes , 2006, IEEE Transactions on Neural Networks.

[10]  Andrea Vedaldi,et al.  MatConvNet: Convolutional Neural Networks for MATLAB , 2014, ACM Multimedia.

[11]  Andrew Zisserman,et al.  Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Maps , 2013, ICLR.

[12]  Geoffrey E. Hinton,et al.  Visualizing Data using t-SNE , 2008 .

[13]  Pascal Vincent,et al.  Visualizing Higher-Layer Features of a Deep Network , 2009 .

[14]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[15]  Rob Fergus,et al.  Visualizing and Understanding Convolutional Networks , 2013, ECCV.