Application of UNETR for automatic cochlear segmentation in temporal bone CTs

Abstract

ObjectiveTo investigate the feasibility of a deep learning method based on a UNETR model for fully automatic segmentation of the cochlea in temporal bone CT images. MethodsThe normal temporal bone CTs of 77 patients were used in 3D U-Net and UNETR model automatic cochlear segmentation. Tests were performed on two types of CT datasets and cochlear deformity datasets. ResultsThrough training the UNETR model, when batch_size=1, the Dice coefficient of the normal cochlear test set was 0.92, which was higher than that of the 3D U-Net model; on the GE 256 CT, SE-DS CT and Cochlear Deformity CT dataset tests, the Dice coefficients were 0.91, 0.93, 0 93, respectively. ConclusionAccording to the anatomical characteristics of the temporal bone, the use of the UNETR model can achieve fully automatic segmentation of the cochlea and obtain an accuracy close to manual segmentation. This method is feasible and has high accuracy.