akhan390/DRO_COX

Distributionally Robust Survival Analysis: A Novel Fairness Loss Without Demographics

Shu Hu, George H. Chen

---

Some of codes are extracted from FairSurv.

Requirements

To install requirements:

pip install -r requirements.txt

How to run Dro-COX code

1. To run Dro-COX on FLC

For FLC (Linear),

python -u run_dro_cox.py --dataset FLC --model Linear --eps 0.15 --seed 7 > FLC_Linear_joint_dro_split.log 2>&1 & 

For FLC (MLP),

python -u run_dro_cox.py --dataset FLC --model MLP --eps 0.3 --seed 7 > FLC_MLP_joint_dro_split.log 2>&1 &  

2. To run Dro-COX on SUPPORT

For SUPPORT (Linear),

python -u run_dro_cox.py --dataset SUPPORT --model Linear --eps 0.15 --seed 7 > SUPPORT_Linear_joint_dro_split.log 2>&1 & 

For SUPPORT (MLP),

python -u run_dro_cox.py --dataset SUPPORT --model MLP --eps 0.5 --seed 7 > SUPPORT_MLP_joint_dro_split.log 2>&1 & 

3. To run Dro-COX on SEER

For SEER (Linear),

python -u run_dro_cox.py --dataset SEER --model Linear --eps 0.2 --seed 7 > SEER_Linear_joint_dro_split.log 2>&1 &

For SEER (MLP),

python -u run_dro_cox.py --dataset SEER --model MLP --eps 0.15 --seed 7 > SEER_MLP_joint_dro_split.log 2>&1 & 

How to run Dro-COX (SPLIT) code

1. To run Dro-COX (SPLIT) on FLC

For FLC (Linear),

python -u run_dro_cox_split.py --dataset FLC --model Linear --eps 0.1 --seed 7 > FLC_Linear_joint_dro_split.log 2>&1 & 

For FLC (MLP),

python -u run_dro_cox_split.py --dataset FLC --model MLP --lr 0.0001 --eps 0.05 --seed 7 > FLC_MLP_joint_dro_split.log 2>&1 &  

2. To run Dro-COX (SPLIT) on SUPPORT

For SUPPORT (Linear),

python -u run_dro_cox_split.py --dataset SUPPORT --model Linear --eps 0.15 --seed 7 > SUPPORT_Linear_joint_dro_split.log 2>&1 & 

For SUPPORT (MLP),

python -u run_dro_cox_split.py --dataset SUPPORT --model MLP --lr 0.0001 --eps 0.2 --seed 7 > SUPPORT_MLP_joint_dro_split.log 2>&1 & 

3. To run Dro-COX (SPLIT) on SEER

For SEER (Linear),

python -u run_dro_cox_split.py --dataset SEER --model Linear --eps 0.15 --seed 7 > SEER_Linear_joint_dro_split.log 2>&1 &

For SEER (MLP),

python -u run_dro_cox_split.py --dataset SEER --model MLP --lr 0.0001 --eps 0.2 --seed 7 > SEER_MLP_joint_dro_split.log 2>&1 & 

How to plot Figure 1 ( Dro-COX )

Setting dataset = 'FLC' in plot.py

For FLC,

python plot.py

Setting dataset = 'SUPPORT' in plot.py

For SUPPORT,

python plot.py

Setting dataset = 'SEER' in plot.py

For SEER,

python plot.py

Citation

Please kindly consider citing our paper in your publications.

@inproceedings{hu2022distributionally,
  title={Distributionally Robust Survival Analysis: A Novel Fairness Loss Without Demographics},
  author={Hu, Shu and Chen, George H},
  booktitle={Machine Learning for Health},
  pages={62--87},
  year={2022},
  organization={PMLR}
}