Automated Intensity Modulated Radiation Therapy Treatment Planning for Cervical Cancer Based on Convolution Neural Network

PURPOSE: To develop and evaluate an automatic intensity-modulated radiation therapy (IMRT) program for cervical cancer, including a Convolution Neural Network (CNN)-based prediction model and an automated optimization strategy. METHODS: A CNN deep learning model was trained to predict a patient-specify set of IMRT objectives based on overlap volume histograms (OVH) and high-quality plan of previous patients. A total of 140 cervical cancer patients were enrolled in this study, including 100 patients in the training set, 20 patients in the validation set and 20 patients in the testing set. The input of this model was OVH data and the output were values of IMRT plan objectives. For patients in the testing set, the set of planning objectives were predicted by the CNN model and used to automatically generate IMRT plans. Meanwhile, manual plans of these patients were generated by 1 beginner planner and 1 senior planner respectively. Finally, dose distribution, dosimetric parameters and planning time were analyzed. In addition, the 3 types of plans were blinded compared and ranked by an experienced oncologist. RESULTS: There were almost no statistically differences among these 3 types of plans in target coverage and dose conformity. Dose homogeneity were slightly decreased while the average dose and parameters for most organs-at-risk (OARs) were decreased in automatic plans. Especially in comparison with manual plans by the beginner planner, V(40) of bladder and rectum decreased 6.3% and 12.3%, while mean dose of rectum and marrow were 1.1 Gy and 1.8 Gy lower with automatic plans (either P < 0.017). In the blinded comparison, automatic plans were chosen as best plan in 14 cases. CONCLUSIONS: For cervical cancer, automatic IMRT plans optimized from the CNN generated objectives have superior dose sparing without compromising of target dose. It significantly reduced the planning time.