Organ-at-risk dose prediction using a machine learning algorithm: Clinical validation and treatment planning benefit for lung SBRT

Objective: To quantify the clinical performance of a machine learning (ML) algorithm for organ-at-risk (OAR) dose prediction for lung stereotactic body radiation therapy (SBRT) and estimate the treatment planning benefit from having upfront access to these dose predictions. Methods: ML models were trained using multi-center data consisting of 209 patients previously treated with lung SBRT. Two prescription levels were investigated, 50 Gy in five fractions and 54 Gy in three fractions. Models were generated using a gradient-boosted regression tree algorithm using grid searching with fivefold cross-validation. Twenty patients not included in the training set were used to test OAR dose prediction performance, ten for each prescription. We also performed blinded re-planning based on OAR dose predictions but without access to clinically delivered plans. Differences between predicted and delivered doses were assessed by root-mean square deviation (RMSD), and statistical differences between predicted, delivered, and re-planned doses were evaluated with one-way analysis of variance (ANOVA) tests. Results: ANOVA tests showed no significant differences between predicted, delivered, and replanned OAR doses (all p ≥ 0.36). The RMSD was 2.9, 3.9, 4.3, and 1.7Gy for max dose to the spinal cord, great vessels, heart, and trachea, respectively, for 50 Gy in five fractions. Average improvements of 1.0, 1.4, and 2.0 Gy were seen for spinal cord, esophagus, and trachea max doses in blinded replans compared to clinically delivered plans with 54 Gy in three fractions, and 1.8, 0.7, and 1.5 Gy, respectively, for the esophagus, heart and bronchus max doses with 50 Gy in five fractions. Target coverage was similar with an average PTV V100% of 94.7% for delivered plans compared to 97.3% for blinded re-plans for 50 Gy in five fractions, and respectively 98.4% versus 99.2% for 54 Gy in three fractions. Conclusion: This study validated ML-based OAR dose prediction for lung SBRT, showing potential for improved OAR dose sparing and more consistent plan quality using dose predictions for patient-specific planning guidance.