Abstract:Purpose: External radiation therapy planning is a highly complex and tedious process as it involves treating large target volumes, prescribing several levels of doses, as well as avoiding irradiating critical structures such as organs at risk close to the tumor target. This requires highly trained dosimetrists and physicists to generate a personalized plan and adapt it as treatment evolves, thus affecting the overall tumor control and patient outcomes. Our aim is to achieve accurate dose predictions for head and neck (H&N) cancer patients on a challenging in-house dataset that reflects realistic variability and to further compare and validate the method on a public dataset. Methods: We propose a three-dimensional (3D) deep neural network that combines a hierarchically dense architecture with an attention U-net (HDA U-net). We investigate a domain knowledge objective, incorporating a weighted mean squared error (MSE) with a dose-volume histogram (DVH) loss function. The proposed HDA U-net using the MSE-DVH loss function is compared with two state-of-the-art U-net variants on two radiotherapy datasets of H&N cases. These include reference dose plans, computed tomography (CT) information, organs at risk (OARs), and planning target volume (PTV) delineations. All models were evaluated using coverage, homogeneity, and conformity metrics as well as mean dose error and DVH curves. Results: Overall, the proposed architecture outperformed the comparative state-of-the-art methods, reaching 0.95 (0.98) on D95 coverage, 1.06 (1.07) on the maximum dose value, 0.10 (0.08) on homogeneity, 0.53 (0.79) on conformity index, and attaining the lowest mean dose error on PTVs of 1.7% (1.4%) for the in-house (public) dataset. The improvements are statistically significant ( p < 0.05 $p<0.05$ ) for the homogeneity and maximum dose value compared with the closest baseline. All models offer a near real-time prediction, measured between 0.43 and 0.88 s per volume. Conclusion: The proposed method achieved similar performance on both realistic in-house data and public data compared to the attention U-net with a DVH loss, and outperformed other methods such as HD U-net and HDA U-net with standard MSE losses. The use of the DVH objective for training showed consistent improvements to the baselines on most metrics, supporting its added benefit in H&N cancer cases. The quick prediction time of the proposed method allows for real-time applications, providing physicians a method to generate an objective end goal for the dosimetrist to use as reference for planning. This could considerably reduce the number of iterations between the two expert physicians thus reducing the overall treatment planning time.

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

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

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

Deep-learning-based Modulated Radiotherapy Dose Plan Prediction with Integration of Non-Modulated Dose Distribution

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

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

Deep learning prediction of scenario doses for direct plan robustness evaluations in IMPT for head-and-neck

Deep Learning Based Dosimetry Evaluation at Organs-at-Risk in Esophageal Radiation Treatment Planning.

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

Dose Prediction Using a Deep Neural Network for Accelerated Planning of Rectal Cancer Radiotherapy.

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

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

Domain Knowledge Driven 3D Dose Prediction Using Moment-Based Loss Function

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

Evaluation of deep learning based dose prediction in head and neck cancer patients using two different types of input contours

Data-driven dose calculation algorithm based on deep learning

Deep learning for high-resolution dose prediction in high dose rate brachytherapy for breast cancer treatment

Deep learning-based dose prediction for magnetic resonance-guided prostate radiotherapy

Dose Prediction with Deep Learning for Prostate Cancer Radiation Therapy: Model Adaptation to Different Treatment Planning Practices

Combining dense elements with attention mechanisms for 3D radiotherapy dose prediction on head and neck cancers

A Feasibility Study for Predicting 3D Radiotherapy Dose Distribution of Lung VMAT Patients