Scientific article

Hierarchical approach for pulmonary-nodule identification from CT images using YOLO model and a 3D neural network classifier

Published inRadiological physics and technology, vol. 17, no. 1, p. 124-134
First online date2023-11-18

This study aimed to assist doctors in detecting early-stage lung cancer. To achieve this, a hierarchical system that can detect nodules in the lungs using computed tomography (CT) images was developed. In the initial phase, a preexisting model (YOLOv5s) was used to detect lung nodules. A 0.3 confidence threshold was established for identifying nodules in this phase to enhance the model's sensitivity. The primary objective of the hierarchical model was to locate and categorize all lung nodules while minimizing the false-negative rate. Following the analysis of the results from the first phase, a novel 3D convolutional neural network (CNN) classifier was developed to examine and categorize the potential nodules detected by the YOLOv5s model. The objective was to create a detection framework characterized by an extremely low false positive rate and high accuracy. The Lung Nodule Analysis 2016 (LUNA 16) dataset was used to evaluate the effectiveness of this framework. This dataset comprises 888 CT scans that include the positions of 1186 nodules and 400,000 non-nodular regions in the lungs. The YOLOv5s technique yielded numerous incorrect detections owing to its low confidence level. Nevertheless, the addition of a 3D classification system significantly enhanced the precision of nodule identification. By integrating the outcomes of the YOLOv5s approach using a 30% confidence limit and the 3D CNN classification model, the overall system achieved 98.4% nodule detection accuracy and an area under the curve of 98.9%. Despite producing some false negatives and false positives, the suggested method for identifying lung nodules from CT scans is promising as a valuable aid in decision-making for nodule detection.

  • CT scan
  • Classification
  • Lung cancer
  • Object detection
  • Pulmonary nodules
Citation (ISO format)
AHMADYAR, Yashar et al. Hierarchical approach for pulmonary-nodule identification from CT images using YOLO model and a 3D neural network classifier. In: Radiological physics and technology, 2023, vol. 17, n° 1, p. 124–134. doi: 10.1007/s12194-023-00756-9
Main files (1)
Article (Published version)
ISSN of the journal1865-0333

Technical informations

Creation11/19/2023 10:03:10 AM
First validation02/20/2024 4:14:38 PM
Update time03/04/2024 2:24:34 PM
Status update03/04/2024 2:24:34 PM
Last indexation05/06/2024 5:58:14 PM
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack