A comprehensive framework for automatic bias mitigation in AI systems
Project description
FairHealthGrid: A Systematic Framework for Evaluating Bias Mitigation Strategies in Healthcare Machine Learning
Paper Link* | Slides (soon)
Table of Contents:
Overview*
The integration of machine learning (ML) into healthcare demands rigorous fairness assurance to prevent algorithmic biases from exacerbating disparities in treatment. This study introduces FairHealthGrid, a systematic framework for evaluating bias mitigation strategies across healthcare ML models. Our framework combines grid search with a composite fairness score, incorporating fairness metrics weighted by risk tolerances. As output, we present a trade-off map concurrently evaluating accuracy and fairness, categorizing solutions (model + bias mitigation) into the following regions: Win-Win, Good, Poor, Inverted, or Lose-Lose. We apply the framework on three different healthcare datasets. Results reveal significant variability across different healthcare applications. The framework identifies model-bias mitigation for balancing equity and accuracy, yet highlights the absence of universal solutions. By enabling systematic trade-off analysis, FairHealthGrid allows healthcare stakeholders to audit, compare, and select ethically aligned ML models for specific healthcare applications—advancing towards equitable AI in healthcare.
News
2025/1/6 Our paper got accepted on AMCIS 2025.
Getting Started*
To install and use FairHealthGrid, follow these steps:
1. Installation
You can install the package directly from PyPI:
pip install callmefair
2. Documentation
For detailed documentation and usage examples, visit: CallMeFair Documentation
Results
Key Findings
Our evaluation across three healthcare datasets reveals that preprocessing and in-processing strategies offer the best balance between fairness and accuracy, while multi-stage bias mitigation approaches risk over-correction. The framework demonstrates significant variability in bias mitigation effectiveness across different healthcare applications, highlighting the absence of universal solutions.
Main Contributions
We introduce FairHealthGrid, a comprehensive framework that addresses critical gaps in healthcare AI fairness evaluation:
- Systematic Evaluation Framework: Exhaustive grid search optimization for model-strategy combinations in healthcare AI
- Unified Fairness Score (𝓕): Integrates five fairness metrics with risk tolerances for comprehensive bias assessment
- Interpretable Trade-off Maps: Context-driven model selection through fairness-accuracy visualization
Framework Benefits
- Model-agnostic design enabling evaluation across diverse ML architectures
- Context-specific evaluations prioritizing datasets with sensitive attributes
- Actionable insights for clinical deployment decisions
- Systematic methodology for selecting appropriate bias mitigation strategies
Limitations & Future Work
Current evaluations focus on single protected attributes. Future research will address intersectional bias from interacting sensitive features, expand to new data modalities, and formalize a comprehensive bias taxonomy using causal fairness frameworks.
Figure 1: Proposed approach consists of three main stages: (A) Grid Search, where multiple machine learning models (ML) and bias mitigation (BM) strategies are evaluated; (B) Computation of the Combined Fairness Score (𝓕), aggregating multiple fairness measures to quantify bias; and (C) Fairness Trade-off Mapping, which utilizes the baseline fairness score 𝓕baseline and naive mutation strategies (𝓕10–𝓕100) to generate a trade-off curve. The final trade-off map categorizes models into trade-off regions.
Citation
If you find our project useful for your research, please consider citing our paper and codebase with the following BibTeX:
@proceedings{bento2022deep,
title={FairHealthGrid: A Systematic Framework for Evaluating Bias Mitigation Strategies in Healthcare Machine Learning},
author={Paiva, Pedro and Dehghani, Farzaneh and Anzum, Fahim and Singhal, Mansi and Metwali, Ayah and Gavrilova, Marina and Bento, Mariana},
year={2025},
journal={31st Americas Conference on Information Systems, {AMCIS} 2025},
publisher = {Association for Information Systems},
}
License
In the License section of your GitHub project or research paper, include the type of license under which your code or content is released. Briefly explain the permissions and limitations associated with that license. Common licenses include MIT, GPL, or Apache.
Related Resources
- Related Literature (ai2lab)
- Web Content
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
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file callmefair-0.0.4.tar.gz.
File metadata
- Download URL: callmefair-0.0.4.tar.gz
- Upload date:
- Size: 32.6 MB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
59c1b37312bd00154a3efc098a54c2414f61cd676988f0133acb1affe3621d59
|
|
| MD5 |
faab017b00c0197014ce08361e30e92a
|
|
| BLAKE2b-256 |
adca9ffb621e967b5ded002a1c397b44f040b10bb92611467422f1a66a389471
|
Provenance
The following attestation bundles were made for callmefair-0.0.4.tar.gz:
Publisher:
release.yaml on enemy537/callmefair
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
callmefair-0.0.4.tar.gz -
Subject digest:
59c1b37312bd00154a3efc098a54c2414f61cd676988f0133acb1affe3621d59 - Sigstore transparency entry: 269473581
- Sigstore integration time:
-
Permalink:
enemy537/callmefair@a0101afa086fa0f39896baad7dac38742ee9cc2d -
Branch / Tag:
refs/tags/0.0.4 - Owner: https://github.com/enemy537
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yaml@a0101afa086fa0f39896baad7dac38742ee9cc2d -
Trigger Event:
push
-
Statement type:
File details
Details for the file callmefair-0.0.4-py3-none-any.whl.
File metadata
- Download URL: callmefair-0.0.4-py3-none-any.whl
- Upload date:
- Size: 31.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
31932fd1d9ac32908c4060a24fa61d0a8c77825a2b5c17eb5c908900ab37bcd7
|
|
| MD5 |
cb931508b725c4cbe1c87a818868f911
|
|
| BLAKE2b-256 |
d951d43ccc4ffc01ed78c52f59be28cff9cc0ecbf2fc43265c5ceaf7b03f53a4
|
Provenance
The following attestation bundles were made for callmefair-0.0.4-py3-none-any.whl:
Publisher:
release.yaml on enemy537/callmefair
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
callmefair-0.0.4-py3-none-any.whl -
Subject digest:
31932fd1d9ac32908c4060a24fa61d0a8c77825a2b5c17eb5c908900ab37bcd7 - Sigstore transparency entry: 269473606
- Sigstore integration time:
-
Permalink:
enemy537/callmefair@a0101afa086fa0f39896baad7dac38742ee9cc2d -
Branch / Tag:
refs/tags/0.0.4 - Owner: https://github.com/enemy537
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yaml@a0101afa086fa0f39896baad7dac38742ee9cc2d -
Trigger Event:
push
-
Statement type: