Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Extension to DeepHit #23

Open
Anivader opened this issue Jul 26, 2023 · 2 comments
Open

Extension to DeepHit #23

Anivader opened this issue Jul 26, 2023 · 2 comments

Comments

@Anivader
Copy link

Hello,

This is great work!

I was wondering if SurvSHAP can be applied to Deep Learning survival Networks such as DeepHit that can produce survival function estimates without being constrained by the "Proportional Hazards (PH)" assumption.

Please let me know. I am keen to try it out.
Ani

@krzyzinskim
Copy link
Collaborator

Hi, thanks for your interest!

SurvSHAP(t) can definitely be used for deep learning models, there are no theoretical counter-arguments for that. In fact, it is a good idea because the lack of PH assumption makes DL models more flexible, but thus also more complex and complicated - it is useful to know what is going on inside such a black-box model.

As for the practical side of this problem - currently this Python package is adapted to work with models from the scikit-survival package, where no DL models are available. Other models can, of course, be explained, but their prediction interface must then be adapted to return predictions in the form of sksurv.functions.StepFunction.

@Nadam0707
Copy link

This is an important contribution. Do you have an example implementation of survival models such as DeepHit or Cox-Time?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants