Feasibility of applying deep learning methods as universal function approximators to accelerate and improve the accuracy and speed of dose field computation in radiotherapy was investigated. The principal objective was to develop and evaluate models capable of predicting a high-accuracy dose field from its low-accuracy. For photon irradiation, the Pencil Beam (PB) algorithm was employed for fast low-accuracy dose field calculation, while the Collapsed Cone Convolution (CCC) algorithm was used to obtain the reference high-accuracy dose field. For particle irradiation, both low- and high-accuracy dose field calculations were performed using Monte Carlo (MC) simulation, using 10000 and 1000000 particles for respective calculations of the input and the output (reference) dose fields.
Standard model architectures were the U-Net and a Cascaded 3D U-Net (C3D) model for
the MC based proton and CCC based photon dose field predictions, respectively. Several loss
functions were evaluated; namely, mean absolute error, shrinkage loss, perceptual loss, and
gradient loss in order to assess their individual impact on the accuracy of the predicted dose
fields. In addition to the standard architectures, an Attention U-Net (AU-Net) and a
Generative Adversarial Network (GAN), in which a U-Net served as both generator and
discriminator were employed for the MC based proton dose field prediction. A GAN was also
applied to CCC based photon dose field prediction, with the C3D model used as the generator. For MC based proton dose field prediction, the model architectures of lower complexity with fewer trainable parameters were adopted, as overfitting was observed despite extensive regularization and data augmentation.
The CCC based photon dose field prediction models were trained and evaluated on two datasets. An internal prostate cancer dataset and the open source OpenKBP dataset comprising head and neck cancer cases. The MC based proton dose field prediction models were evaluated exclusively on the internal prostate cancer dataset. Obtained results demonstrated that the C3D model successfully predicted CCC based photon dose fields with high accuracy with respect to the reference values. On the internal prostate cancer dataset, the model achieved a mean absolute error of 0,0753 Gy, an average gamma pass rate of 98,86% under the 1 % / 1 mm criterion, and 99,95% under the 2 % / 2 mm criterion. Performance was somewhat lower on the OpenKBP dataset, attributable to greater anatomical complexity of the head and neck cancer cases as compared to the prostate cancer cases, with a mean absolute error of 0,18 Gy and average gamma index pass rates of 90,5% and 96,54% under the 1 % / 1 mm and 2 % / 2 mm criteria, respectively. These were the best results and were achieved using the C3D model trained with the gradient loss function. The GAN model also produced dose field distributions closely resembling the CCC based photon dose field reference, though with lower accuracy than the C3D model trained with the gradient loss. On the OpenKBP dataset of head and neck cancer cases, gamma pass rates were 87,0% and 95,0% under the 1 % / 1 mm and 2 % / 2 mm criteria, respectively. The U-Net model used for MC based proton dose field prediction achieved a mean absolute error of 0,17 Gy, with gamma index pass rates of 90,0 % and 98,5 % under the 1 % / 1 mm and 2 % / 2 mm criteria.
The primary advantages of the proposed approaches include high predictive accuracy, short
computation times, and potential applicability in real time treatment planning workflows.
Limitations include the restricted size of the training datasets and the need for further validation on clinical data spanning diverse anatomical regions. Future work should consider incorporating larger and more heterogeneous datasets, extending the framework to diffusion and/or transformer based architectures, and exploring the integration of clinical objectives directly into the training process.
Validation results confirm that deep learning methods can reliably replace computationally
intensive physical dose calculation algorithms, enable fast dose field approximation, thereby
contributing to the automation and acceleration of radiotherapy treatment planning.
|