Skip to main content

Achieve error-rate parity between protected groups for any predictor

Project description

error-parity

Tests status PyPI status PyPI version OSI license Python compatibility

Fast postprocessing of any score-based predictor to meet fairness criteria.

The error-parity package can achieve strict or relaxed fairness constraint fulfillment, which can be useful to compare ML models at equal fairness levels.

Installing

Install package from PyPI:

pip install error-parity

Or, for development, you can clone the repo and install from local sources:

git clone https://github.com/socialfoundations/error-parity.git
pip install ./error-parity

Getting started

See detailed example notebooks under the examples folder.

from error_parity import RelaxedThresholdOptimizer

# Given any trained model that outputs real-valued scores
fair_clf = RelaxedThresholdOptimizer(
    predictor=lambda X: model.predict_proba(X)[:, -1],   # for sklearn API
    # predictor=model,  # use this for a callable model
    constraint="equalized_odds",
    tolerance=0.05,     # fairness constraint tolerance
)

# Fit the fairness adjustment on some data
# This will find the optimal _fair classifier_
fair_clf.fit(X=X, y=y, group=group)

# Now you can use `fair_clf` as any other classifier
# You have to provide group information to compute fair predictions
y_pred_test = fair_clf(X=X_test, group=group_test)

How it works

Given a callable score-based predictor (i.e., y_pred = predictor(X)), and some (X, Y, S) data to fit, RelaxedThresholdOptimizer will:

  1. Compute group-specific ROC curves and their convex hulls;
  2. Compute the r-relaxed optimal solution for the chosen fairness criterion (using cvxpy);
  3. Find the set of group-specific binary classifiers that match the optimal solution found.
    • each group-specific classifier is made up of (possibly randomized) group-specific thresholds over the given predictor;
    • if a group's ROC point is in the interior of its ROC curve, partial randomization of its predictions may be necessary.

Features and implementation road-map

We welcome community contributions for cvxpy implementations of other fairness constraints.

Currently implemented fairness constraints:

  • equality of odds (Hardt et al., 2016);
    • i.e., equal group-specific TPR and FPR;
    • use constraint="equalized_odds";
  • equal opportunity;
    • i.e., equal group-specific TPR;
    • use constraint="true_positive_rate_parity";
  • predictive equality;
    • i.e., equal group-specific FPR;
    • use constraint="false_positive_rate_parity";

Road-map:

  • demographic parity;
    • i.e., equal group-specific predicted prevalence;

Citing

This repository contains code and supplementary materials for the following preprint:

André F. Cruz and Moritz Hardt. "Unprocessing Seven Years of Algorithmic Fairness." arXiv preprint, 2023.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

error-parity-0.3.7.tar.gz (35.6 kB view details)

Uploaded Source

Built Distribution

error_parity-0.3.7-py3-none-any.whl (39.1 kB view details)

Uploaded Python 3

File details

Details for the file error-parity-0.3.7.tar.gz.

File metadata

  • Download URL: error-parity-0.3.7.tar.gz
  • Upload date:
  • Size: 35.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.18

File hashes

Hashes for error-parity-0.3.7.tar.gz
Algorithm Hash digest
SHA256 fa9f40af9989ffa7f1ce5ea46e166e5bc7f317605df4e576791678e676c80451
MD5 0c9b7a8ca7e2189900847618d971aa22
BLAKE2b-256 1d80f23f4c4b2cbcfa3d647a500a42d38a2f5ab38629378eeccbb59109f84c86

See more details on using hashes here.

File details

Details for the file error_parity-0.3.7-py3-none-any.whl.

File metadata

  • Download URL: error_parity-0.3.7-py3-none-any.whl
  • Upload date:
  • Size: 39.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.18

File hashes

Hashes for error_parity-0.3.7-py3-none-any.whl
Algorithm Hash digest
SHA256 6e9a5f595900a58d7b00bbea3e158431bb4fc151e7b162b4fd0a49e5334f2f5f
MD5 490fe6cefa105c5849faad58a356cbaa
BLAKE2b-256 d277087d6827c28a387eb5aa5449fa309f6264c8c93aaa5282c6c907fc7d165f

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page