Development of a machine learning-based risk model for postoperative complications of lung cancer surgery

Purpose To develop a comorbidity risk score specifically for lung resection surgeries. Methods We reviewed the medical records of patients who underwent lung resections for lung cancer, and developed a risk model using data from 2014 to 2017 (training dataset), validated using data from 2018 to 2019 (validation dataset). Forty variables were analyzed, including 35 factors related to the patient’s overall condition and five factors related to surgical techniques and tumor-related factors. The risk model for postoperative complications was developed using an elastic net regularized generalized linear model. The performance of the risk model was evaluated using receiver operating characteristic curves and compared with the Charlson Comorbidity Index (CCI). Results The rate of postoperative complications was 34.7% in the training dataset and 21.9% in the validation dataset. The final model consisted of 20 variables, including age, surgical-related factors, respiratory function tests, and comorbidities, such as chronic obstructive pulmonary disease, a history of ischemic heart disease, and 12 blood test results. The area under the curve (AUC) for the developed risk model was 0.734, whereas the AUC for the CCI was 0.521 in the validation dataset. Conclusions The new machine learning model could predict postoperative complications with acceptable accuracy. Clinical registration number 2020–0375.