Abstract:Lung cancer is a leading cause of cancer-related deaths worldwide, and early detection is crucial for improving patient outcomes. Nevertheless, early diagnosis of cancer is a major challenge, particularly in low-resource settings where access to medical resources and trained radiologists is limited. The objective of this study is to propose an automated end-to-end deep learning-based framework for the early detection and classification of lung nodules, specifically for low-resource settings. The proposed framework consists of three stages: lung segmentation using a modified 3D U-Net named 3D Res-U-Net, nodule detection using YOLO-v5, and classification with a Vision Transformer-based architecture. We evaluated the proposed framework on a publicly available dataset, LUNA16. The proposed framework's performance was measured using the respective domain's evaluation matrices. The proposed framework achieved a 98.82% lung segmentation dice score while detecting the lung nodule with 0.76 mAP@50 from the segmented lung, at a low false-positive rate. The performance of both networks of the proposed framework was compared with other studies and found to outperform them regarding segmentation and detection accuracy. Additionally, our proposed Vision transformer network obtained an accuracy of 93.57%, which is 1.21 higher than the state-of-the-art networks. Our proposed end-to-end deep learning-based framework can effectively segment lungs, and detect and classify lung nodules, specifically in low-resource settings with limited access to radiologists. The proposed framework outperforms existing studies regarding all the respective evaluation metrics. The proposed framework can potentially improve the accuracy and efficiency of lung cancer screening in low-resource settings, ultimately leading to better patient outcomes.

What problem does this paper attempt to address?

### Problems the Paper Attempts to Solve This paper aims to propose an automated end-to-end deep learning framework for the early detection and classification of lung nodules, particularly targeting resource-limited environments. Specifically, the paper attempts to address the following issues: 1. **Challenges of Early Lung Cancer Diagnosis**: Lung cancer is the leading cause of cancer-related deaths worldwide, and early diagnosis is crucial for improving patient prognosis. However, early diagnosis is particularly challenging in resource-limited environments due to the lack of medical resources and trained radiologists. 2. **Accuracy of Lung Image Segmentation**: Accurate detection of lung nodules requires precise segmentation of the lungs. Existing methods have limitations in segmentation accuracy, especially when dealing with complex lung structures. 3. **High Precision in Lung Nodule Detection**: Current lung nodule detection methods perform poorly in detecting small nodules and irregularly shaped nodules, leading to a high rate of missed detections. 4. **High Accuracy in Lung Nodule Classification**: Distinguishing between benign and malignant lung nodules is crucial for early lung cancer diagnosis. Existing classification methods still have room for improvement in accuracy. ### Main Contributions of the Framework 1. **Integration of Multi-Stage Models**: A cascaded three-stage framework (CTSF) is proposed, which reliably identifies and classifies lung nodules, reducing the burden on clinicians and lowering the misdiagnosis rate. 2. **Novel 3D Res-U-Net Algorithm**: A new 3D Res-U-Net algorithm is proposed, achieving state-of-the-art performance in lung segmentation tasks. 3. **High-Confidence YOLO-v5 Detection Algorithm**: The YOLO-v5 detection algorithm is utilized to detect lung nodules with high confidence. 4. **Vision Transformer-Based Classification Network**: A classification network based on vision transformers is proposed, effectively classifying malignant nodules. 5. **Experimental Results**: On the LUNA16 dataset, the framework achieved excellent performance metrics in lung segmentation, lung nodule detection, and classification tasks, with a Dice score of 89.5%, mAP@50 of 89%, and cancer classification accuracy of 86.75%, respectively. ### Conclusion The end-to-end deep learning framework proposed in this study demonstrates outstanding performance in lung segmentation, lung nodule detection, and classification tasks, with significant application prospects in resource-limited environments. This framework is expected to improve the accuracy and efficiency of lung cancer screening, ultimately enhancing patient prognosis. The study also shows that deep learning methods can be effectively applied to automated lung cancer detection and classification, even in environments with limited medical resources and professionals.

An automated end-to-end deep learning-based framework for lung cancer diagnosis by detecting and classifying the lung nodules

Lung cancer classification and identification framework with automatic nodule segmentation screening using machine learning

Automated Lung Nodule Detection and Classification Using Deep Learning Combined with Multiple Strategies

Lung Images Segmentation and Classification Based on Deep Learning: A New Automated CNN Approach

An efficient U-Net framework for lung nodule detection using densely connected dilated convolutions

Automated Pulmonary Nodule Detection via 3D ConvNets with Online Sample Filtering and Hybrid-Loss Residual Learning

A Hybrid deep learning model for effective segmentation and classification of lung nodules from CT images

A user-friendly deep learning application for accurate lung cancer diagnosis

An integrated method for detecting lung cancer via CT scanning via optimization, deep learning, and IoT data transmission

An Improved Convolution Neural Network and Modified Regularized K-Means-Based Automatic Lung Nodule Detection and Classification

Automated Pulmonary Nodule Detection in CT Images Using Deep Convolutional Neural Networks.

DeepLung: 3D Deep Convolutional Nets for Automated Pulmonary Nodule Detection and Classification

Cloud-Based Automated Clinical Decision Support System for Detection and Diagnosis of Lung Cancer in Chest CT

Early detection of lung cancer from CT images: nodule segmentation and classification using deep learning

A 3D Probabilistic Deep Learning System for Detection and Diagnosis of Lung Cancer Using Low-Dose CT Scans

Lung Nodule Detection For CT-Guided Biopsy Images Using Deep Learning

AWEU-Net: An Attention-Aware Weight Excitation U-Net for Lung Nodule Segmentation

Detection-Guided Deep Learning-Based Model with Spatial Regularization for Lung Nodule Segmentation

An End-to-End Framework for Integrated Pulmonary Nodule Detection and False Positive Reduction.

Improved UNet Deep Learning Model for Automatic Detection of Lung Cancer Nodules

VGG19 Network Assisted Joint Segmentation and Classification of Lung Nodules in CT Images