Wavelet Transform and Hierarchical Hybrid Matching for Enhancing End-to-End Pediatric Wrist Fracture Detection

Abstract

With the increasing frequency of daily physical activities among children and adolescents, the incidence of wrist fractures has been rising annually. Without precise and prompt diagnosis, these fractures may remain undetected, potentially leading to complications. Recent advancements in computer-aided diagnosis (CAD) technologies have facilitated the development of sophisticated diagnostic tools, which significantly improve the accuracy of fracture detection. To enhance the capability of detecting pediatric wrist fractures, this study presents the WH-DETR model, specifically designed for pediatric wrist fracture detection. WH-DETR is configured as a DEtection TRansformer framework, an end-to-end object detection algorithm that obviates the need for non-maximum suppression post-processing. To further enhance its performance, this study first introduces a wavelet transform projection module to capture different frequency features from the feature maps extracted by the backbone. This module allows the network to effectively capture multi-scale and multi-frequency information, improving the detection of subtle and complex features in medical images. Secondly, this study designs a hierarchical hybrid matching framework that decouples the prediction tasks of different decoder layers during training, thereby improving the final predictive capabilities of the model. The framework improves prediction robustness while maintaining inference efficiency. Extensive experiments on the GRAZPEDWRI-DX dataset demonstrate that our WH-DETR model achieves state-of-the-art performance with only 43 M parameters, attaining an <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"><mml:msub><mml:mtext>mAP</mml:mtext> <mml:mn>50</mml:mn></mml:msub> </mml:math> score of 68.8%, an <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"><mml:msub><mml:mtext>mAP</mml:mtext> <mml:mrow><mml:mn>50</mml:mn> <mml:mo>-</mml:mo> <mml:mn>90</mml:mn></mml:mrow> </mml:msub> </mml:math> score of 48.3%, and an F1 score of 64.1%. These results represent improvements of 1.78% in <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"><mml:msub><mml:mtext>mAP</mml:mtext> <mml:mn>50</mml:mn></mml:msub> </mml:math> , 1.69% in <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML"><mml:msub><mml:mtext>mAP</mml:mtext> <mml:mrow><mml:mn>50</mml:mn> <mml:mo>-</mml:mo> <mml:mn>90</mml:mn></mml:mrow> </mml:msub> </mml:math> , and 1.75% in F1 score, respectively, over the next best-performing model, highlighting its superior efficiency and robustness in pediatric wrist fracture detection.

Ähnliche Arbeiten