Dose prediction for cervical cancer in radiotherapy based on the beam channel generative adversarial network

Background: Existing deep learning methods, such as generative adversarial network (GAN) technology, face challenges when dealing with mixed datasets, which involve a combination of Intensity Modulated Radiotherapy (IMRT) and Volumetric Modulated Arc Therapy (VMAT). This issue significantly complicates the application of dose prediction in the field of radiotherapy. In this study, we propose a novel approach called beam channel GAN (Bc-GAN) to address the task of radiation dose prediction for mixed datasets. Bc-GAN introduces a dose prediction calculation method that requires less precision. By defining an approximate range for dose prediction, Bc-GAN limits the physical range of GAN prediction, resulting in more reasonable dose distribution predictions. Methods: We adopt a beam angle weighting method to determine the beam angle in the dose calculation. The dose of the beam with the highest weight is calculated using medical images and is then inputted into the artificial intelligence dose prediction model as the input channel. Additionally, we collect data from a total of 346 patients with Cervical Cancer (CC) for dataset. After cleaning the data, we exclude 51 cases with incomplete organ delineation, leaving us with 295 cases (IMRT: VMAT = 137:158) randomly divided into three sets: the training set, the validation set, and the test set, with proportions of 205:60:30, respectively. The assessment of model predictions was conducted via an analysis of dose distributions on the tomographic plane, dose volume histogram (DVH), and dosimetric parameters within the target zones and organs at risk (OAR). Results: After DVH analysis, minimal discrepancy was found between predicted and actual dose distributions in PTV and OAR. The predicted distribution aligned with clinical standards. Dosimetric parameters for PTV were generally lower in the predicted model, except for homogeneity index (HI) (0.238 ± 0.024, P = 0.017) and Dmax (53.599 ± 0.710 Gy, P = 1.8e-05). The prediction model varied in estimating doses for six organs. Specifically, small intestine showed higher V₂₀ (67.92 ± 51.64 %, P = 0.019) and V₃₀ (57.171 ± 1.213 %, P = 0.024) than manual planning. A similar trend was seen in colon's V₃₀ (37.13 ± 61.14 %, P = 0.016). However, predicted bladder V30 (87.51 ± 41.44 %, P = 2.03e-16) was lower, indicating significant dosimetric differences. Conclusion: Overall, this study presents an innovative prediction method for CC in radiotherapy using the Bc-GAN model, addressing the challenges posed by different radiotherapy techniques. The proposed approach allows IMRT and VMAT in radiotherapy to be used as training sets, enabling the potential for large-scale engineering and commercialization applications of artificial intelligence (AI). The Bc-GAN-based prediction method for CC in radiotherapy not only reduces the amount of data needed for the training set but also expedites the model generation process. This approach can be applied to guide the development of clinical radiation therapy plans. Furthermore, future studies should consider extending the dose prediction method to encompass other types of tumors.