YOLOv8-AM: YOLOv8 Based on Effective Attention Mechanisms for Pediatric Wrist Fracture Detection
學年 113
學期 2
出版(發表)日期 2025-03-10
作品名稱 YOLOv8-AM: YOLOv8 Based on Effective Attention Mechanisms for Pediatric Wrist Fracture Detection
作品名稱(其他語言)
著者 [1] Chun-Tse Chien, Rui-Yang Ju, Kuang-Yi Chou, and Jen-Shiun Chiang
單位
出版者
著錄名稱、卷期、頁數 IEEE Access, vol. 13, pp. 52461 – 52477
摘要 Wrist trauma and even fractures occur frequently in daily life, particularly among children who account for a significant proportion of fracture cases. Before performing surgery, surgeons often request patients to undergo X-ray imaging first and prepare for surgery based on the analysis of the radiologists. With the development of neural networks, You Only Look Once (YOLO) series models have been widely used in fracture detection as computer-assisted diagnosis (CAD) tools. Ultralytics presented the latest version of the YOLO models in 2023, which has been employed for detecting fractures across various parts of the body. Attention mechanism is one of the most popular methods to improve the model performance. This work proposes YOLOv8-AM, which incorporates the attention modules into the YOLOv8 architecture. Specifically, we respectively employ four different attention modules, ResBlock with Convolutional Block Attention Module (ResCBAM), Shuffle Attention (SA), Efficient Channel Attention (ECA), and ResBlock with Global Attention Mechanism (ResGAM), to improve the model architecture, and train these models on the GRAZPEDWRI-DX dataset. Experimental results demonstrate that the mean Average Precision at IoU 50 (mAP@50) of one of the variants of YOLOv8-AM model (i.e., YOLOv8+ResCBAM) increased from 63.6% to 65.8%, which achieves the state-of-the-art (SOTA) performance. In addition, the proposed YOLOv8-AM models can detect the single-label category “fracture” with mAP@50 value of 95.7% in pediatric wrist trauma X-ray images. The implementation code for this work is available on GitHub at https://github.com/RuiyangJu/Fracture_Detection_Improved_YOLOv8
關鍵字 Computer vision, deep learning, fracture detection, medical image diagnostics, medical image processing, object detection, radiology, X-ray imaging, you only look once (YOLO)
語言 en_US
ISSN
期刊性質 國內
收錄於 SCI
產學合作
通訊作者 Jen-Shiun Chiang
審稿制度 1
國別 USA
公開徵稿
出版型式 ,電子版
SDGS 優質教育