Debiasing pipeline improves deep learning model generalization for X-ray based lung nodule detection
Not on the map. Why?
Abstract (arXiv)
Lung cancer is the leading cause of cancer death worldwide and a good prognosis depends on early diagnosis. Unfortunately, screening programs for the early diagnosis of lung cancer are uncommon. This is in-part due to the at-risk groups being located in rural areas far from medical facilities. Reaching these populations would require a scaled approach that combines mobility, low cost, speed, accuracy, and privacy. We can resolve these issues by combining the chest X-ray imaging mode with a federated deep-learning approach, provided that the federated model is trained on homogenous data to ensure that no single data source can adversely bias the model at any point in time. In this study we show that an image pre-processing pipeline that homogenizes and debiases chest X-ray images can improve both internal classification and external generalization, paving the way for a low-cost and accessible deep learning-based clinical system for lung cancer screening. An evolutionary pruning mechanism is used to train a nodule detection deep learning model on the most informative images from a publicly available lung nodule X-ray dataset. Histogram equalization is used to remove systematic differences in image brightness and contrast. Model training is performed using all combinations of lung field segmentation, close cropping, and rib suppression operators. We show that this pre-processing pipeline results in deep learning models that successfully generalize an independent lung nodule dataset using ablation studies to assess the contribution of each operator in this pipeline. In stripping chest X-ray images of known confounding variables by lung field segmentation, along with suppression of signal noise from the bone structure we can train a highly accurate deep learning lung nodule detection algorithm with outstanding generalization accuracy of 89% to nodule samples in unseen data.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 25 of its 76 references that are in the map. First posted on arXiv 24 Jan 2022.
Builds on
25 of 76 references.
References · 25
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2019CheXpert: A Large Chest Radiograph Dataset with Uncertainty Labels and Expert Comparisoncited by 1,410
- 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 969
- 2018Confounding variables can degrade generalization performance of radiological deep learning modelscited by 374
- 2017Road Extraction by Deep Residual U-Netcited by 333
- 2016Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challengecited by 198
- 2019Flexibly Fair Representation Learning by Disentanglementcited by 196
- 2020Common pitfalls and recommendations for using machine learning to detect and prognosticate for COVID-19 using chest radiographs and CT scanscited by 158
- 2020CheXclusion: Fairness gaps in deep chest X-ray classifierscited by 153
13 more
- 2017Controllable Invariance through Adversarial Feature Learningcited by 127
- 2018Comparison of Deep Learning Approaches for Multi-Label Chest X-Ray Classificationcited by 57
- 2020Interpreting Chest X-rays via CNNs that Exploit Hierarchical Disease Dependencies and Uncertainty Labelscited by 56
- 2018Deep Learning Predicts Hip Fracture using Confounding Patient and Healthcare Variablescited by 50
- 2019Mitigating Information Leakage in Image Representations: A Maximum Entropy Approachcited by 48
- 2018SDFN: Segmentation-based Deep Fusion Network for Thoracic Disease Classification in Chest X-ray Imagescited by 14
- 2020Impact of lung segmentation on the diagnosis and explanation of COVID-19 in chest X-ray imagescited by 14
- 2017Fully Convolutional Architectures for Multi-Class Segmentation in Chest Radiographscited by 13
- 2018Automatic Lung Cancer Prediction from Chest X-ray Images Using Deep Learning Approachcited by 12
- 2020Segmentation of Lungs in Chest X-Ray Image Using Generative Adversarial Networkscited by 7
- 2018Dimensionality Reduction in Deep Learning for Chest X-Ray Analysis of Lung Cancercited by 5
- 2017Accurate Lung Segmentation via Network-Wise Training of Convolutional Networkscited by 3
- 2020Automated Identification of Thoracic Pathology from Chest Radiographs with Enhanced Training Pipelinecited by 2
Built on by
Citing papers per year: 2023: 1.
1 paper.