Oriented Feature Selection SVM Applied to Cancer Prediction in Precision Medicine

Yang Shen, Chunxue Wu, Cong Liu, Yan Wu, Naixue Xiong

Research output: Contribution to journalArticlepeer-review

30 Scopus citations

Abstract

Advances in the gene sequencing technology and the outbreak of artificial intelligence have made precision medicine a reality recently. Applying machine learning algorithms to cancer prediction using gene expression data helps to discover the link between genetic data and cancer, which will promote the development and application of precision medicine. Considering the natural order of genes, a new classification method that combines fused lasso and elastic net as regularization for linear support vector machine (SVM), which uses huberized hinge loss as the loss function, is proposed in this paper, which we name it oriented feature selection SVM (OFSSVM). Due to the characteristics of the elastic net and fused lasso, the OFSSVM can not only provide automatic feature selection, but also average the adjacent coefficients, resulting in a sparse and smooth solution. We demonstrate its effectiveness in both binary classification and multiclass classification in the sense of comprehensive evaluation that not only the classification accuracy but also the interpretability are considered. The experiments show that the OFSSVM is an appealing compromise between interpretability and classification accuracy, and is superior to other traditional methods in the sense of comprehensive evaluation.

Original languageEnglish
Article number8452888
Pages (from-to)48510-48521
Number of pages12
JournalIEEE Access
Volume6
DOIs
StatePublished - 30 Aug 2018
Externally publishedYes

Keywords

  • Cancer prediction
  • SVM
  • elastic net
  • feature selection
  • fused lasso
  • gene expression data
  • machine learning
  • precision medicine

Fingerprint

Dive into the research topics of 'Oriented Feature Selection SVM Applied to Cancer Prediction in Precision Medicine'. Together they form a unique fingerprint.

Cite this