| Paper ID | IVMSP-7.4 |
| Paper Title |
A SCALE INVARIANT MEASURE OF FLATNESS FOR DEEP NETWORK MINIMA |
| Authors |
Akshay Rangamani, Massachusetts Institute of Technology, United States; Nam Nguyen, IBM Research, United States; Abhishek Kumar, Google Brain, United States; Dzung Phan, IBM Research, United States; Sang Chin, Boston University, United States; Trac D. Tran, Johns Hopkins University, United States |
| Session | IVMSP-7: Machine Learning for Image Processing I |
| Location | Gather.Town |
| Session Time: | Wednesday, 09 June, 13:00 - 13:45 |
| Presentation Time: | Wednesday, 09 June, 13:00 - 13:45 |
| Presentation |
Poster
|
| Topic |
Image, Video, and Multidimensional Signal Processing: [IVTEC] Image & Video Processing Techniques |
| IEEE Xplore Open Preview |
Click here to view in IEEE Xplore |
| Virtual Presentation |
Click here to watch in the Virtual Conference |
| Abstract |
It has been empirically observed that the flatness of minima obtained from training deep networks seems to correlate with better generalization. However, for deep networks with positively homogeneous activations, most measures of flatness are not invariant to rescaling of the network parameters. This means that the measure of flatness can be made as small or as large as possible through rescaling, rendering the quantitative measures meaningless. In this paper we show that for deep networks with positively homogenous activations, these rescalings constitute equivalence relations, and that these equivalence relations induce a quotient manifold structure in the parameter space. Using an appropriate Riemannian metric, we propose a Hessian-based measure for flatness that is invariant to rescaling and perform simulations to empirically verify our claim. Finally we perform experiments to verify that our flatness measure correlates with generalization by using minibatch stochastic gradient descent with different batch sizes to find deep network minima with different generalization properties. |
