| Literature DB >> 30298088 |
Abstract
Medical image classification is a key technique of Computer-Aided Diagnosis (CAD) systems. Traditional methods rely mainly on the shape, color, and/or texture features as well as their combinations, most of which are problem-specific and have shown to be complementary in medical images, which leads to a system that lacks the ability to make representations of high-level problem domain concepts and that has poor model generalization ability. Recent deep learning methods provide an effective way to construct an end-to-end model that can compute final classification labels with the raw pixels of medical images. However, due to the high resolution of the medical images and the small dataset size, deep learning models suffer from high computational costs and limitations in the model layers and channels. To solve these problems, in this paper, we propose a deep learning model that integrates Coding Network with Multilayer Perceptron (CNMP), which combines high-level features that are extracted from a deep convolutional neural network and some selected traditional features. The construction of the proposed model includes the following steps. First, we train a deep convolutional neural network as a coding network in a supervised manner, and the result is that it can code the raw pixels of medical images into feature vectors that represent high-level concepts for classification. Second, we extract a set of selected traditional features based on background knowledge of medical images. Finally, we design an efficient model that is based on neural networks to fuse the different feature groups obtained in the first and second step. We evaluate the proposed approach on two benchmark medical image datasets: HIS2828 and ISIC2017. We achieve an overall classification accuracy of 90.1% and 90.2%, respectively, which are higher than the current successful methods.Entities:
Mesh:
Year: 2018 PMID: 30298088 PMCID: PMC6157177 DOI: 10.1155/2018/2061516
Source DB: PubMed Journal: Comput Intell Neurosci
Figure 1Framework of the approach.
The configuration of the coding network.
| Type | Patch size/stride | Output size |
|---|---|---|
| Convolution | 11 × 11/1 | 130 × 130 × 32 |
| Convolution | 11 × 11/1 | 120 × 120 × 32 |
| Max pool | 5 × 5/2 | 58 × 58 × 32 |
| Convolution | 9 × 9/1 | 50 × 50 × 64 |
| Max pool | 5 × 5/2 | 23 × 23 × 64 |
| Convolution | 8 × 8/1 | 16 × 16 × 128 |
| Convolution | 9 × 9/1 | 8 × 8 × 256 |
| Convolution | 8 × 8/1 | 1 × 1 × 256 |
| Rasterize | 1 × 1 × 4 | |
| Softmax layer | 1 × 1 × 4 |
Summary of the symbols.
| Symbols | The detail description |
|---|---|
|
| The gray-level co-occurrence matrix |
|
| The size of |
|
| The |
|
|
|
|
|
|
|
| A matrix for representation of the image |
|
| The number of pixels in |
|
| The |
|
| The mean of the |
|
| The variance of the |
|
| The skewness of the |
The composition of the HIS2828 dataset.
| Image category | Number of images | Label |
|---|---|---|
| Nervous tissue | 1026 | 1 |
| Connective tissue | 484 | 2 |
| Epithelial tissue | 804 | 3 |
| Muscular tissue | 514 | 4 |
The composition of the ISIC2017 dataset.
| Image category | Number of images | Label |
|---|---|---|
| Melanoma | 374 | 1 |
| Nevus of seborrheic keratosis | 1626 | 2 |
Comparison of the classification algorithms accuracy.
| Algorithm | HIS2828 | ISIC2017 |
|---|---|---|
| SVM (traditional feature) | 72.17% | 66.1% |
| Coding network | 79.5% | 75% |
| CNMP | 90.2% | 90.1% |
| R feature fusion | 86.3% | 88.7% |
| SVM (traditional and deep feature) | 81.1% | 77.6% |
| KPCA feature fusion | 84.9% | 87.4% |
Figure 2Comparison of the confusion matrix on the histology dataset. (a) The confusion matrix of SVM (traditional features). (b) The confusion matrix of coding network. (c) The confusion matrix of R feature fusion. (d) The confusion matrix of CNMP. (e) The confusion matrix of SVM (traditional and deep feature). (f) The confusion matrix of KPCA feature fusion.
Figure 3The ROC curve on the ISIC2017 dataset.
Comparison of the AUCs on the ISIC dataset.
| Algorithm | The AUC of ROC |
|---|---|
| SVM (traditional feature) | 0.7209 |
| Coding network | 0.8087 |
| CNMP | 0.9585 |
| R feature fusion | 0.9436 |
| SVM (traditional and deep feature) | 0.8210 |
| KPCA feature fusion | 0.9326 |
Figure 4The running time of different algorithms.
Figure 5The variation m influence on algorithm's accuracy.
Figure 6The variation m influence on running time.