ZackAXue / Transformer_Explainability_442

[CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize classifications by Transformer based networks.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch Implementation of Our Explainability Method for ViT

We implemented our method in Transformer-explainability_combined.ipynb. We also added the several functions containing get_relprop() and compute_layer_rollout_attention() to baselines/ViT/ViT_LRP.py. Other than that the code is from https://github.com/hila-chefer/Transformer-Explainability.

About

[CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize classifications by Transformer based networks.

License:MIT License


Languages

Language:Jupyter Notebook 92.8%Language:Python 7.2%