Skip to content

mtsang/neural-interaction-detection

Repository files navigation

Neural Interaction Detection (NID)

Algorithm

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

About

Detecting Statistical Interactions from Neural Network Weights

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published