Abstract:Background: The quality of treatment plans for breast cancer can vary greatly. This variation could be reduced by using dose prediction to automate treatment planning. Our work investigates novel methods for training deep-learning models that are capable of producing high-quality dose predictions for breast cancer treatment planning. Purpose: The goal of this work was to compare the performance impact of two novel techniques for deep learning dose prediction models for tangent field treatments for breast cancer. The first technique, a "glowing" mask algorithm, encodes the distance from a contour into each voxel in a mask. The second, a gradient-weighted mean squared error (MSE) loss function, emphasizes the error in high-dose gradient regions in the predicted image. Methods: Four 3D U-Net deep learning models were trained using the planning CT and contours of the heart, lung, and tumor bed as inputs. The dataset consisted of 305 treatment plans split into 213/46/46 training/validation/test sets using a 70/15/15% split. We compared the impact of novel "glowing" anatomical mask inputs and a novel gradient-weighted MSE loss function to their standard counterparts, binary anatomical masks, and MSE loss, using an ablation study methodology. To assess performance, we examined the mean error and mean absolute error (ME/MAE) in dose across all within-body voxels, the error in mean dose to heart, ipsilateral lung, and tumor bed, dice similarity coefficient (DSC) across isodose volumes defined by 0%-100% prescribed dose thresholds, and gamma analysis (3%/3 mm). Results: The combination of novel glowing masks and gradient weighted loss function yielded the best-performing model in this study. This model resulted in a mean ME of 0.40%, MAE of 2.70%, an error in mean dose to heart and lung of -0.10 and 0.01 Gy, and an error in mean dose to the tumor bed of -0.01%. The median DSC at 50/95/100% isodose levels were 0.91/0.87/0.82. The mean 3D gamma pass rate (3%/3 mm) was 93%. Conclusions: This study found the combination of novel anatomical mask inputs and loss function for dose prediction resulted in superior performance to their standard counterparts. These results have important implications for the field of radiotherapy dose prediction, as the methods used here can be easily incorporated into many other dose prediction models for other treatment sites. Additionally, this dose prediction model for breast radiotherapy has sufficient performance to be used in an automated planning pipeline for tangent field radiotherapy and has the major benefit of not requiring a PTV for accurate dose prediction.

Multi-center dose prediction using attention-aware deep learning algorithm based on transformers for cervical cancer radiotherapy

Automated Clinical Target Volume Delineation Using Deep 3D Neural Networks in Radiation Therapy of Non-small Cell Lung Cancer

A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy

The Feasibility Study on the Generalization of Deep Learning Dose Prediction Model for Volumetric Modulated Arc Therapy of Cervical Cancer

Direct Dose Prediction With Deep Learning for Postoperative Cervical Cancer Underwent Volumetric Modulated Arc Therapy

TrDosePred: A deep learning dose prediction algorithm based on transformers for head and neck cancer radiotherapy

Clinical target volume (CTV) automatic delineation using deep learning network for cervical cancer radiotherapy: A study with external validation

Improvement of accumulated dose distribution in combined cervical cancer radiotherapy with deep learning-based dose prediction

Dose prediction via distance-guided deep learning: Initial development for nasopharyngeal carcinoma radiotherapy

Predicting voxel-level dose distributions of single-isocenter volumetric modulated arc therapy treatment plan for multiple brain metastases

Deep learning–based dose prediction to improve the plan quality of volumetric modulated arc therapy for gynecologic cancers

Unsupervised Domain Adaptive Dose Prediction Via Cross-Attention Transformer and Target-Specific Knowledge Preservation

A Feasibility Study on Deep Learning Based Individualized 3D Dose Distribution Prediction

Deep convolutional neural network with transfer learning for rectum toxicity prediction in cervical cancer radiotherapy: a feasibility study

Improving 3D dose prediction for breast radiotherapy using novel glowing masks and gradient-weighted loss functions

Transformer-Integrated Hybrid Convolutional Neural Network for Dose Prediction in Nasopharyngeal Carcinoma Radiotherapy

Exploring the impact of network depth on 3D U-Net-based dose prediction for cervical cancer radiotherapy

Dose Prediction Using a Three-Dimensional Convolutional Neural Network for Nasopharyngeal Carcinoma With Tomotherapy

A feasibility study for predicting optimal radiation therapy dose distributions of prostate cancer patients from patient anatomy using deep learning

Deep-Learning-Based Dose Predictor for Glioblastoma–Assessing the Sensitivity and Robustness for Dose Awareness in Contouring

Data-driven Dose Calculation Algorithm Based on Deep U-Net