A deep learning method for prediction of three-dimensional dose distribution of helical tomotherapy

Abstract

Purpose: To develop a deep learning method for prediction of three-dimensional (3D) voxel-by-voxel dose distributions of helical tomotherapy (HT).

Methods: Using previously treated HT plans as training data, a deep learning model named U-ResNet-D was trained to predict a 3D dose distribution. First, the contoured structures and dose volumes were converted from plan database to 3D matrix with a program based on a developed visualization toolkit (VTK), then transferred to U-ResNet-D for correlating anatomical features and dose distributions at voxel-level. One hundred and ninety nasopharyngeal cancer (NPC) patients treated by HT with multiple planning target volumes (PTVs) in different prescription patterns were studied. The model was typically trained from scratch with weights randomly initialized rather than using transfer-learning method, and used to predict new patient's 3D dose distributions. The predictive accuracy was evaluated with three methods: (a) The dose difference at the position r, δ(r, r) = D_c (r) - D_p (r), was calculated for each voxel. The mean (μ_δ(r,r) ) and standard deviation (σ_δ(r,r) ) of δ(r, r) were calculated to assess the prediction bias and precision; (b) The mean absolute differences of dosimetric indexes (DIs) including maximum and mean dose, homogeneity index, conformity index, and dose spillage for PTVs and organ at risks (OARs) were calculated and statistically analyzed with the paired-samples t test; (c) Dice similarity coefficients (DSC) between predicted and clinical isodose volumes were calculated.

Results: The U-ResNet-D model predicted 3D dose distribution accurately. For twenty tested patients, the prediction bias ranged from -2.0% to 2.3% and prediction error varied from 1.5% to 4.5% (relative to prescription) for 3D dose differences. The mean absolute dose differences for PTVs and OARs are within 2.0% and 4.2%, and nearly all the DIs for PTVs and OARs had no significant differences. The averaged DSC ranged from 0.95 to 1 for different isodose volumes.

Conclusions: The study developed a new deep learning method for 3D voxel-by-voxel dose prediction, and shown to be able to produce accurately dose predictions for nasopharyngeal patients treated by HT. The predicted 3D dose map can be useful for improving radiotherapy planning design, ensuring plan quality and consistency, making clinical technique comparison, and guiding automatic treatment planning.

Keywords: 3D dose prediction; deep learning; helical tomotherapy; nasopharyngeal cancers.