论文标题

通过使用深度学习来提高质子剂量计算精度

Improving Proton Dose Calculation Accuracy by Using Deep Learning

论文作者

Wu, Chao, Nguyen, Dan, Xing, Yixun, Montero, Ana Barragan, Schuemann, Jan, Shang, Haijiao, Pu, Yuehu, Jiang, Steve

论文摘要

准确的剂量计算对于质子治疗至关重要。基于铅笔梁(PB)模型的剂量计算很快,但由于处理不均匀性时的近似值而不准确。蒙特卡洛(MC)剂量计算是最准确的方法,但耗时。我们假设深度学习方法可以将PB剂量计算的准确性提高到MC的水平。在这项工作中,我们开发了一个深度学习模型,该模型将PB转换为不同肿瘤部位的MC剂量。所提出的模型基于我们新开发的层次连接的U-NET(HD U-NET)网络,它使用PB剂量和患者CT图像作为输入来生成MC剂量。我们使用了290例患者(90例头颈,93例肝脏,75例,前列腺癌和32例患有肺癌)来训练,验证和测试模型。对于每个肿瘤部位,我们进行了四个数值实验,以探索训练数据集的各种组合。将模型一起训练来自所有肿瘤部位的数据,并使用每个梁的剂量分布作为输入,在所有四个肿瘤部位都产生了最佳性能。转化剂量和MC剂量之间的平均伽马指数(1mm/1%标准)分别为92.8%,92.7%,89.7%和99.6%,分别为99.7%和99.6%。单场剂量转化的平均时间不到4秒。总之,我们基于深度学习的方法可以迅速提高质子PB剂量分布的准确性,而MC剂量分布的准确性。通过转移学习,可以轻松地将受过训练的模型适应不同肿瘤部位的新数据集和不同医院。可以将该模型添加为质子治疗计划的临床工作流程的插件,以提高质子剂量计算的准确性。

Accurate dose calculation is vitally important for proton therapy. Pencil beam (PB) model-based dose calculation is fast but inaccurate due to the approximation when dealing with inhomogeneities. Monte Carlo (MC) dose calculation is the most accurate method, but it is time consuming. We hypothesize that deep learning methods can boost the accuracy of PB dose calculation to the level of MC. In this work, we developed a deep learning model that converts PB to MC doses for different tumor sites. The proposed model is based on our newly developed hierarchically densely connected U-Net (HD U-Net) network, and it uses the PB dose and patient CT image as inputs to generate the MC dose. We used 290 patients (90 with head and neck, 93 with liver, 75 with prostate, and 32 with lung cancer) 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 index (1mm/1% criteria) between the converted dose 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 time for dose conversion for a single field was less than 4 seconds. In conclusion, our deep learning-based approach can quickly boost the accuracy of proton PB dose distributions to that of MC dose distributions. The trained model can be readily adapted to new datasets for different tumor sites and from different hospitals through transfer learning. This model can be added as a plug-in to the clinical workflow of proton therapy treatment planning to improve the accuracy of proton dose calculation.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源