Improving Accuracy of Lung Nodule Classification Using Deep Learning with Focal Loss

Area of Science:

• Medical Imaging
• Artificial Intelligence
• Oncology

Background:

• Early detection of pulmonary nodules is crucial for reducing lung cancer mortality.
• Computer-aided diagnosis (CAD) systems can assist in identifying and classifying these nodules.
• Existing methods may face challenges in achieving high classification accuracy.

Purpose of the Study:

• To propose a novel deep learning method for improved classification of pulmonary nodules in CT scans.
• To enhance the accuracy of distinguishing between nodules and non-nodules using artificial intelligence.
• To contribute to more effective early lung cancer diagnosis.

Main Methods:

• Development of a 15-layer 2D deep convolutional neural network (CNN) architecture.
• Automatic feature extraction and classification of pulmonary candidates.
• Application of a focal loss function during the training process to improve model performance.
• Evaluation on the LIDC/IDRI dataset from the LUNA16 challenge.

Main Results:

• The proposed deep learning method achieved high classification accuracy (97.2%).
• The model demonstrated excellent sensitivity (96.0%) and specificity (97.3%).
• The focal loss function effectively boosted the classification performance.

Conclusions:

• The developed deep learning approach with focal loss is a high-quality classifier for pulmonary nodules.
• This method shows significant potential for improving computer-aided diagnosis in lung cancer screening.
• Accurate nodule classification can lead to earlier detection and better patient outcomes.