Cargando…

A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys

We demonstrate the capabilities of two model-agnostic local post-hoc model interpretability methods, namely breakDown (BD) and shapley (SHAP), to explain the predictions of a black-box classification learning model that establishes a quantitative relationship between chemical composition and multi-p...

Descripción completa

Detalles Bibliográficos
Autores principales: Lee, Kyungtae, Ayyasamy, Mukil V., Ji, Yangfeng, Balachandran, Prasanna V.
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2022
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9270422/
https://www.ncbi.nlm.nih.gov/pubmed/35804179
http://dx.doi.org/10.1038/s41598-022-15618-4
_version_ 1784744465240424448
author Lee, Kyungtae
Ayyasamy, Mukil V.
Ji, Yangfeng
Balachandran, Prasanna V.
author_facet Lee, Kyungtae
Ayyasamy, Mukil V.
Ji, Yangfeng
Balachandran, Prasanna V.
author_sort Lee, Kyungtae
collection PubMed
description We demonstrate the capabilities of two model-agnostic local post-hoc model interpretability methods, namely breakDown (BD) and shapley (SHAP), to explain the predictions of a black-box classification learning model that establishes a quantitative relationship between chemical composition and multi-principal element alloys (MPEA) phase formation. We trained an ensemble of support vector machines using a dataset with 1,821 instances, 12 features with low pair-wise correlation, and seven phase labels. Feature contributions to the model prediction are computed by BD and SHAP for each composition. The resulting BD and SHAP transformed data are then used as inputs to identify similar composition groups using k-means clustering. Explanation-of-clusters by features reveal that the results from SHAP agree more closely with the literature. Visualization of compositions within a cluster using Ceteris-Paribus (CP) profile plots show the functional dependencies between the feature values and predicted response. Despite the differences between BD and SHAP in variable attribution, only minor changes were observed in the CP profile plots. Explanation-of-clusters by examples show that the clusters that share a common phase label contain similar compositions, which clarifies the similar-looking CP profile trends. Two plausible reasons are identified to describe this observation: (1) In the limits of a dataset with independent and non-interacting features, BD and SHAP show promise in recognizing MPEA composition clusters with similar phase labels. (2) There is more than one explanation for the MPEA phase formation rules with respect to the set of features considered in this work.
format Online
Article
Text
id pubmed-9270422
institution National Center for Biotechnology Information
language English
publishDate 2022
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-92704222022-07-10 A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys Lee, Kyungtae Ayyasamy, Mukil V. Ji, Yangfeng Balachandran, Prasanna V. Sci Rep Article We demonstrate the capabilities of two model-agnostic local post-hoc model interpretability methods, namely breakDown (BD) and shapley (SHAP), to explain the predictions of a black-box classification learning model that establishes a quantitative relationship between chemical composition and multi-principal element alloys (MPEA) phase formation. We trained an ensemble of support vector machines using a dataset with 1,821 instances, 12 features with low pair-wise correlation, and seven phase labels. Feature contributions to the model prediction are computed by BD and SHAP for each composition. The resulting BD and SHAP transformed data are then used as inputs to identify similar composition groups using k-means clustering. Explanation-of-clusters by features reveal that the results from SHAP agree more closely with the literature. Visualization of compositions within a cluster using Ceteris-Paribus (CP) profile plots show the functional dependencies between the feature values and predicted response. Despite the differences between BD and SHAP in variable attribution, only minor changes were observed in the CP profile plots. Explanation-of-clusters by examples show that the clusters that share a common phase label contain similar compositions, which clarifies the similar-looking CP profile trends. Two plausible reasons are identified to describe this observation: (1) In the limits of a dataset with independent and non-interacting features, BD and SHAP show promise in recognizing MPEA composition clusters with similar phase labels. (2) There is more than one explanation for the MPEA phase formation rules with respect to the set of features considered in this work. Nature Publishing Group UK 2022-07-08 /pmc/articles/PMC9270422/ /pubmed/35804179 http://dx.doi.org/10.1038/s41598-022-15618-4 Text en © The Author(s) 2022 https://creativecommons.org/licenses/by/4.0/Open AccessThis article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Lee, Kyungtae
Ayyasamy, Mukil V.
Ji, Yangfeng
Balachandran, Prasanna V.
A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
title A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
title_full A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
title_fullStr A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
title_full_unstemmed A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
title_short A comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
title_sort comparison of explainable artificial intelligence methods in the phase classification of multi-principal element alloys
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9270422/
https://www.ncbi.nlm.nih.gov/pubmed/35804179
http://dx.doi.org/10.1038/s41598-022-15618-4
work_keys_str_mv AT leekyungtae acomparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT ayyasamymukilv acomparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT jiyangfeng acomparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT balachandranprasannav acomparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT leekyungtae comparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT ayyasamymukilv comparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT jiyangfeng comparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys
AT balachandranprasannav comparisonofexplainableartificialintelligencemethodsinthephaseclassificationofmultiprincipalelementalloys