How to Guess a Gradient

Published: 26 Oct 2023, Last Modified: 13 Dec 2023NeurIPS 2023 Workshop PosterEveryoneRevisionsBibTeX
Keywords: optimization for deep learning, directional derivative
TL;DR: We narrow down the guessing space of gradients by using knowledge architecture and data. This is useful for backprop-free optimization of neural networks
Abstract: What can you say about the gradient of a neural network without \emph{computing a loss} or \emph{knowing the label?} This may sound like a strange question: surely the answer is ``very little.'' However, in this paper, we show that gradients are more structured than previously thought. They lie in a predictable low-dimensional subspace which depends on the network architecture and incoming features. Exploiting this structure can significantly improve gradient-free optimization schemes based on directional derivatives, which until now have struggled to scale beyond small networks trained on MNIST. We study how to narrow the gap in optimization performance between methods that calculate exact gradients and those that use directional derivatives, demonstrate new phenomena that occur when using these methods, and highlight new challenges in scaling these methods.
Submission Number: 43