Classification of gastric slices based on deep learning and sparse representation

Abstract

In this paper, a classification method based on deep learning and sparse representation is proposed for gastric slice images. First of all, the convolution features of images are extracted through the convolutional neural network. Then an overcomplete learning dictionary of convolution features can be gained by K-SVD algorithm. Convolution features can obtain their sparse representation through decomposition on the overcomplete dictionary. Finally, a linear SVM classifier is used to classify the sparse representation of convolution features instead of the commonly used nonlinear classifiers, so as to achieve the purpose of classifying gastric slices. Experiments show that using linear SVM reduces the computation cost of classifier and achieves better classification effect than convolutional neural network with sigmoid output layer or SVM with RBF kernel. Therefore, the method proposed in this paper has better classification effect and lower computation cost.