论文标题

通过贝叶斯优化的高维自动放射治疗治疗计划

High-dimensional Automated Radiation Therapy Treatment Planning via Bayesian Optimization

论文作者

Wang, Qingying, Wang, Ruoxi, Liu, Jiacheng, Jiang, Fan, Yue, Haizhen, Du, Yi, Wu, Hao

论文摘要

放射疗法治疗计划可以看作是迭代的超参数调谐过程,以平衡相互冲突的临床目标。在这项工作中,我们研究了现代贝叶斯优化方法(BO)方法在高维环境中的自动化治疗计划问题。回顾性地选择了20例接受强度调节放射治疗(IMRT)治疗的局部晚期直肠癌患者。我们实施了一个自动治疗计划框架,该框架同时调整了剂量目标和权重,并测试了两种BO方法在治疗计划任务上的性能:一种标准BO方法(GPEI)和一种专用于高维问题(SaaS-BO)的BO方法。随机调整方法还包括作为基线。我们比较并分析了三种自动化方法的计划质量和计划效率以及两种BO方法的不同搜索模式。对于目标结构,SaaS-BO计划实现了可比的热点控制($ p = 0.43 $)和同质性($ p = 0.96 $),其临床计划明显好于GPEI和随机计划($ P <0.05 $)。 SaaS-BO和GPEI的计划都大大优于临床计划,并溢出了剂量($ p <0.05 $)。与临床计划相比,三种自动化方法制定的治疗计划均减少了股头和膀胱的评估剂量学指数。对基本预测模型的分析表明,这两个BO程序都已经确定了相似的重要计划参数。这项工作实施了基于BO的高参数调谐框架,用于自动化治疗计划。两种测试的BO方法都能够制定高质量的治疗计划,模型分析还证实了经过测试的治疗计划问题的内在低维度。

Radiation therapy treatment planning can be viewed as an iterative hyperparameter tuning process to balance conflicting clinical goals. In this work, we investigated the performance of modern Bayesian Optimization (BO) methods on automated treatment planning problems in high-dimensional settings. 20 locally advanced rectal cancer patients treated with intensity-modulated radiation therapy (IMRT) were retrospectively selected as test cases. We implemented an automated treatment planning framework that adjusts dose objectives and weights simultaneously, and tested the performance of two BO methods on the treatment planning task: one standard BO method (GPEI) and one BO method dedicated to high-dimensional problems (SAAS-BO). A random tuning method was also included as the baseline. We compared and analyzed the three automated methods' plan quality and planning efficiency and the different search patterns of the two BO methods. For the target structures, the SAAS-BO plans achieved comparable hot spot control ($p=0.43$) and homogeneity ($p=0.96$) with the clinical plans, significantly better than the GPEI and random plans ($p<0.05$). Both SAAS-BO and GPEI plans significantly outperformed the clinical plans in conformity and dose spillage ($p<0.05$). Compared with clinical plans, the treatment plans generated by the three automated methods all made reductions in evaluated dosimetric indices for the femoral head and the bladder. The analysis of the underlying predictive models has shown that both BO procedures have identified similar important planning parameters. This work implemented a BO-based hyperparameter tuning framework for automated treatment planning. Both tested BO methods were able to produce high-quality treatment plans and the model analysis also confirmed the intrinsic low dimensionality of the tested treatment planning problems.

扫码加入交流群

加入微信交流群

微信交流群二维码

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