A deep learning-based dose calculation method for volumetric modulated arc therapy

Volumetric modulated arc therapy (VMAT) planning optimization involves iterative adjustment of numerous parameters, and hence requires repeatedly dose recalculation. In this study, we used the deep learning method to develop a fast and accurate dose calculation method for VMAT.

The classical 3D UNet was adopted and trained to learn the physics principle of dose calculation. The inputs included the projected fluence map (FM), computed tomography (CT) images, the radiological depth and the source-to-voxel distance (SVD). The projected FM was generated by projecting the accumulated FM between two consecutive control points (CPs) onto the patient's anatomy. The accumulated FM was calculated by simulating the movement of the multi-leaf collimator (MLC) from one CP to the next. The dose, calculated by the treatment planning system (TPS), was used as ground truth. 51 head and neck VMAT plans were used, with 43, 1 and 7 cases as training, validation, and testing datasets, respectively. Correspondingly, 7182, 180 and 1260 CP samples were included in the training, validation, and testing datasets.

This presented method was evaluated by comparing the derived dose distribution to the TPS calculated dose distribution. The dose profiles coincided for both the single CP and the entire plan (summation of all CPs). But the network derived dose was smoother than the TPS calculated dose. Gamma analysis was performed between the network derived dose and the TPS calculated dose. The average gamma pass rate was 96.56%, 98.75%, 98.03% and 99.30% under the criteria of 2% (tolerance) -2 mm (distance to agreement, DTA). 2%-3 mm, 3%-2 mm and 3%-3 mm. No significant difference was observed on the critical indices including the max, mean dose, and the relative volume covered by the 2000 cGy, 4000 cGy and the prescription dose. For one CP, the average computational time of the network and TPS was 0.09s and 0.53s. And for one patient, the average time was 16.51s and 95.60s.

The dose distribution derived by the network showed good agreement with the TPS calculated dose distribution. The computational time was reduced to approximate one-sixth of its original duration. Therefore the presented deep learning-based dose calculation method has the potential to be used for planning optimization.