| Literature DB >> 33157999 |
Furong Huang1, Peiwen Guang1, Fucui Li1, Xuewen Liu1, Weimin Zhang2, Wendong Huang3.
Abstract
Leukemia diagnosis based on bone marrow cell morphology primarily relies on the manual microscopy of bone marrow smears. However, this method is greatly affected by subjective factors and tends to lead to misdiagnosis. This study proposes using bone marrow cell microscopy images and employs convolutional neural network (CNN) combined with transfer learning to establish an objective, rapid, and accurate method for classification and diagnosis of LKA (AML, ALL, and CML). We collected cell microscopy images of 104 bone marrow smears (including 18 healthy subjects, 53 AML patients, 23 ALL patients, and 18 CML patients). The perfect reflection algorithm and a self-adaptive filter algorithm were first used for preprocessing of bone marrow cell images collected from experiments. Subsequently, 3 CNN frameworks (Inception-V3, ResNet50, and DenseNet121) were used to construct classification models for the raw dataset and preprocessed dataset. Transfer learning was used to improve the prediction accuracy of the model. Results showed that the DenseNet121 model based on the preprocessed dataset provided the best classification results, with a prediction accuracy of 74.8%. The prediction accuracy of the DenseNet121 model that was obtained by transfer learning optimization was 95.3%, which was increased by 20.5%. In this model, the prediction accuracies of the normal groups, AML, ALL, and CML were 90%, 99%, 97%, and 95%, respectively. The results showed that the leukemic cell morphology classification and diagnosis based on CNN combined with transfer learning is feasible. Compared with conventional manual microscopy, this method is more rapid, accurate, and objective.Entities:
Mesh:
Year: 2020 PMID: 33157999 PMCID: PMC7647529 DOI: 10.1097/MD.0000000000023154
Source DB: PubMed Journal: Medicine (Baltimore) ISSN: 0025-7974 Impact factor: 1.817
Bone marrow smear sample information statistics table.
| Samples | Age | Male | Female | Total |
| Healthy | 18–40 | 12 | 8 | 18 |
| AML | 13–73 | 30 | 23 | 53 |
| ALL | 3–58 | 13 | 11 | 23 |
| CML | 21–68 | 10 | 8 | 10 |
| Total | 65 | 39 | 104 |
Figure 1Model learning process: (a) traditional machine learning; (b) migration learning.
Figure 2Schematic diagram of migration learning combined with convolutional neural network: (a) using the pre-trained model as a fixed feature extractor; (b) fine-tuning the pre-trained model.
Figure 3Dataset image dataset: (a) normal group; (b) acute myelogenous leukemia (c) acute lymphoblastic leukemia; (d) chronic myelocytic leukemia.
Data set partition results.
| Samples | Healthy | AML | ALL | CML | Total |
| Data set | 380 | 400 | 302 | 240 | 1322 |
| Train set | 285 | 300 | 226 | 180 | 991 |
| Test set | 95 | 100 | 76 | 60 | 331 |
Figure 4Dataset microscopic image preprocessing results (a) Original image (b) perfect reflection algorithm (c) perfect reflection algorithm and adaptive median filtering.
Hyperparameter setting of convolutional neural network model.
| Hyperparameter | Settings |
| Optimizer | SGD (Stochastic Gradient Descent) |
| Pooling method | Max-pooling |
| Activation function | ReLU (Rectifier Linear Unit) |
| Loss function | Cross-Entropy |
| Batch-size | 8 |
| Learning rate | 0.001 |
| Momentum | 0.5 |
Model classification results of the original data set.
| Models | Accuracy of train data | Accuracy of prediction data |
| Inception-V3 | 96.5% | 64.3% |
| ResNet50 | 98.4% | 66.2% |
| DenseNet121 | 98.9% | 70.6% |
Model classification results of preprocessed data sets.
| Models | Accuracy of train data | Accuracy of prediction data |
| Inception-V3 | 97.1% | 60.6% |
| ResNet50 | 98.7% | 69.3% |
| DenseNet121 | 99.2% | 74.8% |
Model classification results of different modeling methods on the data set.
| Methods | Model | Accuracy of train set | Accuracy of test set | Time |
| Non | CNN | 99.2% | 74.8% | 45 min |
| Transfer learning | CNN-1 | 99.4% | 84.9% | 8 min |
| CNN-2 | 99.7% | 95.3% | 20 min |
Figure 5Confusion matrix based on the pre-processing data set fine-tuning model convolutional neural network-2 (1. acute myelogenous leukemia; 2. acute lymphoblastic leukemia; 3. chronic myelocytic leukemia; 4. Normal).
Figure 6Comparison of modeling results based on DenseNet framework.