Comparative analysis of nnU-Net and Auto3Dseg for fat and fibroglandular tissue segmentation in MRI.

Abstract

Purpose: Breast cancer, the most common cancer type among women worldwide, requires early detection and accurate diagnosis for improved treatment outcomes. Segmenting fat and fibroglandular tissue (FGT) in magnetic resonance imaging (MRI) is essential for creating volumetric models, enhancing surgical workflow, and improving clinical outcomes. Manual segmentation is time-consuming and subjective, prompting the development of automated deep-learning algorithms to perform this task. However, configuring these algorithms for 3D medical images is challenging due to variations in image features and preprocessing distortions. Automated machine learning (AutoML) frameworks automate model selection, hyperparameter tuning, and architecture optimization, offering a promising solution by reducing reliance on manual intervention and expert knowledge.

Approach: We compare nnU-Net and Auto3Dseg, two AutoML frameworks, in segmenting fat and FGT on T1-weighted MRI images from the Duke breast MRI dataset (100 patients). We used threefold cross-validation, employing the Dice similarity coefficient (DSC) and Hausdorff distance (HD) metrics for evaluation. The $F$ -test and Tukey honestly significant difference analysis were used to assess statistical differences across methods.

Results: nnU-Net achieved DSC scores of $0.946\pm 0.026$ (fat) and $0.872\pm 0.070$ (FGT), whereas Auto3DSeg achieved $0.940\pm 0.026$ (fat) and $0.871\pm 0.074$ (FGT). Significant differences in fat HD ( $F=6.3020$ , $p<0.001$ ) originated from the full resolution and the 3D cascade U-Net. No evidence of significant differences was found in FGT HD or DSC metrics.

Conclusions: Ensemble approaches of Auto3Dseg and nnU-Net demonstrated comparable performance in segmenting fat and FGT on breast MRI. The significant differences in fat HD underscore the importance of boundary-focused metrics in evaluating segmentation methods.