An interpretable machine learning model for diagnosis of Alzheimer's disease
Authors:
Open Access
Abstract
We present an interpretable machine learning model for medical diagnosis called sparse high-order interaction model with rejection option (SHIMR). A decision tree explains to a patient the diagnosis with a long rule (i.e., conjunction of many intervals), while SHIMR employs a weighted sum of short rules. Using proteomics data of 151 subjects in the Alzheimer’s Disease Neuroimaging Initiative (ADNI) dataset, SHIMR is shown to be as accurate as other non-interpretable methods (Sensitivity, SN = 0.84 ± 0.1, Specificity, SP = 0.69 ± 0.15 and Area Under the Curve, AUC = 0.86 ± 0.09). For clinical usage, SHIMR has a function to abstain from making any diagnosis when it is not confident enough, so that a medical doctor can choose more accurate but invasive and/or more costly pathologies. The incorporation of a rejection option complements SHIMR in designing a multistage cost-effective diagnosis framework. Using a baseline concentration of cerebrospinal fluid (CSF) and plasma proteins from a common cohort of 141 subjects, SHIMR is shown to be effective in designing a patient-specific cost-effective Alzheimer’s disease (AD) pathology. Thus, interpretability, reliability and having the potential to design a patient-specific multistage cost-effective diagnosis framework can make SHIMR serve as an indispensable tool in the era of precision medicine that can cater to the demand of both doctors and patients, and reduce the overwhelming financial burden of medical diagnosis.Keywords
Funding Information
- “Materials research by Information Integration” Initiative (MI2I) project and Core Research for Evolutional Science and Technology (JPMJCR1502 from Japan Science and Technology Agency (JST))
- Ministry of Education, Culture, Sports, Science and Technology (MEXT) as “Priority Issue on Post-K computer” (Building Innovative Drug Discovery Infrastructure Through Functional Control of Biomolecular Systems)
This publication has 18 references indexed in Scilit:
- How the machine ‘thinks’: Understanding opacity in machine learning algorithmsBig Data & Society, 2016
- 2015 Alzheimer's disease facts and figuresAlzheimer's & Dementia, 2015
- Evaluation of Plasma Proteomic Data for Alzheimer Disease State Classification and for the Prediction of Progression From Mild Cognitive Impairment to Alzheimer DiseaseAlzheimer Disease & Associated Disorders, 2013
- An empirical evaluation of the comprehensibility of decision table, tree and rule based predictive modelsDecision Support Systems, 2011
- Mining complex genotypic features for predicting HIV-1 drug resistanceBioinformatics, 2007
- Using Rule Extraction to Improve the Comprehensibility of Predictive ModelsSSRN Electronic Journal, 2006
- Diversity and complexity of HIV-1 drug resistance: A bioinformatics approach to predicting phenotype from genotypeProceedings of the National Academy of Sciences, 2002
- Knowledge discovery from data?IEEE Intelligent Systems and their Applications, 2000
- On-Line Algorithms in Machine Learning1998
- The CN2 induction algorithmMachine Learning, 1989