Building Sparse Deep Feedforward Networks using Tree Receptive Fields

Building Sparse Deep Feedforward Networks using Tree Receptive Fields

Xiaopeng Li, Zhourong Chen, Nevin L. Zhang

Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence
Main track. Pages 5045-5051. https://doi.org/10.24963/ijcai.2018/700

Sparse connectivity is an important factor behind the success of convolutional neural networks and recurrent neural networks. In this paper, we consider the problem of learning sparse connectivity for feedforward neural networks (FNNs). The key idea is that a unit should be connected to a small number of units at the next level below that are strongly correlated. We use Chow-Liu's algorithm to learn a tree-structured probabilistic model for the units at the current level, use the tree to identify subsets of units that are strongly correlated, and introduce a new unit with receptive field over the subsets. The procedure is repeated on the new units to build multiple layers of hidden units. The resulting model is called a TRF-net. Empirical results show that, when compared to dense FNNs, TRF-net achieves better or comparable classification performance with much fewer parameters and sparser structures. They are also more interpretable.
Keywords:
Machine Learning: Neural Networks
Uncertainty in AI: Graphical Models
Machine Learning: Feature Selection ; Learning Sparse Models