Title | Fisher-Rao Metric, Geometry, and Complexity of Neural Networks |
Publication Type | Report |
Year of Publication | 2017 |
Authors | Liang, T, Poggio, T, Rakhlin, A, Stokes, J |
Series Title | arXiv.org |
Date Published | 11/2017 |
Other Numbers | arXiv:1711.01530v1 |
Keywords | capacity control, deep learning, Fisher-Rao metric, generalization error, information geometry, Invariance, natural gradient, ReLU activation, statistical learning theory |
Abstract | We study the relationship between geometry and capacity measures for deep neural networks from an invariance viewpoint. We introduce a new notion of capacity — the Fisher-Rao norm — that possesses desirable in- variance properties and is motivated by Information Geometry. We discover an analytical characterization of the new capacity measure, through which we establish norm-comparison inequalities and further show that the new measure serves as an umbrella for several existing norm-based complexity measures. We discuss upper bounds on the generalization error induced by the proposed measure. Extensive numerical experiments on CIFAR-10 support our theoretical findings. Our theoretical analysis rests on a key structural lemma about partial derivatives of multi-layer rectifier networks. |
URL | https://arxiv.org/abs/1711.01530 |
Research Area:
CBMM Relationship:
- CBMM Related