Patient-specific three-dimensional dose distribution prediction via deep learning for prostate cancer therapy: Improvement with the structure loss

Abstract

PurposeDeep learning (DL)-based dose distribution prediction can potentially reduce the cost of inverse planning process. We developed and introduced a structure-focused loss (Lstruct) for 3D dose prediction to improve prediction accuracy. This study investigated the influence of Lstruct on DL-based dose prediction for patients with prostate cancer. The proposed Lstruct, which is similar in concept to dose–volume histogram (DVH)-based optimization in clinical practice, has the potential to provide more interpretable and accurate DL-based optimization. MethodsThis study involved 104 patients who underwent prostate radiotherapy. We used 3D U-Net-based architecture to predict dose distributions from computed tomography and contours of the planning target volume and organs-at-risk. We trained two models using different loss functions: L2 loss and Lstruct. Predicted doses were compared in terms of dose–volume parameters and the Dice similarity coefficient of isodose volume. ResultsDVH analysis showed that the Lstruct model had smaller errors from the ground truth than the L2 model. The Lstruct model achieved more consistent dose distributions than the L2 model, with errors close to zero. The isodose Dice score of the Lstruct model was greater than that of the L2 model by >20% of the prescribed dose. ConclusionsWe developed Lstruct using labels of inputted contours for DL-based dose prediction for prostate radiotherapy. Lstruct can be generalized to any DL architecture, thereby enhancing the dose prediction accuracy.