Repository files navigation

M. Tsang,D. Cheng,Y. Liu. Detecting Statistical Interactions from Neural Network Weights, ICLR 2018.[pdf]

• Run the demo at "demo.ipynb"
  • the demo trains a multilayer perceptron (MLP) on synthetic data containing interactions with nonlinearities. at the end of the notebook the interactions are found by decoding the learned weights
• requires python 3.6+ and jupyter notebook, tested with pytorch 1.3.1, scikit-learn 0.21.3, numpy 1.17.1

If you need to reproduce paper results, please contact me so I can share the original code used for experiments (in Tensorflow). Email: tsangm at usc dot edu

If you use NID in your research, please cite the following:

@article{tsang2017detecting,  title={Detecting statistical interactions from neural network weights},  author={Tsang, Michael and Cheng, Dehua and Liu, Yan},  journal={arXiv preprint arXiv:1705.04977},  year={2017}}