Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Function
- 1. Memorial Sloan Kettering Cancer Center
- 2. Georgia Tech
Description
We propose a novel moment-based loss function for predicting 3D dose distribution for the challenging conventional lung IMRT plans. The moment-based loss function is convex and differentiable and can easily incorporate clinical dose volume histogram (DVH) domain knowledge in any deep learning framework without computational overhead. A large dataset of 360 (240 for training, 50 for validation and 70 for testing) conventional lung patients with 2Gy × 30 fractions was used to train the deep learning (DL) model using clinically treated plans at our institution. A UNet-like CNN architecture was trained 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) The popular Mean Absolute Error (MAE) Loss, (2) the recently developed 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). DVH metrics are widely accepted evaluation criteria in the clinic. However, incorporating them into the 3D dose prediction model is challenging due to their non-convexity and non-differentiability. Moments provide a mathematically rigorous and computationally efficient way to incorporate DVH information in any deep learning architecture. The accompanying code can be found on our DoseRTX GitHub (https://github.com/nadeemlab/DoseRTX).
Files
Files
(555.9 MB)
Name | Size | Download all |
---|---|---|
md5:0cf12782b6fa1bb494b1c8b5b1e60d15
|
98.4 MB | Download |
md5:920e115febee791add5fdb81caff776b
|
98.4 MB | Download |
md5:4d03d4b0653e5960ef741400ed44b900
|
98.4 MB | Download |
md5:f01db3b8be00f815ccf673a17fabd899
|
157.4 MB | Download |
md5:51898b73ec99688b9a6ea44e1fa62edd
|
48.3 MB | Download |
md5:5ee14f96d82f367e45ef1162023158c6
|
25.9 MB | Download |
md5:7a99af00b828b98df9bcf7d76483f978
|
29.1 MB | Download |
Additional details
Related works
- Is cited by
- Journal article: 10.1088/1361-6560/ac8d45 (DOI)