Mean Heart Dose Prediction Using Parameters of Single-Slice Computed Tomography and Body Mass Index: Machine Learning Approach for Radiotherapy of Left-Sided Breast Cancer of Asian Patients

Abstract

Deep inspiration breath-hold (DIBH) is an excellent technique to reduce the incidental radiation received by the heart during radiotherapy in patients with breast cancer. However, DIBH is costly and time-consuming for patients and radiotherapy staff. In Asian countries, the use of DIBH is restricted due to the limited number of patients with a high mean heart dose (MHD) and the shortage of radiotherapy personnel and equipment compared to that in the USA. This study aimed to develop, evaluate, and compare the performance of ten machine learning algorithms for predicting MHD using a patient’s body mass index and single-slice CT parameters to identify patients who may not require DIBH. Machine learning models were built and tested using a dataset containing 207 patients with left-sided breast cancer who were treated with field-in-field radiotherapy with free breathing. The average MHD was 251 cGy. Stratified repeated four-fold cross-validation was used to build models using 165 training data. The models were compared internally using their average performance metrics: F2 score, AUC, recall, accuracy, Cohen’s kappa, and Matthews correlation coefficient. The final performance evaluation for each model was further externally analyzed using 42 unseen test data. The performance of each model was evaluated as a binary classifier by setting the cut-off value of MHD ≥ 300 cGy. The deep neural network (DNN) achieved the highest F2 score (78.9%). Most models successfully classified all patients with high MHD as true positive. This study indicates that the ten models, especially the DNN, might have the potential to identify patients who may not require DIBH.