English

Interpretable Architecture Neural Networks for Function Visualization

Machine Learning 2023-03-08 v1 Human-Computer Interaction Methodology Machine Learning

Abstract

In many scientific research fields, understanding and visualizing a black-box function in terms of the effects of all the input variables is of great importance. Existing visualization tools do not allow one to visualize the effects of all the input variables simultaneously. Although one can select one or two of the input variables to visualize via a 2D or 3D plot while holding other variables fixed, this presents an oversimplified and incomplete picture of the model. To overcome this shortcoming, we present a new visualization approach using an interpretable architecture neural network (IANN) to visualize the effects of all the input variables directly and simultaneously. We propose two interpretable structures, each of which can be conveniently represented by a specific IANN, and we discuss a number of possible extensions. We also provide a Python package to implement our proposed method. The supplemental materials are available online.

Keywords

Cite

@article{arxiv.2303.03393,
  title  = {Interpretable Architecture Neural Networks for Function Visualization},
  author = {Shengtong Zhang and Daniel W. Apley},
  journal= {arXiv preprint arXiv:2303.03393},
  year   = {2023}
}