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.

Machine Learning to Generate Adjustable Dose Distributions in Head-and-Neck Cancer Radiation Therapy

Incorporating human and learned domain knowledge into training deep neural networks: A differentiable dose volume histogram and adversarial inspired framework for generating Pareto optimal dose distributions in radiation therapy

Dose distribution prediction for head-and-neck cancer radiotherapy using a generative adversarial network: influence of input data

Automatic Treatment Planning Based on Three-Dimensional Dose Distribution Predicted from Deep Learning Technique.

Advancing patient care: Machine learning models for predicting grade 3+ toxicities in gynecologic cancer patients treated with HDR brachytherapy

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

A Cascade Transformer-based Model for 3D Dose Distribution Prediction in Head and Neck Cancer Radiotherapy

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

Deep Learning-Based Dose Prediction for Automated, Individualized Quality Assurance of Head and Neck Radiation Therapy Plans

Surrogate-free machine learning-based organ dose reconstruction for pediatric abdominal radiotherapy

AS-NeSt: A Novel 3D Deep Learning Model for Radiation Therapy Dose Distribution Prediction in Esophageal Cancer Treatment with Multiple Prescriptions

3D dose prediction for Gamma Knife radiosurgery using deep learning and data modification

Large‐language‐model empowered 3D dose prediction for intensity‐modulated radiotherapy

Predicting the 3-Dimensional Dose Distribution of Multilesion Lung Stereotactic Ablative Radiation Therapy With Generative Adversarial Networks

Deep Learning 3D Dose Prediction for Conventional Lung IMRT Using Consistent/Unbiased Automated Plans

Deep Learning-Based Statistical Robustness Evaluation of Intensity-Modulated Proton Therapy for Head and Neck Cancer

Generating Pareto Optimal Dose Distributions for Radiation Therapy Treatment Planning

A machine learning approach to using Quality-of-Life patient scores in guiding prostate radiation therapy dosing

Machine Learning for the Prediction of Organ Damage from Cancer Radiotherapy

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