Alert button

Sparsifying networks by traversing Geodesics

Dec 12, 2020
Guruprasad Raghavan, Matt Thomson

Figure 1 for Sparsifying networks by traversing Geodesics
Figure 2 for Sparsifying networks by traversing Geodesics

Share this with someone who'll enjoy it:

The geometry of weight spaces and functional manifolds of neural networks play an important role towards 'understanding' the intricacies of ML. In this paper, we attempt to solve certain open questions in ML, by viewing them through the lens of geometry, ultimately relating it to the discovery of points or paths of equivalent function in these spaces. We propose a mathematical framework to evaluate geodesics in the functional space, to find high-performance paths from a dense network to its sparser counterpart. Our results are obtained on VGG-11 trained on CIFAR-10 and MLP's trained on MNIST. Broadly, we demonstrate that the framework is general, and can be applied to a wide variety of problems, ranging from sparsification to alleviating catastrophic forgetting.

* 5 pages; Presented work at NeurIPS 2020 Workshop (DiffGeo4DL). arXiv admin note: text overlap with arXiv:2005.11603  
View paper onarxiv iconopen_review iconOpenReview

Share this with someone who'll enjoy it: