Attention 3D UNET for dose distribution prediction of high-dose-rate brachytherapy of cervical cancer: Intracavitary applicators.

backgroundFormulating a clinically acceptable plan within the time-constrained clinical setting of brachytherapy poses challenges to clinicians. Deep learning based dose prediction methods have shown favorable solutions for enhancing efficiency, but development has primarily been on external beam radiation therapy. Thus, there is a need for translation to brachytherapy. PURPOSE: This study proposes a dose prediction model utilizing an attention-gating mechanism and a 3D UNET for cervical cancer high-dose-rate intracavitary brachytherapy treatment planning with tandem-and-ovoid/ring applicators.

methodsA multi-institutional data set consisting of 77 retrospective clinical brachytherapy plans was utilized in this study. The data were preprocessed and augmented to increase the number of plans to 252. A 3D UNET architecture with attention gates was constructed and trained for mapping the contour information to dose distribution. The trained model was evaluated on a testing data set using various metrics, including dose statistics and dose-volume indices. We also trained a baseline UNET model for a fair comparison.

resultsThe attention-gated 3D UNET model exhibited competitive accuracy in predicting dose distributions similar to the ground truth. The average values of the mean absolute errors were 0.46 ± 11.71 Gy (vs. 0.47 ± 9.16 Gy for a baseline UNET) in CTV

conclusionAttention-gated 3D UNET revealed a promising capability in predicting voxel-wise dose distributions compared to 3D UNET. This model could be deployed for clinical use to predict 3D dose distributions for near real-time decision-making before planning, quality assurance, and guiding future automated planning, making the current workflow more efficient.