The xgboost model itself computes a notion of feature importance: import mlflow.
sklearn best_run_id = “.
” model = mlflow.
load_model(“runs:/” + best_run_id + “/xgboost”) sorted((zip(model.
columns)), reverse=True)[:6] Factors like years of coding professionally, organization size, and using Windows are most “important”.
This is interesting, but hard to interpret.
The values reflect relative and not absolute importance.
That is, the effect isn’t measured in dollars.
The definition of importance here (total gain) is also specific to how decision trees are built and are hard to map to an intuitive interpretation.
The important features don’t even necessarily correlate positively with salary, either.
More importantly, this is a ‘global’ view of how much features matter in aggregate.
Factors like gender and ethnicity don’t show up on this list until farther along.
This doesn’t mean these factors aren’t still significant.
For one, features can be correlated, or interact.
It’s possible that factors like gender correlate with other features that the trees selected instead, and this to some degree masks their effect.
The more interesting question is not so much whether these factors matter overall — it’s possible that their average effect is relatively small — but, whether they have a significant effect in some individual cases.
These are the instances where the model is telling us something important about individuals’ experience, and to those individuals, that experience is what matters.
Applying SHAP for Developer-Level Explanations Fortunately, a set of techniques for more theoretically sound model interpretation at the individual prediction level has emerged over the past five years or so.
They are collectively “Shapley Additive Explanations”, and conveniently, are implemented in the Python package shap.
Given any model, this library computes “SHAP values” from the model.
These values are readily interpretable, as each value is a feature’s effect on the prediction, in its units.
A SHAP value of 1000 here means “explained +$1,000 of predicted salary”.
SHAP values are computed in a way that attempts to isolate away of correlation and interaction, as well.
import shap explainer = shap.
TreeExplainer(model) shap_values = explainer.
values) SHAP values are also computed for every input, not the model as a whole, so these explanations are available for each input individually.
It can also estimate the effect of feature interactions separately from the main effect of each feature, for each prediction.
Explaining the Features’ Effects Overall Developer-level explanations can aggregate into explanations of the features’ effects on salary over the whole data set by simply averaging their absolute values.
SHAP’s assessment of the overall most important features is similar: The SHAP values tell a similar story.
First, SHAP is able to quantify the effect on salary in dollars, which greatly improves the interpretation of the results.
Above is a plot the absolute effect of each feature on predicted salary, averaged across developers.
Years of professional coding experience still dominates, explaining on average almost $15,000 of effect on salary.
Examining the Effects of Gender with SHAP Values We came to look specifically at the effects of gender, race, and other factors that presumably should not be predictive per se of salary at all.
This example will examine the effect of gender, though this by no means suggests that it’s the only or most important, type of bias to look for.
Gender is not binary, and the survey recognizes responses of “Man”, “Woman”, and “Non-binary, genderqueer, or gender non-conforming” as well as “Trans” separately.
(Note that while the survey also separately records responses about sexuality, these are not considered here.
) SHAP computes the effect on predicted salary for each of these.
For a male developer (identifying only as male), the effect of gender is not just the effect of being male, but of not identifying as female, transgender, and so on.
SHAP values let us read off the sum of these effects for developers identifying as each of the four categories: While male developers’ gender explains about a modest -$230 to +$890 with mean about $225, for females, the range is wider, from about -$4,260 to -$690 with mean -$1,320.
The results for transgender and non-binary developers is similar, though slightly less negative.
When evaluating what this means below, it’s important to recall the limitations of the data and model here: Correlation isn’t causation; ‘explaining’ predicted salary is suggestive, but doesn’t prove, that a feature directly caused salary to be higher or lower The model isn’t perfectly accurate This is just 1 year of data, and only from US developers This reflects only base salary, not bonuses or stock, which can vary more widely Gender and Interacting Features The SHAP library offers interesting visualizations that leverage its ability to isolate the effect of feature interactions.
For example, the values above suggest that developers who identify as male are predicted to earn a slightly higher salary than others, but is there more to it?.A dependence plot like this one can help: Dots are developers.
Developers at the left are those that don’t identify as male, and at the right, those that do, which are predominantly those identifying as only male.
(The points are randomly spread horizontally for clarity.
) The y-axis is SHAP value, or what identifying as male or not explains about predicted salary for each developer.
As above, those not identifying as male show overall negative SHAP values, and one that varies widely, while others consistently show a small positive SHAP value.
What’s behind that variance?.SHAP can select a second feature whose effect varies most given the value of, here, identifying as male or not.
It selects the answer “I work on what seems most important or urgent” to the question “How structured or planned is your work?” Among developers identifying as male, those who answered this way (red points) appear to have slightly higher SHAP values.
Among the rest, the effect is more mixed but seems to have generally lower SHAP values.
Interpretation is left to the reader, but perhaps: are male developers who feel empowered in this sense also enjoying slightly higher salaries, while other developers enjoy this where it goes hand in hand with lower-paying roles?.Exploring Instances with Outsized Gender Effects What about investigating the developer whose salary is most negatively affected?.Just as it’s possible to look at the effect of gender-related features overall, it’s possible to search for the developer whose gender-related features had the largest impact on predicted salary.
This person is female, and the effect is negative.
According to the model, she is predicted to earn about $4,260 less per year because of her gender: The predicted salary, just over $157,000, accurate in this case, as her actual reported salary is $150,000.
The three most positive and negative features influencing predicted salary are that she: Has a college degree (only) (+$18,200) Has 10 years professional experience (+$9,400) Identifies as East Asian (+$9,100) … Works 40 hours per week (-$4,000) Does not identify as male (-$4,250) Works at a medium-sized org of 100-499 employees (-$9,700) Given the magnitude of the effect on the predicted salary of not identifying as male, we might stop here and investigate the details of this case offline to gain a better understanding of the context around this developer and whether her experience, or salary, or both, need a change.
Explaining Interactions There is more detail available within that -$4,260.
SHAP can break down the effects of these features into interactions.
The total effect of identifying as female on the prediction can be broken down into the effect of identifying as female and being an engineering manager, and working with Windows, etc.
The effect on predicted salary explained by the gender factors per se only adds up to about -$630.
Rather, SHAP assigns most of the effects of gender to interactions with other features: gender_interactions = interaction_values[gender_feature_locs].
sum(axis=0) max_c = np.
argmax(gender_interactions) min_c = np.
columns[max_c]) print(gender_interactions[max_c]) print(X.
columns[min_c]) print(gender_interactions[min_c]) DatabaseWorkedWith_PostgreSQL 110.
64005 Ethnicity_East_Asian -1372.
6714 Identifying as female and working with PostgreSQL affects predicted salary slightly positively, whereas also identifying as East Asian predicted salary more negatively.
Interpreting these values at this level of granularity is difficult in this context, but, this additional level of explanation is available.
Applying SHAP with Apache Spark SHAP values are computed independently for each row, given the model, and so this could have also been done in parallel with Spark.
The following example computes SHAP values in parallel and similarly locates developers with outsized gender-related SHAP values: X_df = pruned_parsed_df.
repartition(16) X_columns = X_df.
columns def add_shap(rows): rows_pd = pd.
DataFrame(rows, columns=X_columns) shap_values = explainer.
drop([“Respondent”], axis=1)) return [Row(*([int(rows_pd[“Respondent”][i])] + [float(f) for f in shap_values[i]])) for i in range(len(shap_values))] shap_df = X_df.
toDF(X_columns) effects_df = shap_df.
withColumn(“gender_shap”, col(“Gender_Woman”) + col(“Gender_Man”) + col(“Gender_Non_binary__genderqueer__or_gender_non_conforming”) + col(“Trans”)).
select(“Respondent”, “gender_shap”) top_effects_df = effects_df.
filter(abs(col(“gender_shap”)) >= 2500).
orderBy(“gender_shap”) Clustering SHAP values Applying Spark is advantageous when there are a large number of predictions to assess with SHAP.
Given that output, it’s also possible to use Spark to cluster the results with, for example, bisecting k-means: assembler = VectorAssembler(inputCols=[c for c in to_review_df.
columns if c != “Respondent”], outputCol=”features”) assembled_df = assembler.
cache() clusterer = BisectingKMeans().
setSeed(0) cluster_model = clusterer.
fit(assembled_df) transformed_df = cluster_model.
select(“Respondent”, “prediction”) The cluster whose total gender-related SHAP effects are most negative might bear some further investigation.
What are the SHAP values of those respondents in the cluster?.What do the members of the cluster look like with respect to the overall developer population?.min_shap_cluster_df = transformed_df.
filter(“prediction = 5”).
orderBy(gender_cols) all_shap_df = X_df.
orderBy(gender_cols) expected_ratio = transformed_df.
filter(“prediction = 5”).
count() / X_df.
withColumn(“ratio”, (min_shap_cluster_df[“count”] / all_shap_df[“count”]) / expected_ratio).
orderBy(“ratio”)) Developers identifying as female (only) are represented in this cluster at almost 2.
8x the rate of the overall developer population, for example.
This isn’t surprising given the earlier analysis.
This cluster could be further investigated to assess other factors specific to this group that contribute to overall lower predicted salary.
Conclusion This type of analysis with SHAP can be run for any model, and at scale too.
As an analytical tool, it turns models into data detectives, to surface individual instances whose predictions suggest that they deserve more examination.
The output of SHAP is easily interpretable and yields intuitive plots, that can be assessed case-by-case by business users.
Of course, this analysis isn’t limited to examining questions of gender, age or race bias.
More prosaically, it could be applied to customer churn models.
There, the question is not just “will this customer churn?” but “why is the customer churning?” A customer who is canceling due to price may be offered a discount, while one canceling due to limited usage might need an upsell.
Finally, this analysis can be run as part of a model validation process.
Model validation often focuses on the overall accuracy of a model.
It should also focus on the model’s ‘reasoning’, or what features contributed most to the predictions.
With SHAP, it can also help detect when too many individual predictions’ explanations are at odds with overall feature importance.
Try Databricks for free.
Get started today.