Algorithms for mitigating unfairness in supervised machine learning
Project description
Fairlearn
Fairlearn is a Python package that empowers developers of artificial intelligence (AI) systems to assess their system's fairness and mitigate any observed unfairness issues. Fairlearn contains mitigation algorithms as well as a Jupyter widget for model assessment. Besides the source code, this repository also contains Jupyter notebooks with examples of Fairlearn usage.
- Current release
- What we mean by fairness
- Overview of Fairlearn
- Install Fairlearn
- Usage
- Contributing
- Maintainers
- Issues
Current release
-
The current stable release is available at Fairlearn v0.4.6.
-
Our current version differs substantially from version 0.2 or earlier. Users of these older versions should visit our onboarding guide.
What we mean by fairness
An AI system can behave unfairly for a variety of reasons. In Fairlearn, we define whether an AI system is behaving unfairly in terms of its impact on people – i.e., in terms of harms. We focus on two kinds of harms:
-
Allocation harms. These harms can occur when AI systems extend or withhold opportunities, resources, or information. Some of the key applications are in hiring, school admissions, and lending.
-
Quality-of-service harms. Quality of service refers to whether a system works as well for one person as it does for another, even if no opportunities, resources, or information are extended or withheld.
We follow the approach known as group fairness, which asks: Which groups of individuals are at risk for experiencing harms? The relevant groups need to be specified by the data scientist and are application specific.
Group fairness is formalized by a set of constraints, which require that some aspect (or aspects) of the AI system's behavior be comparable across the groups. The Fairlearn package enables assessment and mitigation of unfairness under several common definitions. To learn more about our definitions of fairness, please visit our user guide on Fairness of AI Systems.
Note: Fairness is fundamentally a sociotechnical challenge. Many aspects of fairness, such as justice and due process, are not captured by quantitative fairness metrics. Furthermore, there are many quantitative fairness metrics which cannot all be satisfied simultaneously. Our goal is to enable humans to assess different mitigation strategies and then make trade-offs appropriate to their scenario.
Overview of Fairlearn
The Fairlearn package has two components:
-
A dashboard for assessing which groups are negatively impacted by a model, and for comparing multiple models in terms of various fairness and accuracy metrics.
-
Algorithms for mitigating unfairness in a variety of AI tasks and along a variety of fairness definitions.
Fairlearn algorithms
Fairlearn contains the following algorithms for mitigating unfairness in binary classification and regression:
algorithm | description | classification/regression | sensitive features | supported fairness definitions |
---|---|---|---|---|
fairlearn. reductions. ExponentiatedGradient |
Black-box approach to fair classification described in A Reductions Approach to Fair Classification | binary classification | categorical | DP, EO |
fairlearn. reductions. GridSearch |
Black-box approach described in Section 3.4 of A Reductions Approach to Fair Classification | binary classification | binary | DP, EO |
fairlearn. reductions. GridSearch |
Black-box approach that implements a grid-search variant of the algorithm described in Section 5 of Fair Regression: Quantitative Definitions and Reduction-based Algorithms | regression | binary | BGL |
fairlearn. postprocessing. ThresholdOptimizer |
Postprocessing algorithm based on the paper Equality of Opportunity in Supervised Learning. This technique takes as input an existing classifier and the sensitive feature, and derives a monotone transformation of the classifier's prediction to enforce the specified parity constraints. | binary classification | categorical | DP, EO |
Note: DP refers to demographic parity, EO to equalized odds, and BGL to bounded group loss. For more information on these and other terms we use in this repository please refer to the terminology page. To request additional algorithms or fairness definitions, please open a new issue.
Fairlearn dashboard
Fairlearn dashboard is a Jupyter notebook widget for assessing how a model's predictions impact different groups (e.g., different ethnicities), and also for comparing multiple models along different fairness and accuracy metrics.
Set-up and a single-model assessment
To assess a single model's fairness and accuracy, the dashboard widget can be launched within a Jupyter notebook as follows:
from fairlearn.widget import FairlearnDashboard
# A_test containts your sensitive features (e.g., age, binary gender)
# sensitive_feature_names containts your sensitive feature names
# y_true contains ground truth labels
# y_pred contains prediction labels
FairlearnDashboard(sensitive_features=A_test,
sensitive_feature_names=['BinaryGender', 'Age'],
y_true=Y_test.tolist(),
y_pred=[y_pred.tolist()])
After the launch, the widget walks the user through the assessment set-up, where the user is asked to select (1) the sensitive feature of interest (e.g., binary gender or age), and (2) the accuracy metric (e.g., model precision) along which to evaluate the overall model performance as well as any disparities across groups. These selections are then used to obtain the visualization of the model's impact on the subgroups (e.g., model precision for females and model precision for males).
The following figures illustrate the set-up steps, where binary gender is selected as a sensitive feature and accuracy rate is selected as the accuracy metric.
After the set-up, the dashboard presents the model assessment in two panels:
Panel | Description |
---|---|
Disparity in accuracy | This panel shows: (1) the accuracy of your model with respect to your selected accuracy metric (e.g., accuracy rate) overall as well as on different subgroups based on your selected sensitive feature (e.g., accuracy rate for females, accuracy rate for males); (2) the disparity (difference) in the values of the selected accuracy metric across different subgroups; (3) the distribution of errors in each subgroup (e.g., female, male). For binary classification, the errors are further split into overprediction (predicting 1 when the true label is 0), and underprediction (predicting 0 when the true label is 1). |
Disparity in predictions | This panel shows a bar chart that contains the selection rate in each group, meaning the fraction of data classified as 1 (in binary classification) or distribution of prediction values (in regression). |
Comparing multiple models
The dashboard also enables comparison of multiple models, such as the models produced by different learning algorithms and different mitigation approaches, including fairlearn.reductions.GridSearch
, fairlearn.reductions.ExponentiatedGradient
and fairlearn.postprocessing.ThresholdOptimizer
.
As before, the user is first asked to select the sensitive feature and the accuracy metric. The model comparison view then depicts the accuracy and disparity of all the provided models in a scatter plot. This allows the user to examine trade-offs between accuracy and fairness. Each of the dots can be clicked to open the assessment of the corresponding model. The figure below shows the model comparison view with binary gender
selected as a sensitive feature and accuracy rate
selected as the accuracy metric.
Install Fairlearn
The package can be installed via
pip install fairlearn
or optionally with a full feature set by adding extras, e.g.
pip install fairlearn[customplots]
.
or you can clone the repository locally via
git clone git@github.com:fairlearn/fairlearn.git
To verify that the cloned repository works (the pip package does not include the tests), run
pip install -r requirements.txt
python -m pytest -s ./test/unit
Onboarding guide for users of version 0.2 or earlier
Up to version 0.2, Fairlearn contained only the exponentiated gradient method. The Fairlearn repository now has a more comprehensive scope and aims to incorporate other methods as specified above. The same exponentiated gradient technique is now the class fairlearn.reductions.ExponentiatedGradient
. While in the past exponentiated gradient was invoked via
import numpy as np
from fairlearn.classred import expgrad
from fairlearn.moments import DP
estimator = LogisticRegression() # or any other estimator
exponentiated_gradient_result = expgrad(X, sensitive_features, y, estimator, constraints=DP())
positive_probabilities = exponentiated_gradient_result.best_classifier(X)
randomized_predictions = (positive_probabilities >= np.random.rand(len(positive_probabilities))) * 1
the equivalent operation is now
from fairlearn.reductions import ExponentiatedGradient, DemographicParity
estimator = LogisticRegression() # or any other estimator
exponentiated_gradient = ExponentiatedGradient(estimator, constraints=DemographicParity())
exponentiated_gradient.fit(X, y, sensitive_features=sensitive_features)
randomized_predictions = exponentiated_gradient.predict(X)
Please open a new issue if you encounter any problems.
Usage
For common usage refer to the Jupyter notebooks and our API guide
Contributing
To contribute please check our contributing guide.
Maintainers
The Fairlearn project is maintained by:
- @MiroDudik
- @riedgar-ms
- @rihorn2
- @romanlutz
For a full list of contributors refer to the authors page
Issues
Usage Questions
Pose questions and help answer them on Stack
Overflow with the tag
fairlearn
or on Gitter.
Regular (non-security) issues
Issues are meant for bugs, feature requests, and documentation improvements. Please submit a report through GitHub issues. A maintainer will respond promptly as appropriate.
Maintainers will try to link duplicate issues when possible.
Reporting security issues
Please take a look at our guidelines for reporting security issues.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file fairlearn-0.4.6.tar.gz
.
File metadata
- Download URL: fairlearn-0.4.6.tar.gz
- Upload date:
- Size: 10.5 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/40.6.2 requests-toolbelt/0.9.1 tqdm/4.46.0 CPython/3.6.10
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c2082419a63c1e6cd1965736e4c65bf6cea8e1d04da835e6b2d1db0c422b929b |
|
MD5 | 3d5a5838d483ecd9b6985e251f031e97 |
|
BLAKE2b-256 | 68bb117218a02ac877e1b22bb1a8eda25948c4fc0a97f301294163a264ed8806 |
File details
Details for the file fairlearn-0.4.6-py3-none-any.whl
.
File metadata
- Download URL: fairlearn-0.4.6-py3-none-any.whl
- Upload date:
- Size: 21.2 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/40.6.2 requests-toolbelt/0.9.1 tqdm/4.46.0 CPython/3.6.10
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 03a794321a0e5d22996f0076e5530cd390407312b0fa16bc24a3d5164385fe78 |
|
MD5 | 8bf7b134d63fc068799baf62e183b247 |
|
BLAKE2b-256 | c0ec15764c20d27f4ec53e826eff160139c937274b646491c4de936a73444fd8 |