yidinghao/whitebox-lstm

Code for the paper "Evaluating Attribution Methods using White-Box LSTMs."

White-Box LSTMs

Code for the paper Evaluating Attribution Methods using White-Box LSTMs, to appear in the proceedings of BlackboxNLP.

Dependencies:

• Python 3
• PyTorch
• Torchtext
• Captum

Dependencies for generating heatmaps:

• Yattag
• Matplotlib
• Jupyter Notebook

Usage

Refer to example.ipynb to see how to reproduce the heatmaps from the paper. The scripts lrp_saturation_test.py and ablation_test.py run the LRP saturation test and the ablation test, respectively.