Lottery Tickets in Evolutionary Optimization: On Sparse Backpropagation-Free Trainability

Robert Tjarko Lange, Henning Sprekeler
5/31/2023
cs.NEcs.AIcs.LG

No official code found. Best implementation: RobertTLange/es-lottery (17 stars, last commit 2023-06-02).

Abstract

Is the lottery ticket phenomenon an idiosyncrasy of gradient-based training or does it generalize to evolutionary optimization? In this paper we establish the existence of highly sparse trainable initializations for evolution strategies (ES) and characterize qualitative differences compared to gradient descent (GD)-based sparse training. We introduce a novel signal-to-noise iterative pruning procedure, which incorporates loss curvature information into the network pruning step. This can enable the discovery of even sparser trainable network initializations when using black-box evolution as compared to GD-based optimization. Furthermore, we find that these initializations encode an inductive bias, which transfers across different ES, related tasks and even to GD-based training. Finally, we compare the local optima resulting from the different optimization paradigms and sparsity levels. In contrast to GD, ES explore diverse and flat local optima and do not preserve linear mode connectivity across sparsity levels and independent runs. The results highlight qualitative differences between evolution and gradient-based learning dynamics, which can be uncovered by the study of iterative pruning procedures.

View on arXivView PDF

Code Implementations(1)

Lottery Tickets in Evolutionary Optimization (Lange & Sprekeler, ICML 2023)

172Python, Jupyter NotebookDec 2, 20223 years ago

Similar Papers

Cite this paper

@article{lange2023lottery,
  title  = {Lottery Tickets in Evolutionary Optimization: On Sparse Backpropagation-Free Trainability},
  author = {Robert Tjarko Lange and Henning Sprekeler},
  year   = {2023},
  eprint = {2306.00045},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2306.00045v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2306.00045.svg)](https://codeofpaper.com/paper/2306.00045)

Discussion