TY - JOUR
T1 - Improving proton dose calculation accuracy by using deep learning
AU - Wu, Chao
AU - Nguyen, Dan
AU - Xing, Yixun
AU - Montero, Ana Barragan
AU - Schuemann, Jan
AU - Shang, Haijiao
AU - Pu, Yuehu
AU - Jiang, Steve
N1 - Publisher Copyright:
© 2021 The Author(s). Published by IOP Publishing Ltd
PY - 2021/3
Y1 - 2021/3
N2 - Pencil beam (PB) dose calculation is fast but inaccurate due to the approximations when dealing with inhomogeneities. Monte Carlo (MC) dose calculation is the most accurate method but it is time consuming. The aim of this study was to develop a deep learning model that can boost the accuracy of PB dose calculation to the level of MC dose by converting PB dose to MC dose for different tumor sites. The proposed model uses the PB dose and computed tomography image as inputs to generate the MC dose. We used 290 patients (90 head and neck, 93 liver, 75 prostate and 32 lung) to train, validate, and test the model. For each tumor site, we performed four numerical experiments to explore various combinations of training datasets. Training the model on data from all tumor sites together and using the dose distribution of each individual beam as input yielded the best performance for all four tumor sites. The average gamma passing rate (1 mm/1%) between the converted and the MC dose was 92.8%, 92.7%, 89.7% and 99.6% for head and neck, liver, lung, and prostate test patients, respectively. The average dose conversion time for a single field was less than 4 s. The trained model can be adapted to new datasets through transfer learning. Our deep learning-based approach can quickly boost the accuracy of PB dose to that of MC dose. The developed model can be added to the clinical workflow of proton treatment planning to improve dose calculation accuracy.
AB - Pencil beam (PB) dose calculation is fast but inaccurate due to the approximations when dealing with inhomogeneities. Monte Carlo (MC) dose calculation is the most accurate method but it is time consuming. The aim of this study was to develop a deep learning model that can boost the accuracy of PB dose calculation to the level of MC dose by converting PB dose to MC dose for different tumor sites. The proposed model uses the PB dose and computed tomography image as inputs to generate the MC dose. We used 290 patients (90 head and neck, 93 liver, 75 prostate and 32 lung) to train, validate, and test the model. For each tumor site, we performed four numerical experiments to explore various combinations of training datasets. Training the model on data from all tumor sites together and using the dose distribution of each individual beam as input yielded the best performance for all four tumor sites. The average gamma passing rate (1 mm/1%) between the converted and the MC dose was 92.8%, 92.7%, 89.7% and 99.6% for head and neck, liver, lung, and prostate test patients, respectively. The average dose conversion time for a single field was less than 4 s. The trained model can be adapted to new datasets through transfer learning. Our deep learning-based approach can quickly boost the accuracy of PB dose to that of MC dose. The developed model can be added to the clinical workflow of proton treatment planning to improve dose calculation accuracy.
KW - Deep learning
KW - Monte Carlo
KW - Pencil beam
KW - Proton dose calculation
UR - http://www.scopus.com/inward/record.url?scp=85104809054&partnerID=8YFLogxK
UR - http://www.scopus.com/inward/citedby.url?scp=85104809054&partnerID=8YFLogxK
U2 - 10.1088/2632-2153/abb6d5
DO - 10.1088/2632-2153/abb6d5
M3 - Article
C2 - 35965743
AN - SCOPUS:85104809054
SN - 2632-2153
VL - 2
JO - Machine Learning: Science and Technology
JF - Machine Learning: Science and Technology
IS - 1
M1 - 015017
ER -