MURAGAKI Yoshihiro
   Department   School of Medicine(Tokyo Women's Medical University Hospital), School of Medicine
   Position   Visiting Professor
Article types Original article
Language English
Peer review Peer reviewed
Title A Method to Extract Feature Variables Contributed in Nonlinear Machine Learning Prediction
Journal Formal name:Methods of information in medicine
Abbreviation:Methods Inf Med
ISSN code:00261270/2511705X
Domestic / ForeginForegin
Publisher Stuttgart : F K Schattauer Verlag
Volume, Issue, Page 59(1),pp.1-8
Author and coauthor SUZUKI Mayumi†*, SHIBAHARA Takuma, MURAGAKI Yoshihiro
Publication date 2020/02
Summary Background: Although advances in prediction accuracy have been made with new machine learning methods, such as support vector machines and deep neural networks, these methods make nonlinear machine learning models and thus lack the ability to explain the basis of their predictions. Improving their explanatory capabilities would increase the reliability of their predictions.
Objective: Our objective was to develop a factor analysis technique that enables the presentation of the feature variables used in making predictions, even in nonlinear machine learning models.
Methods: A factor analysis technique was consisted of two techniques: backward analysis technique and factor extraction technique. We developed a factor extraction technique extracted feature variables that was obtained from the posterior probability distribution of a machine learning model which was calculated by backward analysis technique.
Results: In evaluation, using gene expression data from prostate tumor patients and healthy subjects, the prediction accuracy of a model of deep neural networks was approximately 5% better than that of a model of support vector machines. Then the rate of concordance between the feature variables extracted in an earlier report using Jensen-Shannon divergence and the ones extracted in this report using backward elimination using Hilbert-Schmidt independence criteria was 40% for the top five variables, 40% for the top 10, and 49% for the top 100.
Conclusion: The results showed that models can be evaluated from different viewpoints by using different factor extraction techniques. In the future, we hope to use this technique to verify the characteristics of features extracted by factor extraction technique, and to perform clinical studies using the genes, we extracted in this experiment.
DOI 10.1055/s-0040-1701615
PMID 32380557