Deep Convolutional Representations and Kernel Extreme Learning Machines for Image Classification

Xiaobin Zhu,Zhuangzi Li,Xiao-Yu Zhang,Peng Li,Ziyu Xue,Lei Wang
DOI: https://doi.org/10.1007/s11042-018-6781-z
IF: 2.577
2019-01-01
Multimedia Tools and Applications
Abstract:Convolutional Neural Networks (CNNs) have been established as a powerful class of models for image classification and related tasks. However, the fully-connected layers in CNN are not robust enough to serve as a classifier to discriminate deep convolutional features, due to the local minima problem of back-propagation. Kernel Extreme Learning Machines (KELMs), known as an outstanding classifier, can not only converge extremely fast but also ensure an outstanding generalization performance. In this paper, we propose a novel image classification framework, in which CNN and KELM are well integrated. In our work, Densely connected network (DenseNet) is employed as the feature extractor, while a radial basis function kernel ELM instead of linear fully connected layer is adopted as a classifier to discriminate categories of extracted features to promote the image classification performance. Experiments conducted on four publicly available datasets demonstrate the promising performance of the proposed framework against the state-of-the-art methods.
What problem does this paper attempt to address?