In medical applications of machine learning, it is essential to interpret model behavior and explore the true association between features and clinical outcomes. Conventional Explainable AI approaches typically focus on “model-dependent” global feature importance from trained models, which may not guarantee “data-driven” causal relationships or medical consistency. To address this, we propose CausalAIME, a new framework that combines approximate inverse model explanations (AIME) with the Peter-Clark algorithm for causal discovery. This integration suppresses multicollinearity while enabling global visualization of both feature signs (positive or negative) and class-specific contributions. Furthermore, by choosing either the model’s output \(\hat{Y}\) or the true label Y as input, CausalAIME unifies both model-dependent and data-driven global feature importance within the same framework. Our experiments using breast cancer diagnostic data compared CausalAIME with existing methods such as Random Forest and SHAP, highlighting the advantages of CausalAIME in offering sign-based interpretability and causal perspectives, both critical in clinical settings. We anticipate that CausalAIME will contribute to enhanced explainability across various domains, including healthcare, by accommodating the needs for both true association analysis and model behavior interpretation.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CausalAIME: Leveraging Peter-Clark Algorithms and Inverse Modeling for Unified Global Feature Explanation in Healthcare

  • Takafumi Nakanishi

摘要

In medical applications of machine learning, it is essential to interpret model behavior and explore the true association between features and clinical outcomes. Conventional Explainable AI approaches typically focus on “model-dependent” global feature importance from trained models, which may not guarantee “data-driven” causal relationships or medical consistency. To address this, we propose CausalAIME, a new framework that combines approximate inverse model explanations (AIME) with the Peter-Clark algorithm for causal discovery. This integration suppresses multicollinearity while enabling global visualization of both feature signs (positive or negative) and class-specific contributions. Furthermore, by choosing either the model’s output \(\hat{Y}\) or the true label Y as input, CausalAIME unifies both model-dependent and data-driven global feature importance within the same framework. Our experiments using breast cancer diagnostic data compared CausalAIME with existing methods such as Random Forest and SHAP, highlighting the advantages of CausalAIME in offering sign-based interpretability and causal perspectives, both critical in clinical settings. We anticipate that CausalAIME will contribute to enhanced explainability across various domains, including healthcare, by accommodating the needs for both true association analysis and model behavior interpretation.