| Literature DB >> 35742039 |
Amal H Alharbi1, Hanan A Hosni Mahmoud1.
Abstract
Pneumonia is a common disease that occurs in many countries, more specifically, in poor countries. This disease is an obstructive pneumonia which has the same impression on pulmonary radiographs as other pulmonary diseases, which makes it hard to distinguish even for medical radiologists. Lately, image processing and deep learning models are established to rapidly and precisely diagnose pneumonia disease. In this research, we have predicted pneumonia diseases dependably from the X-ray images, employing image segmentation and machine learning models. A public labelled database is utilized with 4000 pneumonia disease X-rays and 4000 healthy X-rays. ImgNet and SqueezeNet are utilized for transfer learning from their previous computed weights. The proposed deep learning models are trained for classifying pneumonia and non-pneumonia cases. The following processes are presented in this paper: X-ray segmentation utilizing BoxENet architecture, X-ray classification utilizing the segmented chest images. We propose the improved BoxENet model by incorporating transfer learning from both ImgNet and SqueezeNet using a majority fusion model. Performance metrics such as accuracy, specificity, sensitivity and Dice are evaluated. The proposed Improved BoxENet model outperforms the other models in binary and multi-classification models. Additionally, the Improved BoxENet has higher speed compared to other models in both training and classification.Entities:
Keywords: classification; deep learning; pneumonia; pulmonary diseases
Year: 2022 PMID: 35742039 PMCID: PMC9223174 DOI: 10.3390/healthcare10060987
Source DB: PubMed Journal: Healthcare (Basel) ISSN: 2227-9032
Recent research in pneumonia prediction deep learning models.
| Ref. | Method | Model | Database | Average Accuracy |
|---|---|---|---|---|
| [ | Binary classification | Spatial similarity matrix | Contrast- | 90.23% |
| [ | Pneumonia identification | Region CNN | 4064 X-rays of 1200 patients | 89.76% |
| [ | Classification of pneumonia and healthy cases | Capsule CNN | 3770 X-rays | 93.7% |
| [ | Classification of pneumonia into three stages (preliminary, moderate, severe cases) | Deep CNN Architecture | 2044 X-rays of 2000 patients (average 2 X-rays per patient) | 96.4% |
| [ | Pneumonia and tuberculosis classifications | CNN and discrete wavelet transform | 2054 X-rays | 94.5–97.5% |
| [ | Pneumonia classification | Transfer learning deep CNN | 5942 X-rays | 96.5% with less CPU time |
| [ | Pneumonia classification | Deep learning Region-based CNN method | 2115 chest X-ray | 97.67% |
| [ | Pneumonia grading | Textural based feature extraction | 640 lung X-ray | 97.2% |
| [ | Pneumonia classification | Texture and hue feature extraction | 670 cases with 1580 X-rays | 95.8% |
| [ | X-ray pneumonia grading | Genetic algorithms with deep learning | Unknown | achieves better performance than solo CNN |
| [ | Prediction of pneumonia with high speed | High-speed region CNN | 320 chest X-rays | 97.8%, with a small-size database advantage |
The architecture of the BoxENet and its parameters.
| Layer Number | Layer Type | Properties |
|---|---|---|
| 1 | Input Layer | 512 × 512 images |
| 2 | First Dense Block | 60 layers of 128 × 5 × 5 convolutions |
| 3 | Pooling Block | 4 × 4 Max pooling |
| 4 | Second Dense Block | 30 layers of (32 × 3 × 3) convolutions |
| 5 | Pooling Block | 3 × 3 max pooling |
| 6 | Third Dense Block | 40 (16 × 3 × 3) |
| 7 | Pooling Block | 2 × 2 average pooling |
| 8 | Fully Connected (FC) Layer | 2048 hidden neurons |
| 9 | Softmax classifier | Softmax |
| 10 | Binary Classifier Output | Binary output classes: Pneumonia; Healthy. |
| Multiclassifier Output | Multi output classes: Congestion; Red Hepatization; Grey Hepatization; Resolution. |
Figure 1The architecture of the BoxENet.
Figure 2X-ray instance and their analogous mask from the Kaggle database.
Figure 3Example of X-rays from the utilized databases. (A) CHN dataset, (B) MC dataset, (C) BelPnem dataset and (D) RPNA dataset.
Figure 4Detailed structure of the improved BoxENet with transfer learning.
Details of the test dataset for the BoxENet Segmentation phase.
| Database | Total Number of X-rays | Training Subset | Validation Subset | Testing Subset |
|---|---|---|---|---|
| Kaggle | 1000 | 700 | 150 | 150 |
Training, validation and test subsets for the deep learning model.
| Data Set | Training with Whole X-ray/Segmented X-ray | ||||
|---|---|---|---|---|---|
| Training Subset | Validation Subset | Testing Subset | |||
| CHN, MC and BelPnem | Healthy | 4000 | 2800 | 600 | 600 |
| Pneumonia | 4000 | 2800 | 600 | 600 | |
Comparative Performance of Original BoxENet and Improved BoxENet.
| Batch Size | Loss Function | CNN | Testing Loss | Accuracy % |
|---|---|---|---|---|
| 8 | Dice | BoxENet | 0.0321 | 92.5 |
| 16 | Dice | BoxENet | 0.0223 | 93.1 |
| 32 | Dice | BoxENet | 0.0133 | 92.5 |
| 64 | Dice | BoxENet | 0.0132 | 93.2 |
| 8 | Dice | Improved BoxENet (with transfer learning and majority voting) | 0.0012 | 98.7 |
| 16 | Dice | Improved BoxENet | 0.0203 | 95.8 |
| 32 | Dice | Improved BoxENet | 0.0010 | 97.2 |
| 64 | Dice | Improved BoxENet | 0.0092 | 97.3 |
| 8 | SGD | BoxENet | 0.324 | 91.7 |
| 16 | SGD | BoxENet | 0.223 | 91.4 |
| 32 | SGD | BoxENet | 0.343 | 92.6 |
| 64 | SGD | BoxENet | 0.442 | 92.4 |
| 8 | SGD | Improved BoxENet | 0.124 | 95.9 |
| 16 | SGD | Improved BoxENet | 0.203 | 94.8 |
| 32 | SGD | Improved BoxENet | 0.440 | 94.9 |
| 64 | SGD | Improved BoxENet | 0.1392 | 96.3 |
Definitions of TP, TN, FP and FN.
| Term | Definition |
|---|---|
| True positive | Count of pneumonia X-rays detected as pneumonia |
| True negative | Count of normal X-rays detected as normal |
| False positive | Count of normal X-rays detected as pneumonia |
| False negative | Count of pneumonia X-rays detected as normal |
The highest accurate parameters for X-ray segmentation.
| The Highest Accurate Parameters for X-ray Segmentation | ||
|---|---|---|
| X-ray Segmentation Process | Classification Process | |
| Batch size | 16 | 32 |
| Learning rate | 0.0015 | 0.0015 |
| Number of epochs | 80 | 80 |
| Stopping parameter | 3 | 3 |
Figure 5X-ray radiography, with segmented Pulmonary original BoxENet and Segmented Pulmonary Improved BoxENet.
Comparison of various CNNs for pneumonia classification with whole and segmented X-rays.
| Reference | Accuracy [%] | Sensitivity [%] | Specificity [%] | Average CPU Time per Testing of a Single X-ray | Average CPU Time per Training of a Single Epoch | |
|---|---|---|---|---|---|---|
| Whole X-ray (without segmentation) | InceptV5 [ | 94.04 | 94.80 | 93.40 | 0.92 | 72.7 |
| DSNet3 [ | 93.68 | 93.30 | 94.00 | 1.89 | 101.89 | |
| DSNet5 [ | 94.68 | 94.60 | 93.44 | 2.78 | 176.7 | |
| Improved BoxENet | 97.40 | 98.44 | 97.34 | 0.98 | 38.7 | |
| Segmented X-ray | InceptV5 [ | 94.04 | 94.80 | 93.40 | 1.4 | 45.6 |
| DSNet3 [ | 96.68 | 96.30 | 95.31 | 3.34 | 80.9 | |
| DSNet5 [ | 94.68 | 94.60 | 93.44 | 5.89 | 120.76 | |
| Improved BoxENet | 95.40 | 96.44 | 95.34 | 0.56 | 32.5 |
Comparison of various CNNs for pneumonia multi-classification with whole and segmented X-rays.
| Model | Accuracy [%] | Sensitivity [%] | Specificity [%] | Average CPU Time per Testing of a Single X-ray | Average CPU Time per Training of a Single Epoch | |
|---|---|---|---|---|---|---|
| Whole X-ray (without segmentation) | InceptV6 [ | 96.06 | 96.90 | 94.60 | 0.73 | 73.7 |
| DSNet4 [ | 94.69 | 94.40 | 96.00 | 1.77 | 101.77 | |
| DSNet6 [ | 96.69 | 96.60 | 94.66 | 3.77 | 176.7 | |
| Improved BoxENet | 97.60 | 99.66 | 97.46 | 0.77 | 67.7 | |
| Segmented X-ray | InceptV6 [ | 96.06 | 96.90 | 94.60 | 1.6 | 66.6 |
| DSNet4 [ | 96.69 | 96.40 | 96.41 | 6.66 | 70.7 | |
| DSNet6 [ | 96.69 | 96.60 | 94.66 | 6.77 | 130.76 | |
| Improved BoxENet | 98.60 | 98.66 | 98.46 | 0.66 | 63.6 |
Figure 6Confusion matrix for multi-classification for the improved BoxENet model without segmented X-ray.
Figure 7Confusion matrix for multiclassification for the improved BoxENet model with segmented X-ray.
Figure 8Accuracy in the training phase of the proposed improved BoxENet model.
Figure 9Training loss for the proposed improved BoxENet model.