Beam mask and sliding window-facilitated deep learning-based accurate and efficient dose prediction for pencil beam scanning proton therapy

BACKGROUND

Accurate and efficient dose calculation is essential for on-line adaptive planning in proton therapy. Deep learning (DL) has shown promising dose prediction results in photon therapy. However, there is a scarcity of DL-based dose prediction methods specifically designed for proton therapy. Successful dose prediction method for proton therapy should account for more challenging dose prediction problems in pencil beam scanning proton therapy (PBSPT) due to its sensitivity to heterogeneities.

PURPOSE

To develop a DL-based PBSPT dose prediction workflow with high accuracy and balanced complexity to support on-line adaptive proton therapy clinical decision and subsequent replanning.

METHODS

PBSPT plans of 103 prostate cancer patients (93 for training and the other 10 for independent testing) and 83 lung cancer patients (73 for training and the other 10 for independent testing) previously treated at our institution were included in the study, each with computed tomography scans (CTs), structure sets, and plan doses calculated by the in-house developed Monte-Carlo dose engine (considered as the ground truth in the model training and testing). For the ablation study, we designed three experiments corresponding to the following three methods: (1) Experiment 1, the conventional region of interest (ROI) (composed of targets and organs-at-risk [OARs]) method. (2) Experiment 2, the beam mask (generated by raytracing of proton beams) method to improve proton dose prediction. (3) Experiment 3, the sliding window method for the model to focus on local details to further improve proton dose prediction. A fully connected 3D-Unet was adopted as the backbone. Dose volume histogram (DVH) indices, 3D Gamma passing rates with a criterion of 3%/3 mm/10%, and dice coefficients for the structures enclosed by the iso-dose lines between the predicted and the ground truth doses were used as the evaluation metrics. The calculation time for each proton dose prediction was recorded to evaluate the method's efficiency.

RESULTS

Compared to the conventional ROI method, the beam mask method improved the agreement of DVH indices for both targets and OARs and the sliding window method further improved the agreement of the DVH indices (for lung cancer, CTV D98 absolute deviation: 0.74 ± 0.18 vs. 0.57 ± 0.21 vs. 0.54 ± 0.15 Gy[RBE], ROI vs. beam mask vs. sliding window methods, respectively). For the 3D Gamma passing rates in the target, OARs, and BODY (outside target and OARs), the beam mask method improved the passing rates in these regions and the sliding window method further improved them (for prostate cancer, targets: 96.93% ± 0.53% vs. 98.88% ± 0.49% vs. 99.97% ± 0.07%, BODY: 86.88% ± 0.74% vs. 93.21% ± 0.56% vs. 95.17% ± 0.59%). A similar trend was also observed for the dice coefficients. This trend was especially remarkable for relatively low prescription isodose lines (for lung cancer, 10% isodose line dice: 0.871 ± 0.027 vs. 0.911 ± 0.023 vs. 0.927 ± 0.017). The dose predictions for all the testing cases were completed within 0.25 s.

CONCLUSIONS

An accurate and efficient deep learning-augmented proton dose prediction framework has been developed for PBSPT, which can predict accurate dose distributions not only inside but also outside ROI efficiently. The framework can potentially further reduce the initial planning and adaptive replanning workload in PBSPT.

Virtual particle Monte Carlo: A new concept to avoid simulating secondary particles in proton therapy dose calculation

BACKGROUND

In proton therapy dose calculation, Monte Carlo (MC) simulations are superior in accuracy but more time consuming, compared to analytical calculations. Graphic processing units (GPUs) are effective in accelerating MC simulations but may suffer thread divergence and racing condition in GPU threads that degrades the computing performance due to the generation of secondary particles during nuclear reactions.

PURPOSE

A novel concept of virtual particle (VP) MC (VPMC) is proposed to avoid simulating secondary particles in GPU-accelerated proton MC dose calculation and take full advantage of the computing power of GPU.

METHODS

Neutrons and gamma rays were ignored as escaping from the human body; doses of electrons, heavy ions, and nuclear fragments were locally deposited; the tracks of deuterons were converted into tracks of protons. These particles, together with primary and secondary protons, are considered to be the realistic particles. Histories of primary and secondary protons were replaced by histories of multiple VPs. Each VP corresponded to one proton (either primary or secondary). A continuous-slowing-down-approximation model, an ionization model, and a large angle scattering event model corresponding to nuclear interactions were developed for VPs by generating probability distribution functions (PDFs) based on simulation results of realistic particles using MCsquare. For efficient calculations, these PDFs were stored in the Compute Unified Device Architecture textures. VPMC was benchmarked with TOPAS and MCsquare in phantoms and with MCsquare in 13 representative patient geometries. Comparisons between the VPMC calculated dose and dose measured in water during patient-specific quality assurance (PSQA) of the selected 13 patients were also carried out. Gamma analysis was used to compare the doses derived from different methods and calculation efficiencies were also compared.

RESULTS

Integrated depth dose and lateral dose profiles in both homogeneous and inhomogeneous phantoms all matched well among VPMC, TOPAS, and MCsquare calculations. The 3D-3D gamma passing rates with a criterion of 2%/2 mm and a threshold of 10% was 98.49% between MCsquare and TOPAS and 98.31% between VPMC and TOPAS in homogeneous phantoms, and 99.18% between MCsquare and TOPAS and 98.49% between VPMC and TOPAS in inhomogeneous phantoms, respectively. In patient geometries, the 3D-3D gamma passing rates with 2%/2 mm/10% between dose distributions from VPMC and MCsquare were 98.56 ± 1.09% in patient geometries. The 2D-3D gamma analysis with 3%/2 mm/10% between the VPMC calculated dose distributions and the 2D measured planar dose distributions during PSQA was 98.91 ± 0.88%. VPMC calculation was highly efficient and took 2.84 ± 2.44 s to finish for the selected 13 patients running on four NVIDIA Ampere GPUs in patient geometries.

CONCLUSION

VPMC was found to achieve high accuracy and efficiency in proton therapy dose calculation.