Deep learning for automated segmentation of pelvic muscles, fat, and bone from CT studies for body composition assessment

OBJECTIVE: To develop a deep convolutional neural network (CNN) to automatically segment an axial CT image of the pelvis for body composition measures. We hypothesized that a deep CNN approach would achieve high accuracy when compared to manual segmentations as the reference standard.

MATERIALS AND METHODS: We manually segmented 200 axial CT images at the supra-acetabular level in 200 subjects, labeling background, subcutaneous adipose tissue (SAT), muscle, inter-muscular adipose tissue (IMAT), bone, and miscellaneous intra-pelvic content. The dataset was randomly divided into training (180/200) and test (20/200) datasets. Data augmentation was utilized to enlarge the training dataset and all images underwent preprocessing with histogram equalization. Our model was trained for 50 epochs using the U-Net architecture with batch size of 8, learning rate of 0.0001, Adadelta optimizer and a dropout of 0.20. The Dice (F1) score was used to assess similarity between the manual segmentations and the CNN predicted segmentations.

RESULTS: The CNN model with data augmentation of N = 3000 achieved accurate segmentation of body composition for all classes. The Dice scores were as follows: background (1.00), miscellaneous intra-pelvic content (0.98), SAT (0.97), muscle (0.95), IMAT (0.91), and bone (0.92). Mean time to automatically segment one CT image was 0.07 s (GPU) and 2.51 s (CPU).

CONCLUSIONS: Our CNN-based model enables accurate automated segmentation of multiple tissues on pelvic CT images, with promising implications for body composition studies.