As AI becomes more prevalent in society, many stakeholders are requiring AI predictions to be accompanied by explanations. In addition to GDPR (Commission 2017), the state of Illinois recently passed a law about video interviews that requires employers to provide information “explaining how the artificial intelligence works (Illinois 2019). Concurrently, the research community has developed many explainability or interpretability techniques. Most of these take an “inside out” (model-to-consumer) approach to providing explanations based on how the models work. Although this can be useful to those explanation consumers who understand ML models, it can pose a challenge for non-technical consumers. For example, many techniques provide explanations composed of input features of the model, e.g. proprietary risk scores in a loan application scenario, but often these features may not be meaningful to an end user. Other approaches explain by example by showing similar instances from the training set, which can be useful to a domain expert, but not to an end user. The TED explainability framework (Hind et al. 2019) takes a different “outside in” (consumer-to-model) approach to explainable AI. It attempts to provide meaningful explanations by capturing the conceptual domain of the explanation consumer. The framework uses a training dataset that includes explanations (E), in addition to features (X)
and target labels (Y ), to create a model that simultaneously produces predictions and explanations. The framework is general in that it can be used with many supervised machine learning algorithms, thereby leveraging their wellestablished predictive abilities to also predict explanations.
Previous work has shown the efficacy of this approach on two synthetic datasets (Hind et al. 2019) and on two other datasets where explanations are synthetically created (Codella et al. 2019). Based on the promising results from this initial work, this paper more deeply explores some of the practical considerations with the TED approach.
The contributions of this paper are
• Empirical evaluations that address open questions regarding how TED performs with different base classification algorithms, with varying numbers of explanations, and with variability in the explanations for a given set of features. Results show that accuracies are maintained as the number of explanations increases to more than 30 and degrade gradually as variability increases.
• A new algorithm that allows TED to be deployed when some training examples are missing explanations, addressing one of the most significant challenges of using TED. An evaluation of this partially supervised algorithm shows that it can reduce the number of training explanations by 90% with little loss in accuracy.
These contributions demonstrate that TED is a robust explanation technique and useful when meaningful explanations are paramount. Section 2 provides an overview of the TED framework. Section 3 describes the new algorithm for partially supervised learning. Section 4 describes two datasets and the synthesis of labels that we use for our evaluation. Section 5 describes our experiments and results. Section 6 discusses open problems and future work.
The main motivation for the TED (Teaching Explainable Decisions) framework is that a meaningful explanation is quite subjective (Hind 2019; Arya et al. 2019), and thus, requires additional information from the explanation consumer regarding what is meaningful to them. This additional information takes the form of training data composed
of triples corresponding to the input space, output space, and explanation space, respectively. From this training data, TED learns to predict Y and E jointly from X. There are many instantiations of the TED framework. The
initial paper (Hind et al. 2019) describes a simple Cartesian
product approach, where the Y and E labels are combined into a new label, (Y, E), and a variety of supervised learning algorithms can be applied to solve the resulting multi-class problem. Other instantiations include 1) a multi-task learning approach in which predicting labels and explanations constitute two or more tasks, and 2) building upon the tradition of similarity metrics, case-based reasoning and contentbased retrieval (Codella et al. 2019). This work focuses on the Cartesian product instantiation
of TED, referred to as TED-C, and implemented in the AI
Explainability 360 open source toolkit (Arya et al. 2019). We choose this instantiation for its simplicity and generality as discussed above. We pursue the following open research questions:
1. How does TED-C impact target label (Y ) accuracy compared to the base model?
2. How does the effectiveness of TED-C vary with different learning algorithms?
3. How does the effectiveness of TED-C scale with an increase in the number of possible explanations?
4. How does the effectiveness of TED-C depend on variability in the explanations associated with a given input X?
5. How can the TED framework be extended to work in a partially-supervised manner, i.e., when some training instances do not have an explanation?
6. What is the performance of this new algorithm? TED is best suited for use cases where the labels and ex-
planations are consistent, which typically occurs when they
are created at the same time. We explore this scenario in this paper. In the situation where a dataset already has labels, either from a human or from actual outcomes, such as loan defaults, it can be challenging to produce explanations that are consistent with the labels. For example, asking someone to explain why a person defaulted on a loan, given their loan application, is not a trivial matter. We leave such scenarios for future work, noting however that the partially supervised approach described in Section 3 can help reduce the amount of explanation labeling required. One of the strengths of the TED approach is that it en-
ables explanations that involve higher-level concepts that are
more meaningful than features. For example, in Table 1 we enumerate 8 different conceptual reasons (the non-indented descriptions in the table) that someone might be a retention risk, i.e., likely to leave a company. These concepts are independent of the features and, in fact, can be implemented differently for different companies or the same company over time. They are likely to be more meaningful to a non-technical explanation consumer.
One significant drawback of the TED framework is the requirement that training datasets contain explanations in addition to decision labels. Although researchers have reported that producing explanations at the same time as producing labels incurs negligible overhead (Zaidan and Eisner 2007; 2008; Zhang, Marshall, and Wallace 2016; McDonnell et al. 2016), it is desirable to reduce this overhead.
We approach this problem by allowing some training instances to not have explanations. We propose to use imputation to fill in missing E values and thereby allow the general TED framework to be used without further modification. Using the part of the training data with E labels, a model is trained to predict E from X and Y (the latter is available for all training examples). This model is used to impute E labels for the part of the training data that lacks them. The full training set with given and imputed E labels is then passed as input to the TED framework. We evaluate this approach for TED-C in Section 5.3.
Evaluation of TED requires training data containing explanations, ideally provided by domain experts. Since such datasets are not yet readily available, we instead use semi-synthetic data generated as described in this section. Semisynthetic data does have advantages over real-world data in enabling the study of certain effects on the performance of TED, in particular those due to the amount of variability in explanations and the number of possible explanations, which are the subject of this paper (Sections 5.1 and 5.2).
We use two datasets described in the following subsections. For both datasets, we first derive a set of rules that are intended to approximate a domain expert’s knowledge and the decisions and explanations that they might provide. We then retain the original features and synthesize decisions and explanations according to these rules. This rule-based approach is becoming increasingly popular due to the scarcity of true experts for labeling (Ratner et al. 2017) and is common in behavioral science (Heyman et al. 2014).
4.1 FICO Challenge Dataset
The FICO Explainable Machine Learning Challenge dataset (FICO 2018) contains credit information on around 10,000 applicants for home equity lines of credit (HELOC), together with an outcome label of whether they were delinquent for 90 days or longer in repaying. Our goal is to simulate the credit approval decisions and explanations that a loan officer might produce based on applicant features.
To substitute for domain expertise, we used the Boolean Rules via Column Generation (BRCG) algorithm (Dash, G¨unl¨uk, and Wei 2018) as implemented in (AIX 2019) to learn a set of rules for approving applications. We first removed the credit score–like ‘ExternalRiskEstimate’ feature, which was causing explanations to be more opaque, and processed special values as detailed in Appendix A. A 10-fold cross-validation (CV) experiment showed that BRCG regularization parameter values of result in good accuracy in predicting outcomes with a simple rule set.
The rule set learned by BRCG from the entire dataset with has the form shown below but with different thresholds (see Appendix A). To make the rules more human-like, we replaced the thresholds with nearby round numbers to yield
This rule set has an accuracy of 71% in predicting the repayment outcome label. The ‘NetFractionRevolvingBurden’ feature is the applicant’s revolving (e.g. credit card) debt as a percentage of their credit limit. We label a value no greater than 40 as “low debt” and between 40 and 60 as “medium debt”. Conditions 2 and 4 in rule 1 pertain to delinquencies; condition 2 requires the percentage of never-delinquent trades (credit agreements) to be above 85% while condition 4 requires no delinquencies in the last 12 months. Condition 3 in rule 1 requires the average age of existing accounts to be greater than 4 years. While rule 1 may be regarded as a rule for general creditworthiness, rule 2 is a “shortcut” in checking for only two conditions, low debt and no credit inquiries in the last 2 years, as signals of ability to repay.
Rules 1 and 2 above are used to synthesize decision (Y ) and explanation (E) labels alongside the original features (X). The “base” version of this semi-synthetic FICO dataset has 9 possible explanations: 2 for the positive (approve) class corresponding to which of the two rules is satisfied (rule 1 takes precedence if both), and 7 for the negative (deny) class that are minimally sufficient in explaining why neither of the rules is satisfied. The latter 7 explanations are 1) “high debt” (> 60), which is sufficient to violate both rules, 2)–4) medium debt coupled with the violation of one of conditions 2–4 in rule 1 (earlier conditions take precedence if more than one is violated), and 5)–7) low debt but an inquiry in the last 2 years, again coupled with one of conditions 2–4 in rule 1.
To obtain larger numbers of explanations, we increase the maximum number of unsatisfied conditions that are communicated in a credit denial. The 7 explanations above for the denied class consist of only 1 or 2 unsatisfied conditions (e.g. high debt, medium debt plus one other condition), even if there are additional unsatisfied conditions. The set of explanations may be enlarged, for example, by telling high-debt applicants that they also have too many delinquent trades or insufficient credit history. In this manner, the number of explanations can be increased to more than 30.
4.2 Employee Retention Dataset
In the employee retention use case, a company tries to determine which of its employees are risks to leave and thus, for whom it should consider a proactive action to retain the employee. We use the dataset described in (Varshney et al. 2012), which is modeled based on real data. The original dataset contains 14 features and 9,999 employees. We selected the following subset of 8 features: Business unit,
Position, Salary, Months since hire, Months since promotion, Last evaluation, Evaluation slope over the last 3 years, Strong potential indicator.
We then created 26 rules, further grouped in categories as described below, that signify a possible retention issue. The rules were created by an experienced manager with 17 years experience (one of the authors) and enhanced by the BRCG algorithm (Dash, G¨unl¨uk, and Wei 2018) mentioned above. The rules were then applied to the dataset to produce Y labels specifying if the employee was a retention risk. The explanation for the prediction was the rule number for retention risks and a default “No Risk” explanation for nonretention risks. This use case and the rules are inspired by the Proactive Retention tutorial in (AIX 2019).
Before applying the rules, we applied discretization to several of the features. The 33 organizations were mapped to 3 groups of high-, medium-, and low-attrition organizations. The positions were similarly grouped into 4 categories: Entry, Junior, Senior, Established. Salary was mapped into 3 salary competitiveness groups (low, medium, high) based on how it compared to the average salary for the position in the dataset (medium was within 20%). Similarly, evaluations were grouped into low, medium, high. The duration-related features were unchanged; the rules specify bounds, such as 120 < MonthsSincePromotion < 180, to signify an employee who might be at risk for mid-career crisis.
Table 1: Employee Retention Explanations
Table 1 describes the explanations used for the employee retention dataset. The explanations are partitioned into 3 levels (indicated by indentation), depending on the level of detail required by the consumer. For example, at the lowest level we have the explanations at the outermost indentation level: “Promotion Lag”, “New Employee”, “Disappointing Evaluation”, etc. These show the top 8 reasons why employees are in danger of leaving the company. The next level provides additional refinement of these explanations. For example, “Promotion Lag” is decomposed into 3 different types, depending on the organization. Employees in Organization 1 historically have a higher attrition rate and those in Organization 3 have a lower rate, so it is useful to provide this extra level of information in an explanation. This results in a total of 16 explanations. The last 2 explanations, “Mid-Career Crisis” and “Seeking Higher Salary”, are not decomposed, so they remain as in the first level. The third level provides further differentiation in explanations for those consumers who need even more details.
A key advantage of TED over other explainability approaches is the ability to define concepts that provide higher-level semantic meaning than features. For example, the concept of “Promotion Lag” refers to an employee believing that the time since their most recent promotion is excessive. Exactly how this concept is mapped into features may vary among organizations, positions, etc. For example, “recent” can be defined as within the last N months for values of N that depend on organizations, positions, etc. This is how the promotion lag rules are implemented for this use case.
4.3 Adding Variability to Labels
Explanations based on the rules in Sections 4.1 and 4.2 exhibit no variability in that they are a deterministic function of the features. This may not be representative of explanations obtained from humans, who may make errors, disagree with each other, especially in borderline cases, or generally vary in their assessments. To simulate these effects, we experiment with adding variability to the synthesized explanations, which we will refer to as noise for convenience.
We consider two types of noise, both based on the fact that the rules operate on discretized features, e.g. thresholded NetFractionRevolvingBurden in Section 4.1. Perturbations to these features may thus result in changes from one explanation or decision to a “nearby” one. The first type of noise models variability in the thresholds applied to discretize continuous features. This variability may arise, for example, from different loan officers’ opinions about what constitutes a high versus a medium debt burden. Suppose that in the absence of noise, a continuous feature is converted into an ordinal feature
by means of thresholds
such that
if and only if
and
may be infinite). In the presence of noise, the mapping becomes random and is given by the conditional cumulative distribution function
(1) where g is the logistic function and is a parameter that scales the standard deviation
of
. If
is much lower or much higher than the threshold
, then the probability is close to either 1 or 0. However, if
lies in a transition zone around
, whose width is defined by
, then there is uncertainty as reflected in a probability between 0 and 1.
The second type of noise occurs after discretization and perturbs the value of an ordinal feature . This may model human errors in misreading
or more general variability that occurs in this discrete space (for example, one loan of-ficer may have a rule concerning high-debt applicants while another includes both medium and high debt). It is assumed that perturbations to nearby values are most likely. Specifi-cally, we generate a noisy feature
from
according to
so that only mappings to adjacent levels are allowed, with total probability .
This section evaluates the TED-C framework’s performance under the impact of four variables: the number of explanation options, the amount of noise in the explanation (E) and decision (Y ) labels, the availability of E labels, and the spe-cific algorithm used to classify the Cartesian product (Y, E). We report multiple measures of the performance of TEDC: E accuracy, Y accuracy, and the difference between Y accuracy in predicting the (Y, E) Cartesian product and in predicting Y alone. The third measure answers our first research question, i.e., whether and how the TED-C framework affects a classifier’s ability to predict decisions given the requirement to also predict explanations. Both accuracy (proportion of labels predicted correctly) and F1 scores were collected from the experiments. However, because we found that the two were highly correlated, we present only the accuracy results below. We evaluated TED-C’s performance using four classifi-cation algorithms: LightGBM (LGB, Ke et al. 2017), Random Forest (RF), SVM, and Logistic Regression (LR). The first two produce tree ensembles, with LGB being a state-of-the-art gradient boosting method, while the last two are linear classification algorithms, where SVM may use nonlinear kernels. We hypothesize that since our explanations were generated by rules and the first two algorithms are similar to rule-based algorithms, they would likely perform better than the linear classifiers within the TED-C framework. To attempt to maximize the performance of each algorithm, we conducted hyperparameter optimization using the Bayesian Optimization and Hyperband method (Falkner, Klein, and Hutter 2018). This method is efficient because it combines Bayesian optimization with bandit-based optimization. We did not employ its successive halving feature because each run of these four algorithms is reasonably fast. Table 2 (in Appendix B) lists, for each algorithm, the hyperparameters that were searched in the experiment. All other parameters were kept constant at their default values except 1) RF’s number of estimators was set to 100 (default is 10), and 2) SVM and LR’s maximum number of iterations was set to 1000 because little improvement in accuracy was observed beyond this point. However, this restriction was ap-
Figure 1: The effect of the number of possible explanations on TED-C’s E and Y accuracy using four classification algorithms. The circles represent mean accuracy across five folds, and the error bars represent the 95% confidence interval for the mean. The rightmost graph shows the Y accuracy difference between a model that predicts only Y and one that predicts (Y, E) pairs.
Figure 2: The effect of explanation noise parameters and
on TED-C’s E and Y accuracy.
plied only in the parameter tuning phase and was removed during final model fitting for better accuracy.
For each experimental setting, a five-fold CV was conducted for each classification algorithm. Test results are averaged across these five folds and reported below. For each fold, we conducted hyperparameter search using the training data of that fold, and evaluated the resulting optimal model on the test data of that fold. The hyperparameter search was configured to optimize the classifier’s E accuracy since TED-C’s primary purpose is to generate explanations. For SVM, we ran hyperparameter search for 50 iterations, whereas for the other three algorithms we ran the search for only 20 iterations since their hyperparameter spaces are not large. In each search iteration, an internal five-fold CV was conducted to generate an estimate of out-of-sample E accuracy. The remaining parts of this section discuss the experimental results.
5.1 Increasing the Number of Explanations
Under the TED-C approach, the set of possible explanations is enumerated and the size of the set translates directly into the number of classes that the classification algorithm must handle. It is natural to ask how performance may be affected as this number increases. We consider this question by increasing the number of explanations in the FICO and retention datasets beyond the base 9-explanation versions, as described in Section 4.1 and 4.2.
Figure 1 shows the impact of the number of possible explanations on both E and Y accuracy. The left two sets of graphs show that the accuracies of RF (brown) and LR (pink) stay relatively constant as the number of explanations increases. Impressively, RF was able to maintain almost 100% accuracy on both E and Y across all settings. On the other hand, LGB and SVM were unstable when the number of explanations is large, as evidenced by their large error bars. In some cases, their average E accuracy dropped to 50%. This is somewhat surprising given LGB’s state-of-the-art performance on many classification datasets,2 and SVM’s usual outperformance over LR. Figure 4 in Appendix B shows similar behavior on the retention dataset.
The rightmost graph in Figure 1 shows how the TED-C framework may affect Y accuracy under some settings. The graph shows the difference in Y accuracy between a model that is built to predict only Y and a model under the TEDC framework that is built to predict (Y, E) pairs. As can be seen, for all algorithms except LGB, the lines stay near 0%, suggesting that TED-C’s need to predict (Y, E) pairs does not affect its ability to predict Y . In fact, for LR, the accuracy has improved by as much as 3.2% with the introduction of E labels, suggesting that the E labels provided extra information for LR to predict Y . For LGB, however, a model that predicts only Y can sometimes outperform its TED-C counterpart by 40%. This again shows that LGB has a stability issue when there is a large number of classes.
5.2 Increasing Variability in Explanations
Figure 2 shows the effects on TED-C’s accuracy of adding the two types of noise described in Section 4.3 and parametrized by and
respectively. These are added sep-
Figure 3: The effect of missing training explanations on TED-C’s E and Y accuracy.
arately to the FICO dataset with 9 explanations and jointly to the retention dataset with 9 explanations. In addition to curves for the four classification algorithms, we also plot the performance of an “oracle” that knows the noiseless rules in Sections 4.1 and 4.2 for producing explanations and decisions, i.e., without having to learn them from data. The oracle is thus suggestive of the highest accuracy possible for a given noise level. Both RF and LGB perform very close to the oracle, with LGB not exhibiting the instability seen in Figure 1. LR is worse, as expected, due to its linearity but also degrades gracefully, and even closes the gap with respect to the oracle in the FICO plot. The performance of SVM remains uneven.
5.3 Reducing Required Explanations
This subsection explores scenarios where only a subset of the training data has explanations. We simulate this with the FICO and retention datasets by removing at random a fraction of the E labels from the training sets. We use the algorithm described in Section 3 to impute missing E labels. To simplify the experiment, the same classification algorithm (e.g. LGB, SVM) is used for imputation as for TED-C. For imputation, two separate models are trained for Y = 0 and Y = 1 (with X as input), instead of appending Y as an input alongside X. This takes advantage of the fact that Y is binary and restricts the set of possible E’s to those associated with either the positive or negative class.
Figure 3 shows the accuracies obtained as a function of the fraction of training E labels that are given. Remarkably, accuracies decrease only slightly as the fraction decreases to 0.1 (10% of training data have explanations) and moderately even at 0.01. The exception is SVM, which is again unstable. The rightmost plot shows that accuracies are maintained even in the presence of noise (). A possible explanation is that the explanation concepts in these datasets are logical and do not require many examples to learn.
The experimental results in Section 5 have shown that, insofar as human explanations and decisions follow logical rules, the TED-C approach can reproduce them with high accuracy. This is true even as the number of possible explanations increases to more than 30 or as the fraction of training data with explanation labels decreases below 10%. In the presence of noise, accuracies degrade gradually and nearly match those of an oracle that knows the noise-free generating process. Of the four classification algorithms compared, Random Forest stands out for achieving high accuracies and stability across the tested conditions, while the others suffer from lower accuracy (logistic regression), instability (LightGBM), or both (SVM). With the exception of LightGBM, accuracy in predicting class labels Y does not decrease with the addition of explanation labels E and may in fact improve.
We have considered the case where class and explanation labels are produced at the same time. As mentioned in Section 2, in the case where class labels Y have already been assigned by another entity or process, it can be challenging to give explanations E that are consistent with the class labels. Assuming that explanations can be added after the fact, whether consistent or not, TED-C can be applied in exactly the same way to these (Y, E) labels to make predictions. The open problem to be addressed is how to handle instances for which the predicted Y is inconsistent with the predicted E. One option is to use the partially supervised learning described in this work to impute explanations from the assigned labels.
Another area of future work is to pursue use cases where the training explanations are produced by humans rather than automatically constructed from rules. One possibility is to crowd-source explanations, although this falls short of the ideal of having explanations from true experts. Another scenario is to have a human produce labels and explanations based on rules written by experts. This can be useful in dealing with ambiguous or ill-defined rules because the human can apply their intuition to resolve the confusion. However, humans are also prone to mistakes, i.e., misapplying a rule, which could lead to lower quality training data. Either way, this will be an interesting scenario to consider in the future.
2019. AI Explainability 360 toolkit. http://aix360.mybluemix.net/.
Arya, V.; Bellamy, R. K. E.; Chen, P.-Y.; Dhurandhar, A.; Hind, M.; Hoffman, S. C.; Houde, S.; Liao, Q. V.; Luss, R.; Mojsilovi, A.; Mourad, S.; Pedemonte, P.; Raghaven-
dra, R.; Richards, J.; Sattigeri, P.; Shanmugam, K.; Singh, M.; Varshney, K. R.; Wei, D.; and Zhang, Y. 2019. One explanation does not fit all: A toolkit and taxonomy of AI explainability techniques. https://arxiv.org/abs/1909.03012.
Codella, N. C. F.; Hind, M.; Ramamurthy, K. N.; Campbell, M.; Dhurandhar, A.; Varshney, K. R.; Wei, D.; and Mojsilovic, A. 2019. Teaching AI to explain its decisions using embeddings and multi-task learning. In ICML Workshop on Human in the Loop Learning (HILL 2019).
Commission, E. 2017. Guidelines on automated individual decision-making and profiling for the purposes of regulation 2016/679. Technical report, Article 29 Data Protection Working Party, European Union.
Dash, S.; G¨unl¨uk, O.; and Wei, D. 2018. Boolean decision rules via column generation. In Proceedings of the 32nd International Conference on Neural Information Processing Systems (NeurIPS).
Falkner, S.; Klein, A.; and Hutter, F. 2018. BOHB: Robust and efficient hyperparameter optimization at scale. In Dy, J., and Krause, A., eds., Proceedings of the 35th International Conference on Machine Learning, volume 80 of Proceedings of Machine Learning Research, 1437–1446. Stockholmsmssan, Stockholm Sweden: PMLR.
FICO. 2018. Explainable machine learning challenge.
Heyman, R. E.; Lorber, M. F.; Eddy, J. M.; and West, T. V. 2014. Behavioral observation and coding. Cambridge University Press. 345–372.
Hind, M.; Wei, D.; Campbell, M.; Codella, N. C. F.; Dhurandhar, A.; Mojsilovic, A.; Ramamurthy, K. N.; and Varshney, K. R. 2019. TED: Teaching AI to explain its decisions. In Proceedings of the 2019 AAAI/ACM Conference on AI, Ethics, and Society (AIES), 123–129.
Hind, M. 2019. Explaining explainable AI. ACM XRDS Mag. 25(3):16–19.
Illinois. 2019. Public act 101-0260.
Ke, G.; Meng, Q.; Finley, T.; Wang, T.; Chen, W.; Ma, W.; Ye, Q.; and Liu, T.-Y. 2017. LightGBM: A highly effi-cient gradient boosting decision tree. In Guyon, I.; Luxburg, U. V.; Bengio, S.; Wallach, H.; Fergus, R.; Vishwanathan, S.; and Garnett, R., eds., Advances in Neural Information Processing Systems 30. Curran Associates, Inc. 3146–3154.
McDonnell, T.; Lease, M.; Kutlu, M.; and Elsayed, T. 2016. Why is that relevant? Collecting annotator rationales for relevance judgments. In Proc. AAAI Conf. Human Comput. Crowdsourc.
Ratner, A.; Bach, S. H.; Ehrenberg, H.; Fries, J.; Wu, S.; and R, C. 2017. Snorkel. Proceedings of the VLDB Endowment 11(3):269282.
Varshney, K. R.; Rasmussen, J. C.; Mojsilovi, A.; Singh, M.; and DiMicco, J. M. 2012. Interactive visual salesforce analytics. In Thirty Third International Conference on Information Systems.
Zaidan, O. F., and Eisner, J. 2007. Using ‘annotator rationales’ to improve machine learning for text categorization. In In NAACL-HLT, 260–267.
Zaidan, O. F., and Eisner, J. 2008. Modeling annotators: A generative approach to learning from annotator rationales. In Proceedings of EMNLP 2008, 31–40.
Zhang, Y.; Marshall, I. J.; and Wallace, B. C. 2016. Rationale-augmented convolutional neural networks for text classification. In Conference on Empirical Methods in Natural Language Processing (EMNLP).
A Additional FICO Dataset Details The original FICO challenge dataset contains three kinds of special values: (no record),
(no usable/valid trades or inquiries), and
(condition not met). First we removed 588 rows where all features have value
and are thus not usable, leaving 9871 rows. After removing the ExternalRiskEstimate feature as mentioned in Section 4.1, no
values remain in the data. Next, only two features have
values: MSinceMostRecentDelq and MSinceMostRe-centInqexcl7days. By analyzing the rates of repayment as a function of these two features, we concluded that MSinceMostRecentDelq
most likely means that the applicant did not have a delinquency in the past 7 years (and hence “condition was not met”), while MSinceMostRe-centInqexcl7days
means that the applicant in fact had a credit inquiry in the last 7 days, which was excluded. Hence we replaced MSinceMostRecentDelq
with 84 months (7 years) and MSinceMostRecentInqexcl7days
with 0 months. The remaining special values are all
, which we replaced with the generic np.nan null value in NumPy. Some classification algorithms (e.g. LGB) can handle null values while for LR and SVM, null values were further imputed using the median value of the feature. Lastly, we were able to infer that all cases of MaxDelq2PublicRecLast12M > 7 (“other values”) should actually be equal to 7 (current and never delinquent in last 12 months) based on the corresponding value of MaxDelqEver, which implied that the applicant was never delinquent. Below we show the exact rule set learned by BRCG with regularization parameters
, which differs from the rule set in Section 4.1 only in the threshold values.
1. NetFractionRevolvingBurden AND PercentTradesNeverDelq > 86 AND AverageMInFile > 52 AND MaxDelq2PublicRecLast12M > 5; OR
2. NetFractionRevolvingBurden AND MSinceMostRecentInqexcl7days > 24.
B Additional Experimental Details and Results Table 2 lists the hyperparameters that were optimized for each classification algorithm. Figure 4 shows the effect of the number of explanation choices on TED-C’s E and Y accuracy for the retention dataset.
Table 2: Hyperparameter search space definitions. Hyperparameter Search space
Figure 4: The effect of the number of explanations on TED-C’s E and Y accuracy for the retention dataset. The rightmost graph shows the Y accuracy difference between a model that predicts only Y and a model that predicts (Y, E) pairs.