| Literature DB >> 29181087 |
Er-Yang Huan1, Gui-Hua Wen1, Shi-Jun Zhang2, Dan-Yang Li1, Yang Hu1, Tian-Yuan Chang1, Qing Wang1, Bing-Lin Huang1.
Abstract
Body constitution classification is the basis and core content of traditional Chinese medicine constitution research. It is to extract the relevant laws from the complex constitution phenomenon and finally build the constitution classification system. Traditional identification methods have the disadvantages of inefficiency and low accuracy, for instance, questionnaires. This paper proposed a body constitution recognition algorithm based on deep convolutional neural network, which can classify individual constitution types according to face images. The proposed model first uses the convolutional neural network to extract the features of face image and then combines the extracted features with the color features. Finally, the fusion features are input to the Softmax classifier to get the classification result. Different comparison experiments show that the algorithm proposed in this paper can achieve the accuracy of 65.29% about the constitution classification. And its performance was accepted by Chinese medicine practitioners.Entities:
Mesh:
Year: 2017 PMID: 29181087 PMCID: PMC5664380 DOI: 10.1155/2017/9846707
Source DB: PubMed Journal: Comput Math Methods Med ISSN: 1748-670X Impact factor: 2.238
Figure 1The flow chart of the whole algorithm.
The number of samples of different constitution types.
| Gentleness | Qi-deficiency | Yang-deficiency | Yin-deficiency | Phlegm-dampness | Dampness-heat | Blood-stasis | Qi-depression | Sum | |
|---|---|---|---|---|---|---|---|---|---|
| Number | 570 | 750 | 600 | 750 | 750 | 750 | 410 | 750 | 5330 |
Figure 2The structure of convolutional neural networks for extracting features.
The classification results under different feature extraction methods.
| SVM | Random Forest | KNN | Softmax | Decision Tree | Gradient BoostTree | Naive Bayes | |
|---|---|---|---|---|---|---|---|
| Color feature | 23.26% | 25.89% | 26.08% | 19.14% | 14.63% | 19.32% | 16.14% |
| Color and texture features | 29.64% | 40.87% | 29.46% | 22.68% | 19.14% | 22.89% | 17.63% |
| CNN | 63.55% | 64.23% | 63.23% | 64.54% | 60.97% | 62.78% | 63.78% |
The confusion matrix of random forest classification based on color texture feature Fusion.
| Qi-deficiency | Yin-deficiency | Yang-deficiency | Phlegm-dampness | Dampness-heat | Qi-depression | Blood-stasis | Gentleness | |
|---|---|---|---|---|---|---|---|---|
| Qi-deficiency | 38 | 6 | 6 | 2 | 7 | 14 | 0 | 2 |
| Yin-deficiency | 9 | 44 | 1 | 4 | 6 | 10 | 0 | 1 |
| Yang-deficiency | 17 | 10 | 17 | 2 | 3 | 8 | 0 | 3 |
| Phlegm-dampness | 17 | 13 | 1 | 34 | 4 | 6 | 0 | 0 |
| Dampness-heat | 10 | 5 | 2 | 3 | 45 | 9 | 0 | 1 |
| Qi-depression | 4 | 5 | 2 | 4 | 11 | 31 | 0 | 18 |
| Blood-stasis | 14 | 8 | 1 | 3 | 4 | 8 | 3 | 0 |
| Gentleness | 12 | 10 | 2 | 5 | 7 | 16 | 2 | 3 |
The confusion matrix of Softmax classification based on convolutional neural network.
| Qi-deficiency | Yin-deficiency | Yang-deficiency | Phlegm-dampness | Dampness-heat | Qi-depression | Blood-stasis | Gentleness | |
|---|---|---|---|---|---|---|---|---|
| Qi-deficiency | 51 | 11 | 2 | 5 | 5 | 0 | 1 | 0 |
| Yin-deficiency | 7 | 61 | 2 | 2 | 0 | 2 | 1 | 0 |
| Yang-deficiency | 10 | 3 | 43 | 1 | 1 | 1 | 1 | 0 |
| Phlegm-dampness | 5 | 6 | 1 | 62 | 1 | 0 | 0 | 0 |
| Dampness-heat | 8 | 3 | 2 | 4 | 57 | 0 | 1 | 0 |
| Qi-depression | 1 | 4 | 0 | 8 | 3 | 39 | 0 | 20 |
| Blood-stasis | 5 | 0 | 0 | 4 | 2 | 0 | 30 | 0 |
| Gentleness | 26 | 10 | 3 | 6 | 2 | 7 | 2 | 1 |
The classification results based on the convolution neural network feature extraction and color feature fusion.
| SVM | Random Forest | KNN | Softmax | Decision Tree | Gradient BoostTree | Naive Bayes | |
|---|---|---|---|---|---|---|---|
| CNN | 63.55% | 64.23% | 63.23% | 64.54% | 60.97% | 62.78% | 63.78% |
| CNN + color | 63.98% | 64.91% | 62.34% | 65.29% | 59.85% | 64.72% | 63.04% |
The confusion matrix of Softmax classification based on convolutional neural network and color feature fusion.
| Qi-deficiency | Yin-deficiency | Yang-deficiency | Phlegm-dampness | Dampness-heat | Qi-depression | Blood-stasis | Gentleness | |
|---|---|---|---|---|---|---|---|---|
| Qi-deficiency | 36 | 9 | 2 | 8 | 12 | 3 | 5 | 0 |
| Yin-deficiency | 2 | 65 | 2 | 3 | 2 | 0 | 1 | 0 |
| Yang-deficiency | 8 | 2 | 45 | 1 | 2 | 1 | 1 | 0 |
| Phlegm-dampness | 4 | 4 | 0 | 61 | 3 | 1 | 2 | 0 |
| Dampness-heat | 7 | 1 | 1 | 3 | 63 | 0 | 0 | 0 |
| Qi-depression | 0 | 3 | 0 | 5 | 6 | 40 | 1 | 20 |
| Blood-stasis | 2 | 1 | 0 | 2 | 2 | 1 | 33 | 0 |
| Gentleness | 20 | 8 | 2 | 6 | 7 | 7 | 2 | 5 |
Figure 3The ROC curve of different classifiers based on the feature of convolution neural network and color feature fusion. The dotted black line is the baseline in ROC curve. It indicates that the true positive rate (TPR) is equal to the false positive rate (FPR).
Figure 4The precision-recall curve of different classifiers based on the feature of convolution neural network and color feature fusion.
Figure 5The micro-average and macro-average ROC curve in the Softmax based on the convolution neural network and the color feature fusion. The dotted black line is the baseline in ROC curve. It indicates that the true positive rate (TPR) is equal to the false positive rate (FPR).
Figure 6The ROC curve of each label in the Softmax based on the convolution neural network and the color feature fusion. The dotted black line is the baseline in ROC curve. It indicates that the true positive rate (TPR) is equal to the false positive rate (FPR).
The classification results with the increase of data.
| SVM | Random Forest | KNN | Naive Bayes | Softmax | Decision Tree | Gradient BoostTree | |
|---|---|---|---|---|---|---|---|
| 3010 | 44.19% | 46.18% | 46.84% | 38.21% | 43.19% | 37.21% | 41.86% |
| 4470 | 54.36% | 53.69% | 52.35% | 52.57% | 54.14% | 46.53% | 53.69% |
| 5330 | 63.98% | 64.91% | 62.34% | 63.07% | 65.29% | 59.89% | 64.72% |