Downloads provided by UsageCounts
Introduction: We present a collection of lung segmentation models designed for accurate analysis of chest X-ray (CXR) images. These models have undergone rigorous training, validation, hyperparameter tuning, and testing on a diverse dataset consisting of CXRs from three publicly available sources: the Darwin, Montgomery, and Shenzhen datasets. Dataset Description: The dataset includes CXRs from patients diagnosed with various respiratory diseases, including COVID-19, pneumonia, and tuberculosis. While these conditions differ in nature and disease patterns, we intentionally included them in the training data to improve the ability of the models to generalize across a range of pathologies. The dataset is summarized in Table 1 and publicly available at https://data.mendeley.com/datasets/8gf9vpkhgy/2. Model Selection: In our model selection process for lung segmentation, we rigorously evaluated several state-of-the-art models, including U-net, U-net++, DeepLabV3, DeepLabV3+, FPN, Linknet, PSPNet, PAN, and MA-Net. These models were trained and tested on a Linux machine equipped with an NVIDIA GeForce RTX 3090 boasting 24 GB of memory. The training procedure closely followed the methodology described in the paper "Automatic scoring of COVID-19 severity in X-ray imaging based on a novel deep learning workflow", specifically as outlined in the Model Training section. Model Comparison: After a thorough evaluation, DeepLabV3+ emerged as our top choice. It exhibited exceptional performance with a DSCtest score of 0.963, while maintaining computational efficiency with a total of 7.4 million parameters and 2.2 billion MACs. This combination of high performance and computational efficiency solidified its position as the optimal model for our lung segmentation task. In Table 2, we provide a comprehensive overview of the performance metrics for each model, including batch size, memory usage, parameters, and MACs. Additionally, we present the DSC values, enabling a detailed comparison of each model's performance across the training, validation, and testing phases. Requirements: The models were originally trained and tested using PyTorch v1.8.1 and timm v0.3.2. Subsequently, we conducted additional tests on PyTorch v1.13.1 and timm v0.6.12, and all models performed well. However, it's worth noting that Linknet and PSPNet encountered errors on the latter version and are not compatible with it. Access to the Study: Further information about this study, including curated source code, dataset details, and trained models, can be accessed through the following repositories: Source code: https://github.com/ViacheslavDanilov/covid_scoring Dataset (Lung segmentation): https://data.mendeley.com/datasets/8gf9vpkhgy/2 Dataset (COVID-19 segmentation): https://data.mendeley.com/datasets/36fjrg9s69/1 Models: https://doi.org/10.5281/zenodo.8393555 Table 1. Description of the datasets used for lung segmentation Dataset Training Validation Testing Total Darwin 4884 611 611 6106 / 90% Montgomery 110 14 14 138 / 2% Shenzhen 452 57 57 566 / 8% Total 5446 / 80% 682 / 10% 682 / 10% 6810 / 100% Table 2. Results of the fully trained lung segmentation networks Model Batch size Memory, Gb Parameters, M MAC, G DSCtrain DSCval DSCtest U-net 24 21 55.9 35.6 0.960 0.960 0.962 U-net++ 32 24 9.1 11.5 0.956 0.959 0.960 DeepLabV3 16 24 7.3 13.4 0.960 0.960 0.961 DeepLabV3+ 20 22 7.4 2.2 0.962 0.960 0.963 FPN 32 23 5.8 9.1 0.962 0.960 0.962 Linknet 24 24 29.3 39.9 0.965 0.959 0.961 PSPNet 40 23 29.7 14.4 0.959 0.959 0.960 PAN 32 24 4.1 0.3 0.961 0.960 0.962 MA-Net 24 24 13.4 9.9 0.956 0.959 0.961
deep learning models, lung segmentation, chest X-ray
deep learning models, lung segmentation, chest X-ray
| selected citations These citations are derived from selected sources. This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | 0 | |
| popularity This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network. | Average | |
| influence This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | Average | |
| impulse This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network. | Average |
| views | 31 | |
| downloads | 1 |

Views provided by UsageCounts
Downloads provided by UsageCounts