| Literature DB >> 33968160 |
Shangjie Yao1, Yaowu Chen2, Xiang Tian3, Rongxin Jiang3.
Abstract
Pneumonia remains a threat to human health; the coronavirus disease 2019 (COVID-19) that began at the end of 2019 had a major impact on the world. It is still raging in many countries and has caused great losses to people's lives and property. In this paper, we present a method based on DeepConv-DilatedNet of identifying and localizing pneumonia in chest X-ray (CXR) images. Two-stage detector Faster R-CNN is adopted as the structure of a network. Feature Pyramid Network (FPN) is integrated into the residual neural network of a dilated bottleneck so that the deep features are expanded to preserve the deep feature and position information of the object. In the case of DeepConv-DilatedNet, the deconvolution network is used to restore high-level feature maps into its original size, and the target information is further retained. On the other hand, DeepConv-DilatedNet uses a popular fully convolution architecture with computation shared on the entire image. Then, Soft-NMS is used to screen boxes and ensure sample quality. Also, K-Means++ is used to generate anchor boxes to improve the localization accuracy. The algorithm obtained 39.23% Mean Average Precision (mAP) on the X-ray image dataset from the Radiological Society of North America (RSNA) and got 38.02% Mean Average Precision (mAP) on the ChestX-ray14 dataset, surpassing other detection algorithms. So, in this paper, an improved algorithm that can provide doctors with location information of pneumonia lesions is proposed.Entities:
Mesh:
Year: 2021 PMID: 33968160 PMCID: PMC8081632 DOI: 10.1155/2021/8854892
Source DB: PubMed Journal: Comput Math Methods Med ISSN: 1748-670X Impact factor: 2.238
Number of marking frames of lesions in the training set and test set.
| Dataset/lesion areas in each picture | 1 | 2 | 3 | 4 |
|---|---|---|---|---|
| Training set | 2068 | 2628 | 104 | 10 |
| Test set | 545 | 629 | 25 | 3 |
Figure 1Flip processed image.
Figure 2Average gray value distribution histogram.
Figure 3Chest X-ray image preprocessing.
Figure 4The comparison of upsampling and deconvolution.
Figure 5DeepConv-DilatedNet.
Figure 6Network structure for pneumonia detection.
Figure 7Classification loss and regression training loss.
Assessment results for different IoU thresholds.
| AP@0.4 | AP@0.5 | AP@0.6 | AP@0.7 | mAP | |
|---|---|---|---|---|---|
| DetNet59 | 0.6317 | 0.4201 | 0.2068 | 0.0657 | 0.3311 |
| ResNet50 | 0.6066 | 0.3791 | 0.1863 | 0.0513 | 0.3058 |
| ResNet101 | 0.5539 | 0.3508 | 0.1540 | 0.0406 | 0.2748 |
| VGG16 | 0.5506 | 0.3559 | 0.1881 | 0.0660 | 0.4210 |
| DeepConv-DilatedNet |
|
|
|
|
|
Assessment results via Soft-NMS.
| Network | mAP@0.4 | mAP@0.5 | mAP@0.6 | mAP@0.7 | mAP |
|---|---|---|---|---|---|
| DetNet59+Soft-NMS | 0.6617 | 0.4751 | 0.2638 | 0.0879 | 0.3721 |
| DeepConv-DilatedNet+Soft-NMS |
|
|
|
|
|
| ResNet50+Soft-NMS | 0.6234 | 0.4268 | 0.2495 | 0.0842 | 0.3460 |
| ResNet101+Soft-NMS | 0.5790 | 0.3992 | 0.2077 | 0.0630 | 0.3122 |
| VGG16+Soft-NMS | 0.5925 | 0.4179 | 0.2462 | 0.0940 | 0.3377 |
Figure 8The blue and yellow, respectively, represent the AP values before and after filtering the proposals using Soft-NMS, and the numbers on polyline indicates the difference of AP between the two.
Comparison of results for different networks.
| Network | MS |
|---|---|
| Mask R-CNN | 0.2181 |
| DeepConv-DilatedNet+Soft-NMS |
|
Figure 9ROC curve.
Figure 10Precision/recall curve.
Figure 11Comparison of test results of different models.
Assessment results of chest X-ray.
| Network | AP@0.4 | AP@0.5 | AP@0.6 | AP@0.7 | mAP |
|---|---|---|---|---|---|
| DetNet59+Soft-NMS | 0.6343 | 0.4614 | 0.2397 | 0.0875 | 0.3557 |
| DeepConv-DilatedNet+Soft-NMS |
|
|
|
|
|
| ResNet50+Soft-NMS | 0.6053 | 0.4317 | 0.1884 | 0.0737 | 0.3248 |
| ResNet101+Soft-NMS | 0.5813 | 0.4005 | 0.1656 | 0.0636 | 0.3028 |
| VGG16+Soft-NMS | 0.5914 | 0.4109 | 0.1784 | 0.0603 | 0.3103 |