Investigation of the support vector machine algorithm to predict lung radiation-induced pneumonitis

Abstract

The purpose of this study is to build and test a support vector machine (SVM) model to predict for the occurrence of lung radiation-induced Grade 2+ pneumonitis. SVM is a sophisticated statistical technique capable of separating the two categories of patients (with/without pneumonitis) using a boundary defined by a complex hypersurface. Despite the complexity, the SVM boundary is only minimally influenced by outliers that are difficult to separate. By contrast, the simple hyperplane boundary computed by the more commonly used and related linear discriminant analysis method is heavily influenced by outliers. Two SVM models were built using data from 219 patients with lung cancer treated using radiotherapy (34 diagnosed with pneumonitis). One model (SVMall) selected input features from all dose and non-dose factors. For comparison, the other model (SVMdose) selected input features only from lung dose-volume factors. Model predictive ability was evaluated using ten-fold cross-validation and receiver operating characteristics (ROC) analysis. For the model SVMall, the area under the cross-validated ROC curve was 0.76 (sensitivityspecificity= 74%75%). Compared to the corresponding SVMdose area of 0.71 (sensitivityspecificity=68%68%), the predictive ability of SVMall was improved, indicating that non-dose features are important contributors to separating patients with and without pneumonitis. Among the input features selected by model SVMall, the two with highest importance for predicting lung pneumonitis were: (a) generalized equivalent uniform doses close to the mean lung dose, and (b) chemotherapy prior to radiotherapy. The model SVMall is publicly available via internet access. © 2007 American Association of Physicists in Medicine.