Opening the black box of neural networks: methods for interpreting neural network models in clinical applications
Author(s) -
Zhongheng Zhang,
Marcus W. Beck,
David A. Winkler,
Bin Huang,
Wilbert Sibanda,
Hemant Goyal
Publication year - 2018
Publication title -
annals of translational medicine
Language(s) - English
Resource type - Journals
eISSN - 2305-5847
pISSN - 2305-5839
DOI - 10.21037/atm.2018.05.32
Subject(s) - artificial neural network , computer science , outcome (game theory) , black box , plot (graphics) , artificial intelligence , machine learning , variable (mathematics) , code (set theory) , quartile , data mining , statistics , mathematics , programming language , mathematical economics , mathematical analysis , confidence interval , set (abstract data type)
Artificial neural networks (ANNs) are powerful tools for data analysis and are particularly suitable for modeling relationships between variables for best prediction of an outcome. While these models can be used to answer many important research questions, their utility has been critically limited because the interpretation of the "black box" model is difficult. Clinical investigators usually employ ANN models to predict the clinical outcomes or to make a diagnosis; the model however is difficult to interpret for clinicians. To address this important shortcoming of neural network modeling methods, we describe several methods to help subject-matter audiences (e.g., clinicians, medical policy makers) understand neural network models. Garson's algorithm describes the relative magnitude of the importance of a descriptor (predictor) in its connection with outcome variables by dissecting the model weights. The Lek's profile method explores the relationship of the outcome variable and a predictor of interest, while holding other predictors at constant values (e.g., minimum, 20th quartile, maximum). While Lek's profile was developed specifically for neural networks, partial dependence plot is a more generic version that visualize the relationship between an outcome and one or two predictors. Finally, the local interpretable model-agnostic explanations (LIME) method can show the predictions of any classification or regression, by approximating it locally with an interpretable model. R code for the implementations of these methods is shown by using example data fitted with a standard, feed-forward neural network model. We offer codes and step-by-step description on how to use these tools to facilitate better understanding of ANN.
Accelerating Research
Robert Robinson Avenue,
Oxford Science Park, Oxford
OX4 4GP, United Kingdom
Address
John Eccles HouseRobert Robinson Avenue,
Oxford Science Park, Oxford
OX4 4GP, United Kingdom