Skip to content

Commit

Permalink
Add citation (#17)
Browse files Browse the repository at this point in the history
  • Loading branch information
RAMitchell authored Oct 29, 2020
1 parent ddb9d0b commit ffcdd9d
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,18 @@

GPUTreeShap is a cuda implementation of the TreeShap algorithm by Lundberg et al. [1] for Nvidia GPUs. It is a header only module designed to be included in decision tree libraries as a fast backend for model interpretability using SHAP values.

See the associated publication [here](https://arxiv.org/abs/2010.13972)
```
@misc{mitchell2020gputreeshap,
title={GPUTreeShap: Fast Parallel Tree Interpretability},
author={Rory Mitchell and Eibe Frank and Geoffrey Holmes},
year={2020},
eprint={2010.13972},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
```

## Performance
Using the benchmark script `benchmark/benchmark.py` we run GPUTreeShap as a backend for xgboost and compare its performance against multithreaded CPU based implementation. Test models are generated on four different datasets at different sizes. The below comparison is run on an Nvidia DGX-1 system, comparing a single V100 to 2X 20-Core Intel Xeon
E5-2698 CPUs (40 physical cores total).
Expand Down

0 comments on commit ffcdd9d

Please sign in to comment.