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.

Predicting 3D dose distribution with scale attention network for prostate cancer radiotherapy

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

Explainable Attention Guided Adversarial Deep Network for 3D Radiotherapy Dose Distribution Prediction

A New Deep-Learning-based Model for Predicting 3D Radiotherapy Dose Distribution In Various Scenarios

DSANet: Dual-path Segmentation-guided Attention Network for Radiotherapy Dose Prediction from CT Images Only

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

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

ARANet: Attention-based Residual Adversarial Network with Deep Supervision for Radiotherapy Dose Prediction of Cervical Cancer

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

Voxel-Level Radiotherapy Dose Prediction Using Densely Connected Network with Dilated Convolutions.

Prostate Dose Prediction in HDR Brachytherapy Using Unsupervised Multi-Atlas Fusion.

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

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

Predicting voxel-level dose distributions for esophageal radiotherapy using densely connected network with dilated convolutions

Prediction of Three-Dimensional Radiotherapy Optimal Dose Distributions for Lung Cancer Patients with Asymmetric Network

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

Utilizing Pre-Determined Beam Orientation Information in Dose Prediction by 3D Fully-Connected Network for Intensity Modulated Radiotherapy.

Deep learning-based dose map prediction for high-dose-rate brachytherapy

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

Prediction of dose distributions for non‐small cell lung cancer patients using MHA‐ResUNet

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