arXiv:2411.18767physics.med-phcs.CV2024-11被引 1

用多任务学习统一勾画器官和预测剂量,提升放疗自动化效率

Multi-Task Learning for Integrated Automated Contouring and Voxel-Based Dose Prediction in Radiotherapy

  • 通过多任务学习共享器官勾画与剂量预测的特征信息
  • 前列腺和头颈癌的剂量预测误差分别降低19.82%和16.33%
  • 在保持甚至提升勾画精度的同时优化剂量预测

基于深度学习的自动化器官勾画与放疗计划已证明可提升放疗效率与准确性。然而传统流程将勾画与计划分为独立步骤,且深度学习中两者也独立进行。本研究采用多任务学习(MTL)方法,无缝整合自动化勾画与体素级剂量预测任务。利用自研前列腺癌数据集与公开的头颈癌数据集OpenKBP进行训练。相比顺序执行的勾画与计划模型,所提方法使前列腺与头颈部位剂量体积直方图指标的平均绝对差分别降低19.82%和16.33%。同时,该MTL模型在保持或提升勾画准确性的前提下,显著改善了剂量预测性能:前列腺与头颈数据集的Dice分数分别达到0.824和0.716,优于基线模型的0.818和0.674。结果表明,该多任务框架具备推动高效精准自动化放疗计划发展的潜力。

原文摘要 · Abstract (English)

Deep learning-based automated contouring and treatment planning has been proven to improve the efficiency and accuracy of radiotherapy. However, conventional radiotherapy treatment planning process has the automated contouring and treatment planning as separate tasks. Moreover in deep learning (DL), the contouring and dose prediction tasks for automated treatment planning are done independently. In this study, we applied the multi-task learning (MTL) approach in order to seamlessly integrate automated contouring and voxel-based dose prediction tasks, as MTL can leverage common information between the two tasks and be able able to increase the efficiency of the automated tasks. We developed our MTL framework using the two datasets: in-house prostate cancer dataset and the publicly available head and neck cancer dataset, OpenKBP. Compared to the sequential DL contouring and treatment planning tasks, our proposed method using MTL improved the mean absolute difference of dose volume histogram metrics of prostate and head and neck sites by 19.82% and 16.33%, respectively. Our MTL model for automated contouring and dose prediction tasks demonstrated enhanced dose prediction performance while maintaining or sometimes even improving the contouring accuracy. Compared to the baseline automated contouring model with the dice score coefficients of 0.818 for prostate and 0.674 for head and neck datasets, our MTL approach achieved average scores of 0.824 and 0.716 for these datasets, respectively. Our study highlights the potential of the proposed automated contouring and planning using MTL to support the development of efficient and accurate automated treatment planning for radiotherapy.

放疗自动化多任务学习剂量预测器官勾画

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。