Sliding transformer with uncertainty estimation for vestibular schwannoma automatic segmentation

Abstract

Abstract Objective. Automated segmentation of vestibular schwannoma (VS) using magnetic resonance imaging (MRI) can enhance clinical efficiency. Though many advanced methods exist for automated VS segmentation, the accuracy is hindered by ambivalent tumor borders and cystic regions in some patients. In addition, these methods provide results that do not indicate segmentation uncertainty, making their translation into clinical workflows difficult due to potential errors. Providing a definitive segmentation result along with segmentation uncertainty or self-confidence is crucial for the conversion of automated segmentation programs to clinical aid diagnostic tools. Approach. To address these issues, we propose a U-shaped cascade transformer structure with a sliding window that utilizes multiple sliding samples, a segmentation head, and an uncertainty head to obtain both the segmentation mask and uncertainty map. We collected multimodal MRI data from 60 clinical patients with VS from Xuanwu Hospital. Each patient case includes T1-weighted images, contrast-enhanced T1-weighted images, T2-weighted images, and a tumor mask. The images exhibit an in-plane resolution ranging from 0.70 × 0.70 to 0.76 × 0.76 mm, an in-plane matrix spanning from 216 × 256 to 284 × 256, a slice thickness varying between 0.50 and 0.80 mm, and a range of slice numbers from 72 to 120. Main results. Extensive experimental results show that our method achieves comparable or higher results than previous state-of-the-art brain tumor segmentation methods. On our collected multimodal MRI dataset of clinical VS, our method achieved the dice similarity coefficient (DSC) of 96.08% ± 1.30. On a publicly available VS dataset, our method achieved the mean DSC of 94.23% ± 2.53. Significance. The method efficiently solves the VS segmentation task while providing an uncertainty map of the segmentation results, which helps clinical experts review the segmentation results more efficiently and helps to transform the automated segmentation program into a clinical aid diagnostic tool.