Improved YOLO-Based Pulmonary Nodule Detection with Spatial-SE Attention and an Aspect Ratio Penalty

• Published in: Sensors (Basel, Switzerland) Vol. 25; no. 14; p. 4245
• Main Authors: Song, Xinhang, Xie, Haoran, Gao, Tianding, Cheng, Nuo, Gou, Jianping
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 08.07.2025; MDPI
• Subjects: Accuracy; Algorithms; aspect ratio penalty; Boxes; channel attention; EAPIoU loss; Experiments; Fines & penalties; Humans; Localization; Lung Neoplasms - diagnosis; Lung Neoplasms - diagnostic imaging; Optimization; pulmonary nodule detection; Solitary Pulmonary Nodule - diagnosis; Solitary Pulmonary Nodule - diagnostic imaging; spatial attention; Tomography, X-Ray Computed - methods; YOLOV11; YOLOV11; pulmonary nodule detection; spatial attention; aspect ratio penalty; channel attention; EAPIoU loss
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s25144245

The accurate identification of pulmonary nodules is critical for the early diagnosis of lung diseases; however, this task remains challenging due to inadequate feature representation and limited localization sensitivity. Current methodologies often utilize channel attention mechanisms and intersection over union (IoU)-based loss functions. Yet, they frequently overlook spatial context and struggle to capture subtle variations in aspect ratios, which hinders their ability to detect small objects. In this study, we introduce an improved YOLOV11 framework that addresses these limitations through two primary components: a spatial squeeze-and-excitation (SSE) module that concurrently models channel-wise and spatial attention to enhance the discriminative features pertinent to nodules and explicit aspect ratio penalty IoU (EAPIoU) loss that imposes a direct penalty on the squared differences in aspect ratios to refine the bounding box regression process. Comprehensive experiments conducted on the LUNA16, LungCT, and Node21 datasets reveal that our approach achieves superior precision, recall, and mean average precision (mAP) across various IoU thresholds, surpassing previous state-of-the-art methods while maintaining computational efficiency. Specifically, the proposed SSE module achieves a precision of 0.781 on LUNA16, while the EAPIoU loss boosts mAP@50 to 92.4% on LungCT, outperforming mainstream attention mechanisms and IoU-based loss functions. These findings underscore the effectiveness of integrating spatially aware attention mechanisms with aspect ratio-sensitive loss functions for robust nodule detection.