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.

Data-driven dose calculation algorithm based on deep learning

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

A Feasibility Study on Deep Learning-Based Radiotherapy Dose Calculation

Technical Note: A feasibility study on deep learning-based radiotherapy dose calculation

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

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

Boosting radiotherapy dose calculation accuracy with deep learning

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

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

Fast dose calculation in x-ray guided interventions by using deep learning

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

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

A preliminary study of a photon dose calculation algorithm using a convolutional neural network

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

Single energy CT-based mass density and relative stopping power estimation for proton therapy using deep learning method

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

A deep learning approach to radiation dose estimation

Convolutional neural network‐based dosimetry evaluation of esophageal radiation treatment planning

Deep learning based linear energy transfer calculation for proton therapy

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

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