Multi-objective deep learning for lung cancer detection in CT images: enhancements in tumor classification,...

This study aims to develop and evaluate an advanced deep learning framework for the detection, classification, and localization of lung tumors in computed tomography (CT) scan images.

The research utilized a dataset of 1608 CT scan images, including 623 cancerous and 985 non-cancerous cases, all carefully labeled for accurate tumor detection, classification (benign or malignant), and localization. The preprocessing involved optimizing window settings, adjusting slice thickness, and applying advanced data augmentation techniques to enhance the model's robustness and generalizability. The proposed model incorporated innovative components such as transformer-based attention layers, adaptive anchor-free mechanisms, and an improved feature pyramid network. These features enabled the model to efficiently handle detection, classification, and localization tasks. The dataset was split into 70% for training, 15% for validation, and 15% for testing. A multi-task loss function was used to balance the three objectives and optimize the model's performance. Evaluation metrics included mean average precision (mAP), intersection over union (IoU), accuracy, precision, and recall.

The proposed model demonstrated outstanding performance, achieving a mAP of 96.26%, IoU of 95.76%, precision of 98.11%, and recall of 98.83% on the test dataset. It outperformed existing models, including You Only Look Once (YOLO)v9 and YOLOv10, with YOLOv10 achieving a mAP of 95.23% and YOLOv9 achieving 95.70%. The proposed model showed faster convergence, better stability, and superior detection capabilities, particularly in localizing smaller tumors. Its multi-task learning framework significantly improved diagnostic accuracy and operational efficiency.

The proposed model offers a robust and scalable solution for lung cancer detection, providing real-time inference, multi-task learning, and high accuracy. It holds significant potential for clinical integration to improve diagnostic outcomes and patient care.