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

Shifeng Chen, Sumin Zhou, Fang Fang Yin, Lawrence B. Marks, Shiva K. Das

Research output: Contribution to journalArticle

53 Scopus citations

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.

Original languageEnglish (US)
Pages (from-to)3808-3814
Number of pages7
JournalMedical physics
Volume34
Issue number10
DOIs
StatePublished - 2007

Keywords

  • Modeling
  • Prediction
  • Radiation pneumonitis
  • Support vector machines

ASJC Scopus subject areas

  • Biophysics
  • Radiology Nuclear Medicine and imaging

Fingerprint Dive into the research topics of 'Investigation of the support vector machine algorithm to predict lung radiation-induced pneumonitis'. Together they form a unique fingerprint.

  • Cite this