DC-YOLOv5-based target detection algorithm for cervical vertebral maturation.

Abstract

The cervical vertebral maturation (CVM) method is essential to determine the timing of orthodontic and orthopedic treatment. In this paper, a target detection model called DC-YOLOv5 is proposed to achieve fully automated detection and staging of CVM. A total of 1800 cephalometric radiographs were labeled and categorized based on the CVM stages. We introduced a model named DC-YOLOv5, optimized for the specific characteristics of CVM based on YOLOv5. This optimization includes replacing the original bounding box regression loss calculation method with Wise-IOU to address the issue of mutual interference between vertical and horizontal losses in Complete-IOU (CIOU), which made model convergence challenging. We incorporated the Res-dcn-head module structure to enhance the focus on small target features, improving the model's sensitivity to subtle sample differences. Additionally, we introduced the Convolutional Block Attention Module (CBAM) dual-channel attention mechanism to enhance focus and understanding of critical features, thereby enhancing the accuracy and efficiency of target detection. Loss functions, precision, recall, mean average precision (mAP), and F1 scores were used as the main algorithm evaluation metrics to assess the performance of these models. Furthermore, we attempted to analyze regions important for model predictions using gradient Class Activation Mapping (CAM) techniques. The final F1 scores of the DC-YOLOv5 model for CVM identification were 0.993, 0.994 for mAp0.5 and 0.943 for mAp0.5:0.95, with faster convergence, more accurate and more robust detection than the other four models. The DC-YOLOv5 algorithm shows high accuracy and robustness in CVM identification, which provides strong support for fast and accurate CVM identification and has a positive effect on the development of medical field and clinical diagnosis.