6533b85bfe1ef96bd12bb563
RESEARCH PRODUCT
Comparison of feature importance measures as explanations for classification models
Susanne JauhiainenMirka Saarelasubject
feature importanceComputer scienceGeneral Chemical EngineeringGeneral Physics and Astronomy02 engineering and technologyinterpretable modelstekoälyMachine learningcomputer.software_genreLogistic regressionDomain (software engineering)020204 information systems0202 electrical engineering electronic engineering information engineeringFeature (machine learning)General Materials ScienceGeneral Environmental Scienceluokitus (toiminta)explainable artificial intelligencebusiness.industrylogistic regressionGeneral EngineeringRandom forestkoneoppiminenTrustworthinessInjury dataGeneral Earth and Planetary Sciences020201 artificial intelligence & image processingArtificial intelligencebusinesscomputerrandom forestdescription
AbstractExplainable artificial intelligence is an emerging research direction helping the user or developer of machine learning models understand why models behave the way they do. The most popular explanation technique is feature importance. However, there are several different approaches how feature importances are being measured, most notably global and local. In this study we compare different feature importance measures using both linear (logistic regression with L1 penalization) and non-linear (random forest) methods and local interpretable model-agnostic explanations on top of them. These methods are applied to two datasets from the medical domain, the openly available breast cancer data from the UCI Archive and a recently collected running injury data. Our results show that the most important features differ depending on the technique. We argue that a combination of several explanation techniques could provide more reliable and trustworthy results. In particular, local explanations should be used in the most critical cases such as false negatives.
year | journal | country | edition | language |
---|---|---|---|---|
2021-02-01 | SN Applied Sciences |