Neural Interaction Detection (NID)

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

Usage

• 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

Reproducibility

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

Reference

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}
}