Loading…
Causality-based counterfactual explanation for classification models
Counterfactual explanation is one branch of interpretable machine learning that produces a perturbation sample to change the model’s original decision. The generated samples can act as a recommendation for end-users to achieve their desired outputs. Most of the current counterfactual explanation app...
Saved in:
Published in: | Knowledge-based systems 2024-09, Vol.300, p.112200, Article 112200 |
---|---|
Main Authors: | , , |
Format: | Article |
Language: | English |
Subjects: | |
Citations: | Items that this one cites |
Online Access: | Get full text |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
Summary: | Counterfactual explanation is one branch of interpretable machine learning that produces a perturbation sample to change the model’s original decision. The generated samples can act as a recommendation for end-users to achieve their desired outputs. Most of the current counterfactual explanation approaches are the gradient-based method, which can only optimize the differentiable loss functions with continuous variables. Accordingly, the gradient-free methods are proposed to handle the categorical variables, which however have several major limitations: (1) causal relationships among features are typically ignored when generating the counterfactuals, possibly resulting in impractical guidelines for decision-makers; (2) the counterfactual explanation algorithm requires a great deal of effort into parameter tuning for determining the optimal weight for each loss functions which must be conducted repeatedly for different datasets and settings. In this work, to address the above limitations, we propose a prototype-based counterfactual explanation framework (ProCE). ProCE is capable of preserving the causal relationship underlying the features of the counterfactual data. In addition, we design a novel gradient-free optimization based on the multi-objective genetic algorithm that generates the counterfactual explanations for the mixed-type of continuous and categorical features. Numerical experiments demonstrate that our method compares favorably with state-of-the-art methods and therefore is applicable to existing prediction models. All the source codes and data are available at https://github.com/tridungduong16/multiobj-scm-cf.
•Proposed method produces counterfactuals satisfying the causal constraints.•Utilize the auto-encoder model and class prototype to to speed up the searching speed.•Design a novel multi-objective optimization. |
---|---|
ISSN: | 0950-7051 |
DOI: | 10.1016/j.knosys.2024.112200 |