arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2207.03414cs.CV
收藏

领域知识驱动的基于矩的损失函数三维剂量预测

Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Function

Memorial Sloan-Kettering Cancer Center(纪念斯隆-凯特琳癌症中心);Georgia Institute of Technology(佐治亚理工学院)

Gourav Jhanwar, Navdeep Dahiya, Parmida Ghahremani, Masoud Zarepisheh, Saad Nadeem

摘要:

剂量体积直方图(DVH)指标是临床上广泛接受的评价标准。然而,将这些度量纳入深度学习剂量预测模型是具有挑战性的,因为它们是非凸性和不可微的。我们提出了一种新的基于矩的损失函数,用于预测具有挑战性的传统肺调强放射治疗(IMRT)计划的三维剂量分布。基于矩的损失函数是凸的和可微的,并且可以在没有计算开销的情况下容易地将DVH度量并入任何深度学习框架中。还可以定制矩以反映3D剂量预测中的临床优先级。例如,使用高阶矩可以更好地预测连续结构的高剂量区域。我们使用了360名接受2Gy $\times$ 30次治疗的传统肺部患者的大型数据集,使用临床治疗计划训练深度学习(DL)模型。我们使用计算机断层扫描(CT)、计划靶体积(PTV)和器官危险轮廓(OAR)作为输入来训练类似UNet的CNN架构,以推断相应的逐体素3D剂量分布。我们评估了三种不同的损失函数:(1)平均绝对误差(MAE)损失,(2)MAE + DVH损失,和(3)建议的MAE +矩损失。预测的质量进行了比较,使用不同的DVH指标以及剂量分数和DVH分数,最近推出的AAPM知识为基础的规划大挑战。具有(MAE +矩)损失函数的模型通过显著提高DVH评分(11%,p $<$ 0.01)而优于具有MAE损失的模型,同时具有相似的计算成本。通过显著提高计算成本(48%)和DVH评分(8%,p $<$ 0.01),它也优于使用(MAE+DVH)训练的模型。代码、模型、Docker容器和Google Colab项目可以在我们的DoseRTX GitHub(https://github.com/nadeemlab/DoseRTX)上找到。

英文摘要:

Dose volume histogram (DVH) metrics are widely accepted evaluation criteria in the clinic. However, incorporating these metrics into deep learning dose prediction models is challenging due to their non-convexity and non-differentiability. We propose a novel moment-based loss function for predicting 3D dose distribution for the challenging conventional lung intensity modulated radiation therapy (IMRT) plans. The moment-based loss function is convex and differentiable and can easily incorporate DVH metrics in any deep learning framework without computational overhead. The moments can also be customized to reflect the clinical priorities in 3D dose prediction. For instance, using high-order moments allows better prediction in high-dose areas for serial structures. We used a large dataset of 360 conventional lung patients with 2Gy $\times$ 30 fractions to train the deep learning (DL) model using clinically treated plans. We trained a UNet-like CNN architecture using computed tomography (CT), planning target volume (PTV) and organ-at-risk contours (OAR) as input to infer corresponding voxel-wise 3D dose distribution. We evaluated three different loss functions: (1) Mean Absolute Error (MAE) Loss, (2) MAE + DVH Loss, and (3) the proposed MAE + Moments Loss. The quality of the predictions was compared using different DVH metrics as well as dose-score and DVH-score, recently introduced by the AAPM knowledge-based planning grand challenge. Model with (MAE + Moment) loss function outperformed the model with MAE loss by significantly improving the DVH-score (11%, p$<$0.01) while having similar computational cost. It also outperformed the model trained with (MAE+DVH) by significantly improving the computational cost (48%) and the DVH-score (8%, p$<$0.01). The code, models, docker container, and Google Colab project are available on our DoseRTX GitHub (https://github.com/nadeemlab/DoseRTX).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑