AFE-YOLO: An Image Detection Algorithm for Pediatric Wrist Fractures
摘要
To improve the accuracy of pediatric wrist fracture detection and reduce false positives and missed detections, this study proposes AFE-YOLO, an adaptive feature enhancement algorithm based on YOLOv8. First, receptive field attention convolution was incorporated into the backbone network to enhance feature perception, enabling the model to capture multi-scale fracture features more effectively and optimize computational efficiency. Second, an adaptive hybrid feature enhancement module was introduced in the neck network to improve the recognition of fine-grained lesion regions through feature fusion and semantic representation. Finally, a weighted loss fusion strategy was proposed, combining binary cross entropy and focal loss to address class imbalance and improve sensitivity to hard-to-classify samples. Experimental results demonstrate that AFE-YOLO improves average precision and recall by 4.2 and 7.3 percentage points, respectively, compared with the baseline model on the GRAZPEDWRI-DX dataset, while also reducing computational overhead.