| Literature DB >> 35047515 |
Ziquan Zhu1, Siyuan Lu1, Shui-Hua Wang1, Juan Manuel Górriz2, Yu-Dong Zhang1,3.
Abstract
Aims: Most blood diseases, such as chronic anemia, leukemia (commonly known as blood cancer), and hematopoietic dysfunction, are caused by environmental pollution, substandard decoration materials, radiation exposure, and long-term use certain drugs. Thus, it is imperative to classify the blood cell images. Most cell classification is based on the manual feature, machine learning classifier or the deep convolution network neural model. However, manual feature extraction is a very tedious process, and the results are usually unsatisfactory. On the other hand, the deep convolution neural network is usually composed of massive layers, and each layer has many parameters. Therefore, each deep convolution neural network needs a lot of time to get the results. Another problem is that medical data sets are relatively small, which may lead to overfitting problems.Entities:
Keywords: ResNet-18; blood cells; convolutional neural network; randomized neural network; transfer learning
Year: 2022 PMID: 35047515 PMCID: PMC8762289 DOI: 10.3389/fcell.2021.813996
Source DB: PubMed Journal: Front Cell Dev Biol ISSN: 2296-634X
The details of the data set.
| Eosinophil | Lymphocyte | Monocyte | |
|---|---|---|---|
| Training set | 2,497 | 2,483 | 2,478 |
| Testing set | 623 | 620 | 620 |
FIGURE 1Explanations of CNN. (A) Convolution layer flow chart. (B) An example of average and max pooling. (C) Activation function.
FIGURE 2Explanation of the proposed BCNet. (A) The flowchart of the proposed BCNet. (B) Original block. (C) Residual learning. (D) The transfer learning in the ResNet-18.
Pseudocode of the proposed BCNet.
| Step 1: Load the pre-trained ResNet-18. |
| Step 2: Divide the blood cell data set into training and testing sets. |
| Step 3: Preprocessing |
| Resize samples in the training and testing set based on the input size of ResNet-18. |
| Step 4: Generate the transferred ResNet-18. |
| Step 4.1: Remove FC1000, softmax, and classification layer from the pre-trained ResNet-18. |
| Step 4.2: Add FC256, ReLU, FC3, softmax, and classification layer. |
| Step 5: Train the transferred ResNet-18. |
| Step 5.1: Input is the processed training set. |
| Step 5.2: Target is the corresponding labels. |
| Step 6: Replace the last 4 layers of the trained transferred ResNet-18 with three RNNs. |
| Step 7: Extract features |
| Step 8: Train the three RNNs on the extracted features |
| Step 8.1: Input is the extracted features |
| Step 8.2: Target is the labels of the processed training set. |
| Step 9: Add the majority voting layer. |
| Step 9.1: Ensemble the predictions of the three RNNs. |
| Step 9.2: Majority voting of the ensemble of the predictions from the three RNNs. |
| Step 9.3: The whole network is named BCNet. |
| Step 10: Test the trained BCNet on the processed testing set. |
| Step 11: Report the classification performance of the trained BCNet. |
FIGURE 3Structure of three RNNs. (A) RVFL. (B) ELM. (C) SNN.
Other proposed models.
| Proposed individual model (Abbreviation) | Meaning | Training |
|---|---|---|
| ResNet-18-RVFL (BCRRNet) | We select RVFL to substitute the end four layers of the trained transferred ResNet-18 and get BCRRNet. | RVFL in the BCRRNet is trained by the features which are extracted from FC256. |
| ResNet-18-ELM (BCRENet) | We select ELM to substitute the end four layers of the trained transferred ResNet-18 and get BCRENet. | ELM in the BCRENet is trained by the features which are extracted from FC256. |
| ResNet-18-SNN (BCRSNet) | We select SNN to substitute the end four layers of the trained transferred ResNet-18 and get BCRSNet. | SNN in the BCRSNet is trained by the features which are extracted from FC256. |
| Proposed ensemble model (Abbreviation) | Meaning | Training |
| AlexNet-RNNs-En (BCARENet) | The pre-trained AlexNet is the backbone of the BCARENet and the results of the BCARENet are generated by the ensemble of the predictions from the three RNNs by the majority voting. | The trained transferred AlexNet is obtained by training the transferred AlexNet on the processed training blood cell data set. Then, three RNNs in the BCARENet are trained by the features which are extracted from FC256. |
| ResNet-50-RNNs-En (BCR5RENet) | The pre-trained ResNet-50 is the backbone of the BCR5RENet and the results of the BCR5RENet are generated by the ensemble of the predictions from the three RNNs by the majority voting. | The trained transferred ResNet-50 is obtained by training the pre-trained ResNet-50 on the processed training blood cell data set. Then, three RNNs in the BCR5RENet are trained by the features which are extracted from FC256. |
| MobileNet-V2-RNNs-En (BCMV2RENet) | The pre-trained MobileNet-V2 is the backbone of the BCMV2RENet and the results of the BCMV2RENet are generated by the ensemble of the predictions from the three RNNs by the majority voting. | The trained transferred MobileNet-V2 is obtained by training the pre-trained MobileNet-V2 on the processed training blood cell data set. Then, three RNNs in the BCMV2RENet are trained by the features which are extracted from FC256. |
The test confusion matrix of BCNet.
| Predicted class | ||||
|---|---|---|---|---|
| Eosinophil | Lymphocyte | Monocyte | ||
| Actual class | Eosinophil | 623 | 0 | 0 |
| Lymphocyte | 0 | 620 | 0 | |
| Monocyte | 60 | 0 | 560 | |
The results of each category.
| Category | Precision (%) | Recall (%) | F1 (%) |
|---|---|---|---|
| Eosinophil | 91.22 | 100 | 95.41 |
| Lymphocyte | 100 | 100 | 100 |
| Monocyte | 100 | 90.32 | 94.92 |
Confusion matrixes of other proposed models.
| Predicted | |||||
|---|---|---|---|---|---|
| Eosinophil | Lymphocyte | Monocyte | |||
| BCRRNet (Individual) | Actual | Eosinophil | 621 | 2 | 0 |
| Lymphocyte | 0 | 619 | 1 | ||
| Monocyte | 72 | 0 | 548 | ||
| BCRENet (Individual) |
| ||||
| Eosinophil | Lymphocyte | Monocyte | |||
| Actual | Eosinophil | 623 | 0 | 0 | |
| Lymphocyte | 0 | 619 | 1 | ||
| Monocyte | 68 | 0 | 552 | ||
| BCRSNet (Individual) |
| ||||
| Eosinophil | Lymphocyte | Monocyte | |||
| Actual | Eosinophil | 623 | 0 | 0 | |
| Lymphocyte | 0 | 619 | 1 | ||
| Monocyte | 65 | 0 | 555 | ||
| BCARENet (Ensemble) |
| ||||
| Eosinophil | Lymphocyte | Monocyte | |||
| Actual | Eosinophil | 432 | 62 | 129 | |
| Lymphocyte | 7 | 600 | 13 | ||
| Monocyte | 176 | 0 | 444 | ||
| BCR5RENet (Ensemble) |
| ||||
| Eosinophil | Lymphocyte | Monocyte | |||
| Actual | Eosinophil | 550 | 73 | 0 | |
| Lymphocyte | 0 | 620 | 0 | ||
| Monocyte | 90 | 0 | 530 | ||
| BCMV2RENet (Ensemble) |
| ||||
| Eosinophil | Lymphocyte | Monocyte | |||
| Actual | Eosinophil | 582 | 35 | 6 | |
| Lymphocyte | 8 | 601 | 11 | ||
| Monocyte | 153 | 7 | 460 | ||
Comparison of the proposed BCNet with other proposed models.
| Model | Accuracy (%) | Average-precision (%) | Average-recall (%) | Average-F1 (%) |
|---|---|---|---|---|
| BCRRNet (Individual) | 95.97 | 96.37 | 95.97 | 96.17 |
| BCRENet (Individual) | 96.30 | 96.66 | 96.29 | 96.47 |
| BCRSNet (Individual) | 96.46 | 96.79 | 96.45 | 96.62 |
| BCARENet (Ensemble) | 79.23 | 78.88 | 79.24 | 79.01 |
| BCR5RENet (Ensemble) | 91.25 | 91.80 | 91.26 | 91.24 |
| BCMV2RENet (Ensemble) | 88.19 | 89.41 | 88.18 | 88.08 |
|
|
|
|
|
|
FIGURE 4Comparison of the proposed BCNet with other proposed models.
FIGURE 5Explainability of our proposed BCNet. (A) One raw image of Eosinophils. (B) One raw image of Lymphocytes. (C) One raw image of Monocytes. (D) Heatmap of (A). (E) Heatmap of (B). (F) Heatmap of (C).
Comparison with other state-of-the-art methods.
| Method | Accuracy | Average-precision | Average-recall | Average-F1 | Source | Category |
|---|---|---|---|---|---|---|
| CNN+RNN | 90.79% | — | — | — | Public | Four |
| ML | 86.70% | 86.19% | 86.25% | 86.22% | Private | Three |
| Q-fuzzy | — | 85% | 86% | 85% | Public | Seven |
| DCRN+R2U | 91.14% | — | — | 81.80% | Public | Four |
| BCNet (Ours) | 96.78% | 97.07% | 96.77% | 96.78% | Public | Three |
FIGURE 6Comparison with other state-of-the-art methods.