Study of three-dimensional dose distribution prediction in cervical cancer brachytherapy based on U-Net fully convolutional network / 中华放射肿瘤学杂志

ABSTRACT

Objective:Topredict the three-dimensional dose distribution of regions of interest (ROI) with brachytherapy for cervical cancer based on U-Net fully convolutional network, and evaluate the accuracy of prediction model.Methods:First, 100 cases of cervical cancer intracavity combined with interstitial implantation were selected as the entire research data set, and divided into the training set ( n=72), validation set ( n=8), and test set ( n=20). Then the U-Net was used to construct two models based on whether the uterine tandem and the implantation needles were included as the distinguishing factors. Finally, dose distribution of 20 cases in the test set were predicted using the trained model, and comparative analysis was performed. The performance of the model was jointly evaluated by , and the mean absolute deviation (MAD). Results:Compared with the model without the uterine tandem and the implantation needles, the of the rectum was increased by (16.83±1.82) cGy ( P<0.05), and the or of the other ROI were not different significantly (all P>0.05). The MAD of the high-risk clinical target volume, rectum, sigmoid, small bowel, and bladder was increased by (11.96±3.78) cGy, (11.43±0.54) cGy, (24.08±1.65) cGy, (17.04±7.17) cGy and (9.52±4.35) cGy, respectively (all P<0.05). The MAD of the intermediate-risk clinical target volume was decreased by (120.85±29.78) cGy ( P<0.05). The mean value of MAD for all ROI was decreased by (7.8±53) cGy ( P<0.05), which was closer to the actual plan. Conclusions:U-Net fully convolutional network can be used to predict three-dimensional dose distribution of patients with cervical cancer undergoing brachytherapy. Combining the uterine tube with the implantation needles as the input parameters yields more accurate predictions than a single use of the ROI structure as the input.