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...
Autores principales: | , , , |
---|---|
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 |