Multi-Backbone U-Net segmentation and interpretable hybrid deep learning classification for lung cancer detection in chest CT images
6 affiliations
- 1Department of Computer Science and Engineering, Oakland University, Rochester, MI USA
- 2Department of Computer Science, Bowling Green State University, Bowling Green, OH USA
- 3Department of Applied Statistics & Operations Research, Bowling Green State University, Bowling Green, OH USA
- 4Department of Software Engineering, Faculty of Engineering and Architecture, Istanbul Beykent University, Istanbul, Türkiye
- 5Department of Biomedical Engineering, College of Engineering, University of Kentucky, Lexington, KY USA
- 6Department of Medical Bioengineering, Faculty of Advanced Medical Sciences, Tabriz University of Medical Sciences, Tabriz, Iran
Abstract
Background: Lung cancer remains the leading cause of cancer-related mortality worldwide, and its early and accurate diagnosis from chest CT imaging is complicated by overlapping radiological features between benign and malignant lesions, as well as by the limited cross-institution validation and interpretability reported in much of the existing computer-aided diagnosis literature. This study evaluates the effectiveness of U-Net architectures integrated with different pre-trained convolutional neural network (CNN) backbones for automated lung segmentation and lung cancer classification in chest CT images, with an emphasis on rigorous, leakage-free validation and cross-institution generalizability relevant to clinical practice.
Methods: A balanced internal dataset of 832 chest CT images (416 cancerous and 416 non-cancerous cases, NIDCH, Bangladesh) was preprocessed using Contrast Limited Adaptive Histogram Equalization (CLAHE) and resized to 224 × 224 pixels. U-Net models with three CNN backbones (VGG16, ResNet50, and Xception) were implemented for lung segmentation, with pseudo-labels verified against expert-annotated masks. A standalone CNN and three hybrid classifiers combining CNN feature extraction with machine learning algorithms (Support Vector Machine, Random Forest, and Gradient Boosting) were then evaluated, alongside a Vision Transformer baseline. All models were assessed using subject-level fivefold cross-validation, followed by fully independent external testing on a second, structurally distinct dataset (IQ-OTH/NCCD; 110 subjects) without retraining. Performance was measured using Dice coefficient, IoU, accuracy, precision, recall, F1-score, and AUC, with 95% confidence intervals and statistical significance testing.
Results: For segmentation, U-Net with the Xception backbone achieved the best overall performance (Dice: 0.9581 ± 0.0104; IoU: 0.9163 ± 0.0167). For classification, the CNN + SVM hybrid model achieved the highest internal cross-validation accuracy (98.21% ± 0.68%) and AUC (0.9881 ± 0.0041), significantly outperforming the standalone CNN (p = 0.041), and retained the strongest performance under external validation (92.73% accuracy), with the smallest accuracy drop among all evaluated models. Ablation analyses confirmed substantial contributions from CLAHE preprocessing and U-Net segmentation, and Grad-CAM analysis showed that segmentation constrains classifier attention to anatomically plausible lung regions.
Conclusion: Integrating U-Net segmentation with hybrid CNN-based classification provides an accurate and reasonably generalizable framework for lung cancer detection, offering a methodological foundation for future clinically oriented computer-aided diagnosis research, pending prospective, multi-institution validation.
Related
- Deep learning based edge segmentation and classification for plasterboard defect detection
- Deep Learning for Real-Time Image Interpretation in Arthroscopy: A Systematic Review of Classification, Segmentation, and Detection Tasks
- Multi-task deep learning framework for brain tumor classification and segmentation using ensemble feature extraction
- Boundary-aware multi-scale residual learning with hybrid radiomic–deep fusion for lung tumor segmentation and classification
- Deep learning-based segmentation, grading and genotype classification of glioma: a literature analysis
BibTeX
@article{golkarieh2026multibackbone,
title = {{Multi-Backbone U-Net segmentation and interpretable hybrid deep learning classification for lung cancer detection in chest CT images}},
author = {Alireza Golkarieh and Kiana Kiashemshaki and Sajjad Rezvani Boroujeni and Sanam Ansari and Nasibeh Asadi Isakan and Hossein Najafzadeh},
journal = {BMC Pulmonary Medicine},
year = {2026},
month = aug,
doi = {10.1186/s12890-026-04526-2},
url = {https://doi.org/10.1186/s12890-026-04526-2},
note = {CC BY-NC-ND 4.0},
}
RIS
TY - JOUR
TI - Multi-Backbone U-Net segmentation and interpretable hybrid deep learning classification for lung cancer detection in chest CT images
AU - Alireza Golkarieh
AU - Kiana Kiashemshaki
AU - Sajjad Rezvani Boroujeni
AU - Sanam Ansari
AU - Nasibeh Asadi Isakan
AU - Hossein Najafzadeh
PY - 2026
DA - 2026/08/01
JO - BMC Pulmonary Medicine
DO - 10.1186/s12890-026-04526-2
UR - https://doi.org/10.1186/s12890-026-04526-2
AB - Background: Lung cancer remains the leading cause of cancer-related mortality worldwide, and its early and accurate diagnosis from chest CT imaging is complicated by overlapping radiological features between benign and malignant lesions, as well as by the limited cross-institution validation and interpretability reported in much of the existing computer-aided diagnosis literature. This study evaluates the effectiveness of U-Net architectures integrated with different pre-trained convolutional neural network (CNN) backbones for automated lung segmentation and lung cancer classification in chest CT images, with an emphasis on rigorous, leakage-free validation and cross-institution generalizability relevant to clinical practice. Methods: A balanced internal dataset of 832 chest CT images (416 cancerous and 416 non-cancerous cases, NIDCH, Bangladesh) was preprocessed using Contrast Limited Adaptive Histogram Equalization (CLAHE) and resized to 224 × 224 pixels. U-Net models with three CNN backbones (VGG16, ResNet50, and Xception) were implemented for lung segmentation, with pseudo-labels verified against expert-annotated masks. A standalone CNN and three hybrid classifiers combining CNN feature extraction with machine learning algorithms (Support Vector Machine, Random Forest, and Gradient Boosting) were then evaluated, alongside a Vision Transformer baseline. All models were assessed using subject-level fivefold cross-validation, followed by fully independent external testing on a second, structurally distinct dataset (IQ-OTH/NCCD; 110 subjects) without retraining. Performance was measured using Dice coefficient, IoU, accuracy, precision, recall, F1-score, and AUC, with 95% confidence intervals and statistical significance testing. Results: For segmentation, U-Net with the Xception backbone achieved the best overall performance (Dice: 0.9581 ± 0.0104; IoU: 0.9163 ± 0.0167). For classification, the CNN + SVM hybrid model achieved the highest internal cross-validation accuracy (98.21% ± 0.68%) and AUC (0.9881 ± 0.0041), significantly outperforming the standalone CNN (p = 0.041), and retained the strongest performance under external validation (92.73% accuracy), with the smallest accuracy drop among all evaluated models. Ablation analyses confirmed substantial contributions from CLAHE preprocessing and U-Net segmentation, and Grad-CAM analysis showed that segmentation constrains classifier attention to anatomically plausible lung regions. Conclusion: Integrating U-Net segmentation with hybrid CNN-based classification provides an accurate and reasonably generalizable framework for lung cancer detection, offering a methodological foundation for future clinically oriented computer-aided diagnosis research, pending prospective, multi-institution validation.
SN - 1471-2466
ER -