Need help with audit-ai?

Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

272 Stars 42 Forks MIT License 48 Commits 2 Opened issues

detect demographic differences in the output of machine learning models or other assessments

Readme

Open Sourced Bias Testing for Generalized Machine Learning Applications

audit-AIis a Python library built on top of

pandasand

sklearnthat implements fairness-aware machine learning algorithms.

audit-AIwas developed by the Data Science team at pymetrics

audit-AIis a tool to measure and mitigate the effects of discriminatory patterns in training data and the predictions made by machine learning algorithms trained for the purposes of socially sensitive decision processes.

The overall goal of this research is to come up with a reasonable way to think about how to make machine learning algorithms more fair. While identifying potential bias in training datasets and by consequence the machine learning algorithms trained on them is not sufficient to solve the problem of discrimination, in a world where more and more decisions are being automated by Artificial Intelligence, our ability to understand and identify the degree to which an algorithm is fair or biased is a step in the right direction.

According to the Uniform Guidelines on Employee Selection Procedures (UGESP; EEOC et al., 1978), all assessment tools should comply to fair standard of treatment for all protected groups. Audit-ai extends this to machine learning methods. Let's say we build a model that makes some prediction about people. This model could theoretically be anything -- a prediction of credit scores, the likelihood of prison recidivism, the cost of a home loan, etc. Audit-ai takes data from a known population (e.g., credit information from people of multiple genders and ethnicities), and runs them through the model in question. The proportional pass rates of the highest-passing demographic group are compared to the lowest-passing group for each demographic category (gender and ethnicity). This proportion is known as the bias ratio.

Audit-ai determines whether groups are different according to a standard of statistical significance (within a statistically different margin of error) or practical significance (whether a difference is large enough to matter on a practical level). The exact threshold of statistical and practical significance depends on the field and use-case. Within the hiring space, the EEOC often uses a statistical significance of p < .05 to determine bias, and a bias ratio below the 4/5ths rule to demonstrate practical significance.

The 4/5ths rule effectively states that the lowest-passing group has to be within 4/5ths of the pass rate of the highest-passing group. Consider an example with 4,000 users, 1,000 of each of the following groups: Asian, Black, Hispanic/Latino, and White, who pass at a frequency of 250, 270, 240 and 260 users, respectively. The highest and lowest passing groups are Black (27%) and Hispanic/Latino (24%), respectively. The bias ratio is therefore 24/27 or .889. As this ratio is greater than .80 (4/5ths), the legal requirement enforced by the EEOC, the model would pass the check for practical significance. Likewise, a chi-squared test (a common statistical test for count data) would report that these groups are above the p = .05 threshold, and therefore pass the check for statistical significance.

Audit-ai also offers tools to check for differences over time or across different regions, using the Cochran-Mantel-Hanzel test, a common test in regulatory circles. To our knowledge this is the first implementation of this measure in an open-source python format.

Here are a few of the bias testing and algorithm auditing techniques that this library implements.

- 4/5th, fisher, z-test, bayes factor, chi squared
- sim
*beta*ratio, classifier*posterior*probabilities

- anova
- 4/5th, fisher, z-test, bayes factor, chi squared
- group proportions at different thresholds

The source code is currently hosted on GitHub: https://github.com/pymetrics/audit-ai

You can install the latest released version with

pip.

# pip pip install audit-AI

If you install with pip, you'll need to install scikit-learn, numpy, and pandas with either pip or conda. Version requirements:

- numpy
- scipy
- pandas

For vizualization: - matplotlib - seaborn

See our implementation paper here: https://github.com/pymetrics/audit-ai/blob/master/examples/implementation_suggestions.md

from auditai.misc import bias_test_checkX = df.loc[:,features] y_pred = clf.predict_proba(X)

## test for bias

bias_test_check(labels=df['gender'], results=y_pred, category='Gender')

>>>

Gender passes 4/5 test, Fisher p-value, Chi-Squared p-value, z-test p-value and Bayes Factor at 50.00

To get a plot of the different tests at different thresholds:

from auditai.viz import plot_threshold_testsX = df.loc[:,features] y_pred = clf.predict_proba(X)

## test for bias

plot_threshold_tests(labels=df['gender'], results=y_pred, category='Gender')