Symmetry Teleportation for Accelerated Optimization

Bo Zhao, Nima Dehmamy, Robin Walters, Rose Yu

Keywords: optimization, parameter space symmetry, teleportation

Abstract: Existing gradient-based optimization methods update parameters locally, in a direction that minimizes the loss function. We study a different approach, symmetry teleportation, that allows parameters to travel a large distance on the loss level set, in order to improve the convergence speed in subsequent steps. Teleportation exploits symmetries in the loss landscape of optimization problems. We derive loss-invariant group actions for test functions in optimization and multi-layer neural networks, and prove a necessary condition for teleportation to improve convergence rate. We also show that our algorithm is closely related to second order methods. Experimentally, we show that teleportation improves the convergence speed of gradient descent and AdaGrad for several optimization problems including test functions, multi-layer regressions, and MNIST classification.

TL;DR: We propose an accelerated gradient-based optimization algorithm that exploits symmetries in the loss landscape.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/arxiv:2205.10637/code)