benhuh/Open-L2O

Open-L2O: A Comprehensive and Reproducible Benchmark for Learning to Optimize Algorithms

Open-L2O

This repository establishes the first comprehensive benchmark efforts of existing learning to optimize (L2O) approaches on a number of problems and settings. We release our software implementation and data as the Open-L2O package, for reproducible research and fair benchmarking in the L2O field. [Paper]

Overview

What is learning to optimize (L2O)?

L2O (Learning to optimize) aims to replace manually designed analytic optimization algorithms (SGD, RMSProp, Adam, etc.) with learned update rules.

How does L2O work?

L2O serves as functions that can be fit from data. L2O gains experience from training optimization tasks in a principled and automatic way.

What can L2O do for you?

L2O is particularly suitable for solving a certain type of optimization over a specific distribution of data repeatedly. In comparison to classic methods, L2O is shown to find higher-quality solutions and/or with much faster convergence speed for many problems.

Open questions for research?

• There are significant theoretical and practicality gaps between manually designed optimizers and existing L2O models.

Main Results

Learning to optimize sparse recovery

Learning to optimize Lasso functions

Learning to optimize non-convex Rastrigin functions

Learning to optimize neural networks

Supported Model-base Learnable Optimizers

All codes are available at here.

1. LISTA (feed-forward form) from Learning fast approximations of sparse coding [Paper]
2. LISTA-CP from Theoretical Linear Convergence of Unfolded ISTA and its Practical Weights and Thresholds [Paper]
3. LISTA-CPSS from Theoretical Linear Convergence of Unfolded ISTA and its Practical Weights and Thresholds [Paper]
4. LFISTA from Understanding Trainable Sparse Coding via Matrix Factorization [Paper]
5. LAMP from AMP-Inspired Deep Networks for Sparse Linear Inverse Problems [Paper]
6. ALISTA from ALISTA: Analytic Weights Are As Good As Learned Weights in LISTA [Paper]
7. GLISTA from Sparse Coding with Gated Learned ISTA [Paper]

Supported Model-free Learnable Optimizers

1. L2O-DM from Learning to learn by gradient descent by gradient descent [Paper] [Code]
2. L2O-RNNProp Learning Gradient Descent: Better Generalization and Longer Horizons from [Paper] [Code]
3. L2O-Scale from Learned Optimizers that Scale and Generalize [Paper] [Code]
4. L2O-enhanced from Training Stronger Baselines for Learning to Optimize [Paper] [Code]
5. L2O-Swarm from Learning to Optimize in Swarms [Paper] [Code]
6. L2O-Jacobian from HALO: Hardware-Aware Learning to Optimize [Paper] [Code]
7. L2O-Minmax from Learning A Minimax Optimizer: A Pilot Study [Paper] [Code]

Supported Optimizees

Convex Functions:

• Quadratic
• Lasso

Non-convex Functions:

• Rastrigin

Minmax Functions:

• Saddle
• Rotated Saddle
• Seesaw
• Matrix Game

Neural Networks:

• MLPs on MNIST
• ConvNets on MNIST and CIFAR-10
• LeNet
• NAS searched archtectures

Other Resources

• This is a Pytorch implementation of L2O-DM. [Code]
• This is the original L2O-Swarm repository. [Code]
• This is the original L2O-Jacobian repository. [Code]

Future Works

• TF2.0 Implementated toolbox v2 with a unified framework and lib dependency.

Cite

@misc{chen2021learning,
      title={Learning to Optimize: A Primer and A Benchmark}, 
      author={Tianlong Chen and Xiaohan Chen and Wuyang Chen and Howard Heaton and Jialin Liu and Zhangyang Wang and Wotao Yin},
      year={2021},
      eprint={2103.12828},
      archivePrefix={arXiv},
      primaryClass={math.OC}
}