search

pytorch / captum

Model interpretability and understanding for PyTorch
https://captum.ai
BSD 3-Clause "New" or "Revised" License
4.94k stars 499 forks source link

LRP for 3D pytorch models #616

Open linhlpv opened 3 years ago

linhlpv commented 3 years ago

🚀 Feature

LRP supports for 3D model, such as 3d video model in torchvision.models

Motivation

Now, 3D models, such as 3D conv for action recognition, video classification, ... is more and more popular. So LRP for 3D model will help to understand how these models work is necessary. E.g: For understanding 3D in Alzheimer’s disease classification, the author use LRP: https://github.com/moboehle/Pytorch-LRP

Pitch

LRP work for 3D models such as r2plus1d, mc3, ...

Alternatives

...

Additional context

None

Mushtaqml commented 3 years ago

Any progress on this feature?. If not then can someone suggest how to interpret or understand a 3d cnn model?