Skip to content
/ CACHE Public

[ML4H 2022] This is the code for our paper `Counterfactual and Factual Reasoning over Hypergraphs for Interpretable Clinical Predictions on EHR'.

License

Notifications You must be signed in to change notification settings

ritaranx/CACHE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CACHE

This repo contains our code for paper Counterfactual and Factual Reasoning over Hypergraphs for Interpretable Clinical Predictions on EHR, in Proceedings of 2nd Machine Learning for Health symposium 2022 (ML4H 2022).

Model Framework

CACHE-Framework

Data

[IMPORTANT, PLEASE READ!]

In order to facilitate the reproducibility, we provide two toy datasets in data. Note that although they're named as mimic3 and cradle (the two datasets we mentioned in our paper), they're in fact randomly generated due to the privacy issue. We include them in this repo only to show the format of the two datasts we used. Thus, their experimental results should NOT reflect the performance we report in the paper.

Package

  • PyTorch 1.4
  • python 3.7
  • tqdm
  • torch-scatter 2.0.4
  • torch-sparse 0.6.0
  • torch-cluster 1.5.2
  • torch-geometric 1.6.3
  • sklearn

Run the Code

Please use run.sh in src to run the code for the two toy datasets in data. It runs four experiments:

  • CACHE for MIMIC-III dataset
  • CACHE for CRADLE dataset
  • vanilla backbone model for MIMIC-III dataset
  • vanilla backbone model for CRADLE dataset

Citation

If you find this paper useful for your research, please cite the following in your publication. Thanks!

@inproceedings{xu2022counterfactual,
  title={Counterfactual and Factual Reasoning over Hypergraphs for Interpretable Clinical Predictions on EHR},
  author={Xu, Ran and Yu, Yue and Zhang, Chao and Ali, Mohammed K and Ho, Joyce C and Yang, Carl},
  booktitle={Machine Learning for Health},
  pages={259--278},
  year={2022},
  organization={PMLR}
}

Acknowledgement

We would like to thank the authors from AllSet for their open-source efforts.

About

[ML4H 2022] This is the code for our paper `Counterfactual and Factual Reasoning over Hypergraphs for Interpretable Clinical Predictions on EHR'.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published