Voxel-Based Dose Prediction with Multi-Patient Atlas Selection for Automated Radiotherapy Treatment Planning

Abstract

Automating the radiotherapy treatment planning process is a technically challenging problem. The majority of automated approaches have focused on customizing and inferring dose volume objectives to used in plan optimization. In this work we outline a multi-patient atlas-based dose prediction approach that learns to predict the dose-per-voxel for a novel patient directly from the computed tomography (CT) planning scan without the requirement of specifying any objectives. Our method learns to automatically select the most effective atlases for a novel patient, and then map the dose from those atlases onto the novel patient. We extend our previous work to include a conditional random field for the optimization of a joint distribution prior that matches the complementary goals of an accurately spatially distributed dose distribution while still adhering to the desired dose volume histograms. The resulting distribution can then be used for inverse-planning with a new spatial dose objective, or to create typical dose volume objectives for the canonical optimization pipeline. We investigated six treatment sites (633 patients for training and 113 patients for testing) and evaluated the mean absolute difference (MAD) in all DVHs for the clinical and predicted dose distribution. The results on average are favorable in comparison to our previous approach (1.91 vs 2.57). Comparing our method with and without atlas-selection further validates that atlas-selection improved dose prediction on average in Whole Breast (0.64 vs 1.59), Prostate (2.13 vs 4.07), and Rectum (1.46 vs 3.29) while it is less important in Breast Cavity (0.79 vs 0.92) and Lung (1.33 vs 1.27) for which there is high conformity and minimal dose shaping. In CNS Brain, atlas-selection has the potential to be impactful (3.65 vs 5.09), but selecting the ideal atlas is the most challenging.

arXiv comment18 pages, 4 figures, submitted to Physics in Medicine and Biology

Cited by in corpus (9)

• Fully Automated Treatment Planning for Head and Neck Radiotherapy using a Voxel-Based Dose Prediction and Dose Mimicking Method
• OpenKBP: The open-access knowledge-based planning grand challenge
• Probabilistic dose prediction using mixture density networks for automated radiation therapy treatment planning
• Knowledge-based Radiation Treatment Planning: A Data-driven Method Survey
• Improving Proton Dose Calculation Accuracy by Using Deep Learning
• Atlas Based Segmentations via Semi-Supervised Diffeomorphic Registrations
• Domain Adaptation of Automated Treatment Planning from Computed Tomography to Magnetic Resonance
• Three-Dimensional Dose Prediction for Lung IMRT Patients with Deep Neural Networks: Robust Learning from Heterogeneous Beam Configurations
• Increased accuracy of planning tools for optimization of dynamic multileaf collimator delivery of radiotherapy through reformulated objective functions