Powered by OpenAIRE graph
Found an issue? Give us feedback
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/ ZENODOarrow_drop_down
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/
ZENODO
Model . 2023
License: CC BY
Data sources: ZENODO
ZENODO
Model . 2023
License: CC BY
Data sources: Datacite
ZENODO
Model . 2023
License: CC BY
Data sources: Datacite
versions View all 2 versions
addClaim

Deep Learning Models for Lung Segmentation in Chest X-ray Radiographs

Authors: Danilov, Viacheslav;

Deep Learning Models for Lung Segmentation in Chest X-ray Radiographs

Abstract

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

Keywords

deep learning models, lung segmentation, chest X-ray

  • BIP!
    Impact byBIP!
    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
    OpenAIRE UsageCounts
    Usage byUsageCounts
    visibility views 31
    download downloads 1
  • 31
    views
    1
    downloads
    Powered byOpenAIRE UsageCounts
Powered by OpenAIRE graph
Found an issue? Give us feedback
visibility
download
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).
BIP!Citations provided by BIP!
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.
BIP!Popularity provided by BIP!
influence
This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Influence provided by BIP!
impulse
This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
BIP!Impulse provided by BIP!
views
OpenAIRE UsageCountsViews provided by UsageCounts
downloads
OpenAIRE UsageCountsDownloads provided by UsageCounts
0
Average
Average
Average
31
1