# Distraction Is All You Need For Fairness
This is The repository for supporting matterial of "Distraction is All You Need for Fairness" [link](https://arxiv.org/abs/2203.07593)

## Architecture
![Distraction](Assets/ModelDiagram.png)

## Training Algorithm
![Algo](Assets/Algo.png)
## Requirements
For installing the requirements run:
```
pip install -r requirements.txt
```
## Replicating the results
For running the model
run:
```
python ablation_study.py --eta --data --fairness_layers --acc_layers --fairness_position --device --epochs --fairness_learning_rate --network_learning_rate --batch_size
```

For running the model with distraction module run:
### UCI Adult dataset:
```
python attribute_classifier.py
```
### Heritage Health dataset:
```
python attribute_classifier_health.py
```

## Results
### Adult Dataset
![adult](Assets/adult_compare.png)

## Health Heritage Dataset
![health](Assets/health_compare.png)
