A 3D multi-task network for the automatic segmentation of CT images featuring hip osteoarthritis.
Authors
Affiliations (3)
Affiliations (3)
- Southwest Jiaotong University School of Mechanical Engineering, No. 111, North Section 1, Second Ring Road, Jinniu District, Chengdu, Chengdu, Sichuan, 610031, CHINA.
- Southwest Jiaotong University School of Mechanical Engineering, No. 111, North Section 1, Second Ring Road, Jinniu District, Chengdu, Chengdu, 610031, CHINA.
- People's Liberation Army The General Hospital of Western Theater Command, No. 270, Tianhui Road, Rongdu Avenue, Jinniu District, Chengdu, Chengdu, Sichuan, 610083, CHINA.
Abstract
Total hip arthroplasty (THA) is the standard surgical treatment for end-stage hip osteoarthritis, with its success dependent on precise preoperative planning, which, in turn, relies on accurate three-dimensional segmentation and reconstruction of the periarticular bone of the hip joint. However, patients with hip osteoarthritis often exhibit pathological characteristics, such as joint space narrowing, femoroacetabular impingement, osteophyte formation, and joint deformity. These changes present significant challenges for traditional manual or semi-automatic segmentation methods. To address these challenges, this study proposed a novel 3D UNet-based multi-task network to achieve rapid and accurate segmentation and reconstruction of the periarticular bone in hip osteoarthritis patients. The bone segmentation main network incorporated the Transformer module during the encoder to effectively capture spatial anatomical features, while a boundary-optimization branch was designed to address segmentation challenges at the acetabular-femoral interface. These branches were jointly optimized through a multi-task loss function, with an oversampling strategy introduced to enhance the network's feature learning capability for complex structures. The experimental results showed that the proposed method achieved excellent performance on the test set with hip osteoarthritis. The average Dice coefficient was 96.09% (96.98% for femur, 95.20% for hip), with an overall precision of 96.66% and recall of 97.32%. In terms of the boundary matching metrics, the average surface distance (ASD) and the 95% Hausdorff distance (HD95) were 0.40 mm and 1.78 mm, respectively. The metrics showed that the proposed automatic segmentation network achieved high accuracy in segmenting the periarticular bone of the hip joint, generating reliable 2D masks and 3D models, thereby demonstrating significant potential for supporting THA surgical planning.