search

triton-inference-server / fil_backend

FIL backend for the Triton Inference Server
Apache License 2.0
68 stars 35 forks source link

Implement CPU shapley values #339

Closed RAMitchell closed 1 year ago

RAMitchell commented 1 year ago

A rough draft implementing shapley values computed on the CPU. Uses the linear tree shap algorithm (https://arxiv.org/abs/2209.08192).

Is limited to depth 32 due to the need for lookup tables.

RAMitchell commented 1 year ago

Have retested after addressing review comments and everything is passing.