Technical Note: Dose prediction for head and neck radiotherapy using a three-dimensional dense dilated U-net architecture.

Abstract

Radiation therapy treatment planning is a time-consuming and iterative manual process. Consequently, plan quality varies greatly between and within institutions. Artificial intelligence shows great promise in improving plan quality and reducing planning times. This technical note describes our participation in the American Association of Physicists in Medicine Open Knowledge-Based Planning Challenge (OpenKBP), a competition to accurately predict radiation therapy dose distributions. A three-dimensional (3D) densely connected U-Net with dilated convolutions was developed to predict 3D dose distributions given contoured CT images of head and neck patients as input. While traditional augmentation techniques such as rotations and translations were explored, it was found that training on random patches alone resulted in the greatest model performance. A custom-weighted mean squared error loss function was employed. Finally, an ensemble of best-performing networks was used to generate the final challenge predictions. Our team (SuperPod) placed second in the dose stream of the OpenKBP challenge. The average mean absolute difference between the predicted and clinical dose distributions of the testing dataset was 2.56Gy. On average, the predicted normalized target DVH metrics were within 3% of the clinical plans, and the predicted organ at risk DVH metrics were within 2Gy of the clinical plans. The developed 3D dense dilated U-Net architecture can accurately predict 3D radiotherapy dose distributions and can be used as part of a fully automated radiation therapy planning pipeline.