A Novel Embedded Feature Selection and Dimensionality Reduction Method for an SVM Type Classifier to Predict Periventricular Leukomalacia (PVL) in Neonates.
Dieter BenderDaniel J LichtChandrasekhar NatarajPublished in: Applied sciences (Basel, Switzerland) (2021)
This paper is concerned with the prediction of the occurrence of periventricular leukomalacia (PVL) in neonates after heart surgery. Our prior work shows that the Support Vector Machine (SVM) classifier can be a powerful tool in predicting clinical outcomes of such complicated and uncommon diseases, even when the number of data samples is low. In the presented work, we first illustrate and discuss the shortcomings of the traditional automatic machine learning (aML) approach. Consequently, we describe our methodology for addressing these shortcomings, while utilizing the designed interactive ML (iML) algorithm. Finally, we conclude with a discussion of the developed method and the results obtained. In sum, by adding an additional (Genetic Algorithm) optimization step in the SVM learning framework, we were able to (a) reduce the dimensionality of an SVM model from 248 to 53 features, (b) increase generalization that was confirmed by a 100% accuracy assessed on an unseen testing set, and (c) improve the overall SVM model's performance from 65% to 100% testing accuracy, utilizing the proposed iML method.
Keyphrases
- machine learning
- deep learning
- big data
- artificial intelligence
- minimally invasive
- acute myeloid leukemia
- neural network
- low birth weight
- heart failure
- risk assessment
- allogeneic hematopoietic stem cell transplantation
- acute coronary syndrome
- percutaneous coronary intervention
- preterm infants
- acute lymphoblastic leukemia